package ipfs

import (
	"bytes"
	"crypto/rand"
	"encoding/hex"
	"encoding/json"
	"fmt"
	"io"
	"net"
	"net/http"
	"net/url"
	"os"
	"os/exec"
	"path/filepath"
	"strings"
	"time"

	"go.uber.org/zap"

	"github.com/DeBrosOfficial/network/pkg/config"
	"github.com/libp2p/go-libp2p/core/host"
	"github.com/multiformats/go-multiaddr"
)

// ClusterConfigManager manages IPFS Cluster configuration files
type ClusterConfigManager struct {
	cfg         *config.Config
	logger      *zap.Logger
	clusterPath string
	secret      string
}

// ClusterServiceConfig represents the structure of service.json
type ClusterServiceConfig struct {
	Cluster struct {
		Peername           string   `json:"peername"`
		Secret             string   `json:"secret"`
		LeaveOnShutdown    bool     `json:"leave_on_shutdown"`
		ListenMultiaddress []string `json:"listen_multiaddress"`
		PeerAddresses      []string `json:"peer_addresses"`
		// ... other fields kept from template
	} `json:"cluster"`
	Consensus struct {
		CRDT struct {
			ClusterName  string   `json:"cluster_name"`
			TrustedPeers []string `json:"trusted_peers"`
			Batching     struct {
				MaxBatchSize int    `json:"max_batch_size"`
				MaxBatchAge  string `json:"max_batch_age"`
			} `json:"batching"`
			RepairInterval string `json:"repair_interval"`
		} `json:"crdt"`
	} `json:"consensus"`
	API struct {
		IPFSProxy struct {
			ListenMultiaddress string `json:"listen_multiaddress"`
			NodeMultiaddress   string `json:"node_multiaddress"`
		} `json:"ipfsproxy"`
		PinSvcAPI struct {
			HTTPListenMultiaddress string `json:"http_listen_multiaddress"`
		} `json:"pinsvcapi"`
		RestAPI struct {
			HTTPListenMultiaddress string `json:"http_listen_multiaddress"`
		} `json:"restapi"`
	} `json:"api"`
	IPFSConnector struct {
		IPFSHTTP struct {
			NodeMultiaddress string `json:"node_multiaddress"`
		} `json:"ipfshttp"`
	} `json:"ipfs_connector"`
	// Keep rest of fields as raw JSON to preserve structure
	Raw map[string]interface{} `json:"-"`
}

// NewClusterConfigManager creates a new IPFS Cluster config manager
func NewClusterConfigManager(cfg *config.Config, logger *zap.Logger) (*ClusterConfigManager, error) {
	// Expand data directory path
	dataDir := cfg.Node.DataDir
	if strings.HasPrefix(dataDir, "~") {
		home, err := os.UserHomeDir()
		if err != nil {
			return nil, fmt.Errorf("failed to determine home directory: %w", err)
		}
		dataDir = filepath.Join(home, dataDir[1:])
	}

	// Determine cluster path based on data directory structure
	// Check if dataDir contains specific node names (e.g., ~/.orama/bootstrap, ~/.orama/bootstrap2, ~/.orama/node2-4)
	clusterPath := filepath.Join(dataDir, "ipfs-cluster")
	nodeNames := []string{"bootstrap", "bootstrap2", "node2", "node3", "node4"}
	for _, nodeName := range nodeNames {
		if strings.Contains(dataDir, nodeName) {
			// Check if this is a direct child
			if filepath.Base(filepath.Dir(dataDir)) == nodeName || filepath.Base(dataDir) == nodeName {
				clusterPath = filepath.Join(dataDir, "ipfs-cluster")
			} else {
				clusterPath = filepath.Join(dataDir, nodeName, "ipfs-cluster")
			}
			break
		}
	}

	// Load or generate cluster secret
	secretPath := filepath.Join(dataDir, "..", "cluster-secret")
	if strings.Contains(dataDir, ".orama") {
		// Try to find cluster-secret in ~/.orama
		home, err := os.UserHomeDir()
		if err == nil {
			secretPath = filepath.Join(home, ".orama", "cluster-secret")
		}
	}

	secret, err := loadOrGenerateClusterSecret(secretPath)
	if err != nil {
		return nil, fmt.Errorf("failed to load/generate cluster secret: %w", err)
	}

	return &ClusterConfigManager{
		cfg:         cfg,
		logger:      logger,
		clusterPath: clusterPath,
		secret:      secret,
	}, nil
}

// EnsureConfig ensures the IPFS Cluster service.json exists and is properly configured
func (cm *ClusterConfigManager) EnsureConfig() error {
	if cm.cfg.Database.IPFS.ClusterAPIURL == "" {
		cm.logger.Debug("IPFS Cluster API URL not configured, skipping cluster config")
		return nil
	}

	serviceJSONPath := filepath.Join(cm.clusterPath, "service.json")

	// Parse ports from URLs
	clusterPort, restAPIPort, err := parseClusterPorts(cm.cfg.Database.IPFS.ClusterAPIURL)
	if err != nil {
		return fmt.Errorf("failed to parse cluster API URL: %w", err)
	}

	ipfsPort, err := parseIPFSPort(cm.cfg.Database.IPFS.APIURL)
	if err != nil {
		return fmt.Errorf("failed to parse IPFS API URL: %w", err)
	}

	// Determine node name
	nodeName := cm.cfg.Node.Type
	if nodeName == "node" || nodeName == "bootstrap" {
		// Try to extract from data dir or ID
		possibleNames := []string{"bootstrap", "bootstrap2", "node2", "node3", "node4"}
		for _, name := range possibleNames {
			if strings.Contains(cm.cfg.Node.DataDir, name) || strings.Contains(cm.cfg.Node.ID, name) {
				nodeName = name
				break
			}
		}
		if nodeName == "node" || nodeName == "bootstrap" {
			nodeName = cm.cfg.Node.Type
		}
	}

	// Calculate ports based on pattern
	// REST API: 9094
	// Proxy: 9094 - 1 = 9093 (NOT USED - keeping for reference)
	// PinSvc: 9094 + 1 = 9095
	// Proxy API: 9094 + 1 = 9095 (actual proxy port)
	// PinSvc API: 9094 + 3 = 9097
	// Cluster LibP2P: 9094 + 4 = 9098
	proxyPort := clusterPort + 1         // 9095 (IPFSProxy API)
	pinSvcPort := clusterPort + 3        // 9097 (PinSvc API)
	clusterListenPort := clusterPort + 4 // 9098 (Cluster LibP2P)

	// If config doesn't exist, initialize it with ipfs-cluster-service init
	// This ensures we have all required sections (datastore, informer, etc.)
	if _, err := os.Stat(serviceJSONPath); os.IsNotExist(err) {
		cm.logger.Info("Initializing cluster config with ipfs-cluster-service init")
		initCmd := exec.Command("ipfs-cluster-service", "init", "--force")
		initCmd.Env = append(os.Environ(), "IPFS_CLUSTER_PATH="+cm.clusterPath)
		if err := initCmd.Run(); err != nil {
			cm.logger.Warn("Failed to initialize cluster config with ipfs-cluster-service init, will create minimal template", zap.Error(err))
		}
	}

	// Load existing config or create new
	cfg, err := cm.loadOrCreateConfig(serviceJSONPath)
	if err != nil {
		return fmt.Errorf("failed to load/create config: %w", err)
	}

	// Update configuration
	cfg.Cluster.Peername = nodeName
	cfg.Cluster.Secret = cm.secret
	cfg.Cluster.ListenMultiaddress = []string{fmt.Sprintf("/ip4/0.0.0.0/tcp/%d", clusterListenPort)}
	cfg.Consensus.CRDT.ClusterName = "debros-cluster"
	cfg.Consensus.CRDT.TrustedPeers = []string{"*"}

	// API endpoints
	cfg.API.RestAPI.HTTPListenMultiaddress = fmt.Sprintf("/ip4/0.0.0.0/tcp/%d", restAPIPort)
	cfg.API.IPFSProxy.ListenMultiaddress = fmt.Sprintf("/ip4/127.0.0.1/tcp/%d", proxyPort)
	cfg.API.IPFSProxy.NodeMultiaddress = fmt.Sprintf("/ip4/127.0.0.1/tcp/%d", ipfsPort) // FIX: Correct path!
	cfg.API.PinSvcAPI.HTTPListenMultiaddress = fmt.Sprintf("/ip4/127.0.0.1/tcp/%d", pinSvcPort)

	// IPFS connector (also needs to be set)
	cfg.IPFSConnector.IPFSHTTP.NodeMultiaddress = fmt.Sprintf("/ip4/127.0.0.1/tcp/%d", ipfsPort)

	// Save configuration
	if err := cm.saveConfig(serviceJSONPath, cfg); err != nil {
		return fmt.Errorf("failed to save config: %w", err)
	}

	cm.logger.Info("IPFS Cluster configuration ensured",
		zap.String("path", serviceJSONPath),
		zap.String("node_name", nodeName),
		zap.Int("ipfs_port", ipfsPort),
		zap.Int("cluster_port", clusterPort),
		zap.Int("rest_api_port", restAPIPort))

	return nil
}

// UpdateBootstrapPeers updates peer_addresses and peerstore with bootstrap peer information
// Returns true if update was successful, false if bootstrap is not available yet (non-fatal)
func (cm *ClusterConfigManager) UpdateBootstrapPeers(bootstrapAPIURL string) (bool, error) {
	if cm.cfg.Database.IPFS.ClusterAPIURL == "" {
		return false, nil // IPFS not configured
	}

	// Skip if this is the bootstrap node itself
	if cm.cfg.Node.Type == "bootstrap" {
		return false, nil
	}

	// Query bootstrap cluster API to get peer ID
	peerID, err := getBootstrapPeerID(bootstrapAPIURL)
	if err != nil {
		// Non-fatal: bootstrap might not be available yet
		cm.logger.Debug("Bootstrap peer not available yet, will retry",
			zap.String("bootstrap_api", bootstrapAPIURL),
			zap.Error(err))
		return false, nil
	}

	if peerID == "" {
		cm.logger.Debug("Bootstrap peer ID not available yet")
		return false, nil
	}

	// Extract bootstrap host and cluster port from URL
	bootstrapHost, clusterPort, err := parseBootstrapHostAndPort(bootstrapAPIURL)
	if err != nil {
		return false, fmt.Errorf("failed to parse bootstrap cluster API URL: %w", err)
	}

	// Bootstrap cluster LibP2P listens on clusterPort + 4
	// (REST API is 9094, LibP2P is 9098 = 9094 + 4)
	bootstrapClusterPort := clusterPort + 4

	// Determine IP protocol (ip4 or ip6) based on the host
	var ipProtocol string
	if net.ParseIP(bootstrapHost).To4() != nil {
		ipProtocol = "ip4"
	} else {
		ipProtocol = "ip6"
	}

	bootstrapPeerAddr := fmt.Sprintf("/%s/%s/tcp/%d/p2p/%s", ipProtocol, bootstrapHost, bootstrapClusterPort, peerID)

	// Load current config
	serviceJSONPath := filepath.Join(cm.clusterPath, "service.json")
	cfg, err := cm.loadOrCreateConfig(serviceJSONPath)
	if err != nil {
		return false, fmt.Errorf("failed to load config: %w", err)
	}

	// CRITICAL: Always update peerstore file to ensure no stale addresses remain
	// Stale addresses (e.g., from old port configurations) cause LibP2P dial backoff,
	// preventing cluster peers from connecting even if the correct address is present.
	// We must clean and rewrite the peerstore on every update to avoid this.
	peerstorePath := filepath.Join(cm.clusterPath, "peerstore")

	// Check if peerstore needs updating (avoid unnecessary writes but always clean stale entries)
	needsUpdate := true
	if peerstoreData, err := os.ReadFile(peerstorePath); err == nil {
		// Only skip update if peerstore contains EXACTLY the correct address and nothing else
		existingAddrs := strings.Split(strings.TrimSpace(string(peerstoreData)), "\n")
		if len(existingAddrs) == 1 && strings.TrimSpace(existingAddrs[0]) == bootstrapPeerAddr {
			cm.logger.Debug("Bootstrap peer address already correct in peerstore", zap.String("addr", bootstrapPeerAddr))
			needsUpdate = false
		}
	}

	if needsUpdate {
		// Write ONLY the correct bootstrap peer address, removing any stale entries
		if err := os.WriteFile(peerstorePath, []byte(bootstrapPeerAddr+"\n"), 0644); err != nil {
			return false, fmt.Errorf("failed to write peerstore: %w", err)
		}
		cm.logger.Info("Updated peerstore with bootstrap peer (cleaned stale entries)",
			zap.String("addr", bootstrapPeerAddr),
			zap.String("peerstore_path", peerstorePath))
	}

	// Then sync service.json from peerstore to keep them in sync
	cfg.Cluster.PeerAddresses = []string{bootstrapPeerAddr}

	// Save config
	if err := cm.saveConfig(serviceJSONPath, cfg); err != nil {
		return false, fmt.Errorf("failed to save config: %w", err)
	}

	cm.logger.Info("Updated bootstrap peer configuration",
		zap.String("bootstrap_peer_addr", bootstrapPeerAddr),
		zap.String("peerstore_path", peerstorePath))

	return true, nil
}

// UpdateAllClusterPeers discovers all cluster peers from the local cluster API
// and updates peer_addresses in service.json. This allows IPFS Cluster to automatically
// connect to all discovered peers in the cluster.
// Returns true if update was successful, false if cluster is not available yet (non-fatal)
func (cm *ClusterConfigManager) UpdateAllClusterPeers() (bool, error) {
	if cm.cfg.Database.IPFS.ClusterAPIURL == "" {
		return false, nil // IPFS not configured
	}

	// Query local cluster API to get all peers
	client := &standardHTTPClient{}
	peersURL := fmt.Sprintf("%s/peers", cm.cfg.Database.IPFS.ClusterAPIURL)
	resp, err := client.Get(peersURL)
	if err != nil {
		// Non-fatal: cluster might not be available yet
		cm.logger.Debug("Cluster API not available yet, will retry",
			zap.String("peers_url", peersURL),
			zap.Error(err))
		return false, nil
	}

	// Parse NDJSON response
	dec := json.NewDecoder(bytes.NewReader(resp))
	var allPeerAddresses []string
	seenPeers := make(map[string]bool)
	peerIDToAddresses := make(map[string][]string)

	// First pass: collect all peer IDs and their addresses
	for {
		var peerInfo struct {
			ID                    string   `json:"id"`
			Addresses             []string `json:"addresses"`
			ClusterPeers          []string `json:"cluster_peers"`
			ClusterPeersAddresses []string `json:"cluster_peers_addresses"`
		}

		err := dec.Decode(&peerInfo)
		if err != nil {
			if err == io.EOF {
				break
			}
			cm.logger.Debug("Failed to decode peer info", zap.Error(err))
			continue
		}

		// Store this peer's addresses
		if peerInfo.ID != "" {
			peerIDToAddresses[peerInfo.ID] = peerInfo.Addresses
		}

		// Also collect cluster peers addresses if available
		// These are addresses of all peers in the cluster
		for _, addr := range peerInfo.ClusterPeersAddresses {
			if ma, err := multiaddr.NewMultiaddr(addr); err == nil {
				// Validate it has p2p component (peer ID)
				if _, err := ma.ValueForProtocol(multiaddr.P_P2P); err == nil {
					addrStr := ma.String()
					if !seenPeers[addrStr] {
						allPeerAddresses = append(allPeerAddresses, addrStr)
						seenPeers[addrStr] = true
					}
				}
			}
		}
	}

	// If we didn't get cluster_peers_addresses, try to construct them from peer IDs and addresses
	if len(allPeerAddresses) == 0 && len(peerIDToAddresses) > 0 {
		// Get cluster listen port from config
		serviceJSONPath := filepath.Join(cm.clusterPath, "service.json")
		cfg, err := cm.loadOrCreateConfig(serviceJSONPath)
		if err == nil && len(cfg.Cluster.ListenMultiaddress) > 0 {
			// Extract port from listen_multiaddress (e.g., "/ip4/0.0.0.0/tcp/9098")
			listenAddr := cfg.Cluster.ListenMultiaddress[0]
			if ma, err := multiaddr.NewMultiaddr(listenAddr); err == nil {
				if port, err := ma.ValueForProtocol(multiaddr.P_TCP); err == nil {
					// For each peer ID, try to find its IP address and construct cluster multiaddr
					for peerID, addresses := range peerIDToAddresses {
						// Try to find an IP address in the peer's addresses
						for _, addrStr := range addresses {
							if ma, err := multiaddr.NewMultiaddr(addrStr); err == nil {
								// Extract IP address (IPv4 or IPv6)
								if ip, err := ma.ValueForProtocol(multiaddr.P_IP4); err == nil && ip != "" {
									clusterAddr := fmt.Sprintf("/ip4/%s/tcp/%s/p2p/%s", ip, port, peerID)
									if !seenPeers[clusterAddr] {
										allPeerAddresses = append(allPeerAddresses, clusterAddr)
										seenPeers[clusterAddr] = true
									}
									break
								} else if ip, err := ma.ValueForProtocol(multiaddr.P_IP6); err == nil && ip != "" {
									clusterAddr := fmt.Sprintf("/ip6/%s/tcp/%s/p2p/%s", ip, port, peerID)
									if !seenPeers[clusterAddr] {
										allPeerAddresses = append(allPeerAddresses, clusterAddr)
										seenPeers[clusterAddr] = true
									}
									break
								}
							}
						}
					}
				}
			}
		}
	}

	if len(allPeerAddresses) == 0 {
		cm.logger.Debug("No cluster peer addresses found in API response")
		return false, nil
	}

	// Load current config
	serviceJSONPath := filepath.Join(cm.clusterPath, "service.json")
	cfg, err := cm.loadOrCreateConfig(serviceJSONPath)
	if err != nil {
		return false, fmt.Errorf("failed to load config: %w", err)
	}

	// Check if peer addresses have changed
	addressesChanged := false
	if len(cfg.Cluster.PeerAddresses) != len(allPeerAddresses) {
		addressesChanged = true
	} else {
		// Check if addresses are different
		currentAddrs := make(map[string]bool)
		for _, addr := range cfg.Cluster.PeerAddresses {
			currentAddrs[addr] = true
		}
		for _, addr := range allPeerAddresses {
			if !currentAddrs[addr] {
				addressesChanged = true
				break
			}
		}
	}

	if !addressesChanged {
		cm.logger.Debug("Cluster peer addresses already up to date",
			zap.Int("peer_count", len(allPeerAddresses)))
		return true, nil
	}

	// Update peerstore file FIRST - this is what IPFS Cluster reads for bootstrapping
	// Peerstore is the source of truth, service.json is just for our tracking
	peerstorePath := filepath.Join(cm.clusterPath, "peerstore")
	peerstoreContent := strings.Join(allPeerAddresses, "\n") + "\n"
	if err := os.WriteFile(peerstorePath, []byte(peerstoreContent), 0644); err != nil {
		cm.logger.Warn("Failed to update peerstore file", zap.Error(err))
		// Non-fatal, continue
	}

	// Then sync service.json from peerstore to keep them in sync
	cfg.Cluster.PeerAddresses = allPeerAddresses

	// Save config
	if err := cm.saveConfig(serviceJSONPath, cfg); err != nil {
		return false, fmt.Errorf("failed to save config: %w", err)
	}

	cm.logger.Info("Updated cluster peer addresses",
		zap.Int("peer_count", len(allPeerAddresses)),
		zap.Strings("peer_addresses", allPeerAddresses))

	return true, nil
}

// RepairBootstrapPeers automatically discovers and repairs bootstrap peer configuration
// Tries multiple methods: config-based discovery, bootstrap peer multiaddr, or discovery service
func (cm *ClusterConfigManager) RepairBootstrapPeers() (bool, error) {
	if cm.cfg.Database.IPFS.ClusterAPIURL == "" {
		return false, nil // IPFS not configured
	}

	// Skip if this is the bootstrap node itself
	if cm.cfg.Node.Type == "bootstrap" {
		return false, nil
	}

	// Method 1: Try to use bootstrap API URL from config if available
	// Check if we have a bootstrap node's cluster API URL in discovery metadata
	// For now, we'll infer from bootstrap peers multiaddr

	var bootstrapAPIURL string

	// Try to extract from bootstrap peers multiaddr
	if len(cm.cfg.Discovery.BootstrapPeers) > 0 {
		if ip := extractIPFromMultiaddrForCluster(cm.cfg.Discovery.BootstrapPeers[0]); ip != "" {
			// Default cluster API port is 9094
			bootstrapAPIURL = fmt.Sprintf("http://%s:9094", ip)
			cm.logger.Debug("Inferred bootstrap cluster API from bootstrap peer",
				zap.String("bootstrap_api", bootstrapAPIURL))
		}
	}

	// Fallback to localhost if nothing found (for local development)
	if bootstrapAPIURL == "" {
		bootstrapAPIURL = "http://localhost:9094"
		cm.logger.Debug("Using localhost fallback for bootstrap cluster API")
	}

	// Try to update bootstrap peers
	success, err := cm.UpdateBootstrapPeers(bootstrapAPIURL)
	if err != nil {
		return false, err
	}

	if success {
		cm.logger.Info("Successfully repaired bootstrap peer configuration")
		return true, nil
	}

	// If update failed (bootstrap not available), return false but no error
	// This allows retries later
	return false, nil
}

// DiscoverClusterPeersFromLibP2P loads IPFS cluster peer addresses from the peerstore file.
// If peerstore is empty, it means there are no peers to connect to.
// Returns true if peers were loaded and configured, false otherwise (non-fatal)
func (cm *ClusterConfigManager) DiscoverClusterPeersFromLibP2P(host host.Host) (bool, error) {
	if cm.cfg.Database.IPFS.ClusterAPIURL == "" {
		return false, nil // IPFS not configured
	}

	// Load peer addresses from peerstore file
	peerstorePath := filepath.Join(cm.clusterPath, "peerstore")
	peerstoreData, err := os.ReadFile(peerstorePath)
	if err != nil {
		// Peerstore file doesn't exist or can't be read - no peers to connect to
		cm.logger.Debug("Peerstore file not found or empty - no cluster peers to connect to",
			zap.String("peerstore_path", peerstorePath))
		return false, nil
	}

	var allPeerAddresses []string
	seenPeers := make(map[string]bool)

	// Parse peerstore file (one multiaddr per line)
	lines := strings.Split(strings.TrimSpace(string(peerstoreData)), "\n")
	for _, line := range lines {
		line = strings.TrimSpace(line)
		if line != "" && strings.HasPrefix(line, "/") {
			// Validate it's a proper multiaddr with p2p component
			if ma, err := multiaddr.NewMultiaddr(line); err == nil {
				if _, err := ma.ValueForProtocol(multiaddr.P_P2P); err == nil {
					if !seenPeers[line] {
						allPeerAddresses = append(allPeerAddresses, line)
						seenPeers[line] = true
						cm.logger.Debug("Loaded cluster peer address from peerstore",
							zap.String("addr", line))
					}
				}
			}
		}
	}

	if len(allPeerAddresses) == 0 {
		cm.logger.Debug("Peerstore file is empty - no cluster peers to connect to")
		return false, nil
	}

	// Get config to update peer_addresses
	serviceJSONPath := filepath.Join(cm.clusterPath, "service.json")
	cfg, err := cm.loadOrCreateConfig(serviceJSONPath)
	if err != nil {
		return false, fmt.Errorf("failed to load config: %w", err)
	}

	// Check if peer addresses have changed
	addressesChanged := false
	if len(cfg.Cluster.PeerAddresses) != len(allPeerAddresses) {
		addressesChanged = true
	} else {
		currentAddrs := make(map[string]bool)
		for _, addr := range cfg.Cluster.PeerAddresses {
			currentAddrs[addr] = true
		}
		for _, addr := range allPeerAddresses {
			if !currentAddrs[addr] {
				addressesChanged = true
				break
			}
		}
	}

	if !addressesChanged {
		cm.logger.Debug("Cluster peer addresses already up to date",
			zap.Int("peer_count", len(allPeerAddresses)))
		return true, nil
	}

	// Update peer_addresses
	cfg.Cluster.PeerAddresses = allPeerAddresses

	// Save config
	if err := cm.saveConfig(serviceJSONPath, cfg); err != nil {
		return false, fmt.Errorf("failed to save config: %w", err)
	}

	cm.logger.Info("Loaded cluster peer addresses from peerstore",
		zap.Int("peer_count", len(allPeerAddresses)),
		zap.Strings("peer_addresses", allPeerAddresses))

	return true, nil
}

// loadOrCreateConfig loads existing service.json or creates a template
func (cm *ClusterConfigManager) loadOrCreateConfig(path string) (*ClusterServiceConfig, error) {
	// Try to load existing config
	if data, err := os.ReadFile(path); err == nil {
		var cfg ClusterServiceConfig
		if err := json.Unmarshal(data, &cfg); err == nil {
			// Also unmarshal into raw map to preserve all fields
			var raw map[string]interface{}
			if err := json.Unmarshal(data, &raw); err == nil {
				cfg.Raw = raw
			}
			return &cfg, nil
		}
	}

	// Create new config from template
	return cm.createTemplateConfig(), nil
}

// createTemplateConfig creates a template configuration matching the structure
func (cm *ClusterConfigManager) createTemplateConfig() *ClusterServiceConfig {
	cfg := &ClusterServiceConfig{}
	cfg.Cluster.LeaveOnShutdown = false
	cfg.Cluster.PeerAddresses = []string{}
	cfg.Consensus.CRDT.TrustedPeers = []string{"*"}
	cfg.Consensus.CRDT.Batching.MaxBatchSize = 0
	cfg.Consensus.CRDT.Batching.MaxBatchAge = "0s"
	cfg.Consensus.CRDT.RepairInterval = "1h0m0s"
	cfg.Raw = make(map[string]interface{})
	return cfg
}

// saveConfig saves the configuration, preserving all existing fields
func (cm *ClusterConfigManager) saveConfig(path string, cfg *ClusterServiceConfig) error {
	// Create directory if needed
	if err := os.MkdirAll(filepath.Dir(path), 0755); err != nil {
		return fmt.Errorf("failed to create cluster directory: %w", err)
	}

	// Load existing config if it exists to preserve all fields
	var final map[string]interface{}
	if data, err := os.ReadFile(path); err == nil {
		if err := json.Unmarshal(data, &final); err != nil {
			// If parsing fails, start fresh
			final = make(map[string]interface{})
		}
	} else {
		final = make(map[string]interface{})
	}

	// Deep merge: update nested structures while preserving other fields
	updateNestedMap(final, "cluster", map[string]interface{}{
		"peername":            cfg.Cluster.Peername,
		"secret":              cfg.Cluster.Secret,
		"leave_on_shutdown":   cfg.Cluster.LeaveOnShutdown,
		"listen_multiaddress": cfg.Cluster.ListenMultiaddress,
		"peer_addresses":      cfg.Cluster.PeerAddresses,
	})

	updateNestedMap(final, "consensus", map[string]interface{}{
		"crdt": map[string]interface{}{
			"cluster_name":  cfg.Consensus.CRDT.ClusterName,
			"trusted_peers": cfg.Consensus.CRDT.TrustedPeers,
			"batching": map[string]interface{}{
				"max_batch_size": cfg.Consensus.CRDT.Batching.MaxBatchSize,
				"max_batch_age":  cfg.Consensus.CRDT.Batching.MaxBatchAge,
			},
			"repair_interval": cfg.Consensus.CRDT.RepairInterval,
		},
	})

	// Update API section, preserving other fields
	updateNestedMap(final, "api", map[string]interface{}{
		"ipfsproxy": map[string]interface{}{
			"listen_multiaddress": cfg.API.IPFSProxy.ListenMultiaddress,
			"node_multiaddress":   cfg.API.IPFSProxy.NodeMultiaddress, // FIX: Correct path!
		},
		"pinsvcapi": map[string]interface{}{
			"http_listen_multiaddress": cfg.API.PinSvcAPI.HTTPListenMultiaddress,
		},
		"restapi": map[string]interface{}{
			"http_listen_multiaddress": cfg.API.RestAPI.HTTPListenMultiaddress,
		},
	})

	// Update IPFS connector section
	updateNestedMap(final, "ipfs_connector", map[string]interface{}{
		"ipfshttp": map[string]interface{}{
			"node_multiaddress":         cfg.IPFSConnector.IPFSHTTP.NodeMultiaddress,
			"connect_swarms_delay":      "30s",
			"ipfs_request_timeout":      "5m0s",
			"pin_timeout":               "2m0s",
			"unpin_timeout":             "3h0m0s",
			"repogc_timeout":            "24h0m0s",
			"informer_trigger_interval": 0,
		},
	})

	// Ensure all required sections exist with defaults if missing
	ensureRequiredSection(final, "datastore", map[string]interface{}{
		"pebble": map[string]interface{}{
			"pebble_options": map[string]interface{}{
				"cache_size_bytes": 1073741824,
				"bytes_per_sync":   1048576,
				"disable_wal":      false,
			},
		},
	})

	ensureRequiredSection(final, "informer", map[string]interface{}{
		"disk": map[string]interface{}{
			"metric_ttl":  "30s",
			"metric_type": "freespace",
		},
		"pinqueue": map[string]interface{}{
			"metric_ttl":         "30s",
			"weight_bucket_size": 100000,
		},
		"tags": map[string]interface{}{
			"metric_ttl": "30s",
			"tags": map[string]interface{}{
				"group": "default",
			},
		},
	})

	ensureRequiredSection(final, "monitor", map[string]interface{}{
		"pubsubmon": map[string]interface{}{
			"check_interval": "15s",
		},
	})

	ensureRequiredSection(final, "pin_tracker", map[string]interface{}{
		"stateless": map[string]interface{}{
			"concurrent_pins":          10,
			"priority_pin_max_age":     "24h0m0s",
			"priority_pin_max_retries": 5,
		},
	})

	ensureRequiredSection(final, "allocator", map[string]interface{}{
		"balanced": map[string]interface{}{
			"allocate_by": []interface{}{"tag:group", "freespace"},
		},
	})

	// Write JSON
	data, err := json.MarshalIndent(final, "", "  ")
	if err != nil {
		return fmt.Errorf("failed to marshal config: %w", err)
	}

	if err := os.WriteFile(path, data, 0644); err != nil {
		return fmt.Errorf("failed to write config: %w", err)
	}

	return nil
}

// updateNestedMap updates a nested map structure, merging values
func updateNestedMap(parent map[string]interface{}, key string, updates map[string]interface{}) {
	existing, ok := parent[key].(map[string]interface{})
	if !ok {
		parent[key] = updates
		return
	}

	// Merge updates into existing
	for k, v := range updates {
		if vm, ok := v.(map[string]interface{}); ok {
			// Recursively merge nested maps
			if _, ok := existing[k].(map[string]interface{}); !ok {
				existing[k] = vm
			} else {
				updateNestedMap(existing, k, vm)
			}
		} else {
			existing[k] = v
		}
	}
	parent[key] = existing
}

// ensureRequiredSection ensures a section exists in the config, creating it with defaults if missing
func ensureRequiredSection(parent map[string]interface{}, key string, defaults map[string]interface{}) {
	if _, exists := parent[key]; !exists {
		parent[key] = defaults
		return
	}
	// If section exists, merge defaults to ensure all required subsections exist
	existing, ok := parent[key].(map[string]interface{})
	if ok {
		updateNestedMap(parent, key, defaults)
		parent[key] = existing
	}
}

// parseBootstrapHostAndPort extracts host and REST API port from bootstrap API URL
func parseBootstrapHostAndPort(bootstrapAPIURL string) (host string, restAPIPort int, err error) {
	u, err := url.Parse(bootstrapAPIURL)
	if err != nil {
		return "", 0, err
	}

	host = u.Hostname()
	if host == "" {
		return "", 0, fmt.Errorf("no host in URL: %s", bootstrapAPIURL)
	}

	portStr := u.Port()
	if portStr == "" {
		// Default port based on scheme
		if u.Scheme == "http" {
			portStr = "9094"
		} else if u.Scheme == "https" {
			portStr = "443"
		} else {
			return "", 0, fmt.Errorf("unknown scheme: %s", u.Scheme)
		}
	}

	_, err = fmt.Sscanf(portStr, "%d", &restAPIPort)
	if err != nil {
		return "", 0, fmt.Errorf("invalid port: %s", portStr)
	}

	return host, restAPIPort, nil
}

// parseClusterPorts extracts cluster port and REST API port from ClusterAPIURL
func parseClusterPorts(clusterAPIURL string) (clusterPort, restAPIPort int, err error) {
	u, err := url.Parse(clusterAPIURL)
	if err != nil {
		return 0, 0, err
	}

	portStr := u.Port()
	if portStr == "" {
		// Default port based on scheme
		if u.Scheme == "http" {
			portStr = "9094"
		} else if u.Scheme == "https" {
			portStr = "443"
		} else {
			return 0, 0, fmt.Errorf("unknown scheme: %s", u.Scheme)
		}
	}

	_, err = fmt.Sscanf(portStr, "%d", &restAPIPort)
	if err != nil {
		return 0, 0, fmt.Errorf("invalid port: %s", portStr)
	}

	// clusterPort is used as the base port for calculations
	// The actual cluster LibP2P listen port is calculated as clusterPort + 4
	clusterPort = restAPIPort

	return clusterPort, restAPIPort, nil
}

// parseIPFSPort extracts IPFS API port from APIURL
func parseIPFSPort(apiURL string) (int, error) {
	if apiURL == "" {
		return 5001, nil // Default
	}

	u, err := url.Parse(apiURL)
	if err != nil {
		return 0, err
	}

	portStr := u.Port()
	if portStr == "" {
		if u.Scheme == "http" {
			return 5001, nil // Default HTTP port
		}
		return 0, fmt.Errorf("unknown scheme: %s", u.Scheme)
	}

	var port int
	_, err = fmt.Sscanf(portStr, "%d", &port)
	if err != nil {
		return 0, fmt.Errorf("invalid port: %s", portStr)
	}

	return port, nil
}

// getBootstrapPeerID queries the bootstrap cluster API to get the peer ID
func getBootstrapPeerID(apiURL string) (string, error) {
	// Simple HTTP client to query /peers endpoint
	client := &standardHTTPClient{}
	resp, err := client.Get(fmt.Sprintf("%s/peers", apiURL))
	if err != nil {
		return "", err
	}

	// The /peers endpoint returns NDJSON (newline-delimited JSON)
	// We need to read the first peer object to get the bootstrap peer ID
	dec := json.NewDecoder(bytes.NewReader(resp))
	var firstPeer struct {
		ID string `json:"id"`
	}
	if err := dec.Decode(&firstPeer); err != nil {
		return "", fmt.Errorf("failed to decode first peer: %w", err)
	}

	return firstPeer.ID, nil
}

// loadOrGenerateClusterSecret loads cluster secret or generates a new one
func loadOrGenerateClusterSecret(path string) (string, error) {
	// Try to load existing secret
	if data, err := os.ReadFile(path); err == nil {
		return strings.TrimSpace(string(data)), nil
	}

	// Generate new secret (32 bytes hex = 64 hex chars)
	secret := generateRandomSecret(64)

	// Save secret
	if err := os.MkdirAll(filepath.Dir(path), 0755); err != nil {
		return "", err
	}
	if err := os.WriteFile(path, []byte(secret), 0600); err != nil {
		return "", err
	}

	return secret, nil
}

// generateRandomSecret generates a random hex string
func generateRandomSecret(length int) string {
	bytes := make([]byte, length/2)
	if _, err := rand.Read(bytes); err != nil {
		// Fallback to simple generation if crypto/rand fails
		for i := range bytes {
			bytes[i] = byte(os.Getpid() + i)
		}
	}
	return hex.EncodeToString(bytes)
}

// standardHTTPClient implements HTTP client using net/http
type standardHTTPClient struct{}

func (c *standardHTTPClient) Get(url string) ([]byte, error) {
	resp, err := http.Get(url)
	if err != nil {
		return nil, err
	}
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		return nil, fmt.Errorf("HTTP %d: %s", resp.StatusCode, resp.Status)
	}

	data, err := io.ReadAll(resp.Body)
	if err != nil {
		return nil, err
	}

	return data, nil
}

// extractIPFromMultiaddrForCluster extracts IP address from a LibP2P multiaddr string
// Used for inferring bootstrap cluster API URL
func extractIPFromMultiaddrForCluster(multiaddrStr string) string {
	// Parse multiaddr
	ma, err := multiaddr.NewMultiaddr(multiaddrStr)
	if err != nil {
		return ""
	}

	// Try to extract IPv4 address
	if ipv4, err := ma.ValueForProtocol(multiaddr.P_IP4); err == nil && ipv4 != "" {
		return ipv4
	}

	// Try to extract IPv6 address
	if ipv6, err := ma.ValueForProtocol(multiaddr.P_IP6); err == nil && ipv6 != "" {
		return ipv6
	}

	return ""
}

// FixIPFSConfigAddresses fixes localhost addresses in IPFS config to use 127.0.0.1
// This is necessary because IPFS doesn't accept "localhost" as a valid IP address in multiaddrs
// This function always ensures the config is correct, regardless of current state
func (cm *ClusterConfigManager) FixIPFSConfigAddresses() error {
	if cm.cfg.Database.IPFS.APIURL == "" {
		return nil // IPFS not configured
	}

	// Determine IPFS repo path from config
	dataDir := cm.cfg.Node.DataDir
	if strings.HasPrefix(dataDir, "~") {
		home, err := os.UserHomeDir()
		if err != nil {
			return fmt.Errorf("failed to determine home directory: %w", err)
		}
		dataDir = filepath.Join(home, dataDir[1:])
	}

	// Try to find IPFS repo path
	// Check common locations: dataDir/ipfs/repo, or dataDir/bootstrap/ipfs/repo, etc.
	possiblePaths := []string{
		filepath.Join(dataDir, "ipfs", "repo"),
		filepath.Join(dataDir, "bootstrap", "ipfs", "repo"),
		filepath.Join(dataDir, "node2", "ipfs", "repo"),
		filepath.Join(dataDir, "node3", "ipfs", "repo"),
		filepath.Join(filepath.Dir(dataDir), "bootstrap", "ipfs", "repo"),
		filepath.Join(filepath.Dir(dataDir), "node2", "ipfs", "repo"),
		filepath.Join(filepath.Dir(dataDir), "node3", "ipfs", "repo"),
	}

	var ipfsRepoPath string
	for _, path := range possiblePaths {
		if _, err := os.Stat(filepath.Join(path, "config")); err == nil {
			ipfsRepoPath = path
			break
		}
	}

	if ipfsRepoPath == "" {
		cm.logger.Debug("IPFS repo not found, skipping config fix")
		return nil // Not an error if repo doesn't exist yet
	}

	// Parse IPFS API port from config
	ipfsPort, err := parseIPFSPort(cm.cfg.Database.IPFS.APIURL)
	if err != nil {
		return fmt.Errorf("failed to parse IPFS API URL: %w", err)
	}

	// Determine gateway port (typically API port + 3079, or 8080 for bootstrap, 8081 for node2, etc.)
	gatewayPort := 8080
	if strings.Contains(dataDir, "node2") {
		gatewayPort = 8081
	} else if strings.Contains(dataDir, "node3") {
		gatewayPort = 8082
	} else if ipfsPort == 5002 {
		gatewayPort = 8081
	} else if ipfsPort == 5003 {
		gatewayPort = 8082
	}

	// Always ensure API address is correct (don't just check, always set it)
	correctAPIAddr := fmt.Sprintf(`["/ip4/0.0.0.0/tcp/%d"]`, ipfsPort)
	cm.logger.Info("Ensuring IPFS API address is correct",
		zap.String("repo", ipfsRepoPath),
		zap.Int("port", ipfsPort),
		zap.String("correct_address", correctAPIAddr))

	fixCmd := exec.Command("ipfs", "config", "--json", "Addresses.API", correctAPIAddr)
	fixCmd.Env = append(os.Environ(), "IPFS_PATH="+ipfsRepoPath)
	if err := fixCmd.Run(); err != nil {
		cm.logger.Warn("Failed to fix IPFS API address", zap.Error(err))
		return fmt.Errorf("failed to set IPFS API address: %w", err)
	}

	// Always ensure Gateway address is correct
	correctGatewayAddr := fmt.Sprintf(`["/ip4/0.0.0.0/tcp/%d"]`, gatewayPort)
	cm.logger.Info("Ensuring IPFS Gateway address is correct",
		zap.String("repo", ipfsRepoPath),
		zap.Int("port", gatewayPort),
		zap.String("correct_address", correctGatewayAddr))

	fixCmd = exec.Command("ipfs", "config", "--json", "Addresses.Gateway", correctGatewayAddr)
	fixCmd.Env = append(os.Environ(), "IPFS_PATH="+ipfsRepoPath)
	if err := fixCmd.Run(); err != nil {
		cm.logger.Warn("Failed to fix IPFS Gateway address", zap.Error(err))
		return fmt.Errorf("failed to set IPFS Gateway address: %w", err)
	}

	// Check if IPFS daemon is running - if so, it may need to be restarted for changes to take effect
	// We can't restart it from here (it's managed by Makefile/systemd), but we can warn
	if cm.isIPFSRunning(ipfsPort) {
		cm.logger.Warn("IPFS daemon appears to be running - it may need to be restarted for config changes to take effect",
			zap.Int("port", ipfsPort),
			zap.String("repo", ipfsRepoPath))
	}

	return nil
}

// isIPFSRunning checks if IPFS daemon is running by attempting to connect to the API
func (cm *ClusterConfigManager) isIPFSRunning(port int) bool {
	client := &http.Client{
		Timeout: 1 * time.Second,
	}
	resp, err := client.Get(fmt.Sprintf("http://127.0.0.1:%d/api/v0/id", port))
	if err != nil {
		return false
	}
	resp.Body.Close()
	return resp.StatusCode == 200
}