network/pkg/client/interface.go

package client

import (
	"context"
	"fmt"
	"os"
	"strconv"
	"time"
)

// NetworkClient provides the main interface for applications to interact with the network
type NetworkClient interface {
	// Database operations (namespaced per app)
	Database() DatabaseClient

	// Key-value storage (namespaced per app)
	Storage() StorageClient

	// Pub/Sub messaging
	PubSub() PubSubClient

	// Network information
	Network() NetworkInfo

	// Lifecycle
	Connect() error
	Disconnect() error
	Health() (*HealthStatus, error)

	// Config access (snapshot copy)
	Config() *ClientConfig
}

// DatabaseClient provides database operations for applications
type DatabaseClient interface {
	Query(ctx context.Context, sql string, args ...interface{}) (*QueryResult, error)
	Transaction(ctx context.Context, queries []string) error
	CreateTable(ctx context.Context, schema string) error
	DropTable(ctx context.Context, tableName string) error
	GetSchema(ctx context.Context) (*SchemaInfo, error)
}

// StorageClient provides key-value storage operations
type StorageClient interface {
	Get(ctx context.Context, key string) ([]byte, error)
	Put(ctx context.Context, key string, value []byte) error
	Delete(ctx context.Context, key string) error
	List(ctx context.Context, prefix string, limit int) ([]string, error)
	Exists(ctx context.Context, key string) (bool, error)
}

// PubSubClient provides publish/subscribe messaging
type PubSubClient interface {
	Subscribe(ctx context.Context, topic string, handler MessageHandler) error
	Publish(ctx context.Context, topic string, data []byte) error
	Unsubscribe(ctx context.Context, topic string) error
	ListTopics(ctx context.Context) ([]string, error)
}

// NetworkInfo provides network status and peer information
type NetworkInfo interface {
	GetPeers(ctx context.Context) ([]PeerInfo, error)
	GetStatus(ctx context.Context) (*NetworkStatus, error)
	ConnectToPeer(ctx context.Context, peerAddr string) error
	DisconnectFromPeer(ctx context.Context, peerID string) error
}

// MessageHandler is called when a pub/sub message is received
type MessageHandler func(topic string, data []byte) error

// Data structures

// QueryResult represents the result of a database query
type QueryResult struct {
	Columns []string        `json:"columns"`
	Rows    [][]interface{} `json:"rows"`
	Count   int64           `json:"count"`
}

// SchemaInfo contains database schema information
type SchemaInfo struct {
	Tables []TableInfo `json:"tables"`
}

// TableInfo contains information about a database table
type TableInfo struct {
	Name    string       `json:"name"`
	Columns []ColumnInfo `json:"columns"`
}

// ColumnInfo contains information about a table column
type ColumnInfo struct {
	Name     string `json:"name"`
	Type     string `json:"type"`
	Nullable bool   `json:"nullable"`
	Default  string `json:"default"`
}

// PeerInfo contains information about a network peer
type PeerInfo struct {
	ID        string    `json:"id"`
	Addresses []string  `json:"addresses"`
	Connected bool      `json:"connected"`
	LastSeen  time.Time `json:"last_seen"`
}

// NetworkStatus contains overall network status
type NetworkStatus struct {
	NodeID       string        `json:"node_id"`
	Connected    bool          `json:"connected"`
	PeerCount    int           `json:"peer_count"`
	DatabaseSize int64         `json:"database_size"`
	Uptime       time.Duration `json:"uptime"`
}

// HealthStatus contains health check information
type HealthStatus struct {
	Status       string            `json:"status"` // "healthy", "degraded", "unhealthy"
	Checks       map[string]string `json:"checks"`
	LastUpdated  time.Time         `json:"last_updated"`
	ResponseTime time.Duration     `json:"response_time"`
}

// ClientConfig represents configuration for network clients
type ClientConfig struct {
	AppName        string        `json:"app_name"`
	DatabaseName   string        `json:"database_name"`
	BootstrapPeers []string      `json:"bootstrap_peers"`
	DatabaseEndpoints []string   `json:"database_endpoints"`
	ConnectTimeout time.Duration `json:"connect_timeout"`
	RetryAttempts  int           `json:"retry_attempts"`
	RetryDelay     time.Duration `json:"retry_delay"`
	QuietMode      bool          `json:"quiet_mode"` // Suppress debug/info logs
}

// DefaultClientConfig returns a default client configuration
func DefaultClientConfig(appName string) *ClientConfig {
    // Base defaults
    peers := []string{}
    endpoints := DefaultDatabaseEndpoints()

    // Development local-only override via env
    if isTruthyEnv("NETWORK_DEV_LOCAL") {
        port := 5001
        if v := os.Getenv("RQLITE_PORT"); v != "" {
            if n, err := strconv.Atoi(v); err == nil && n > 0 { port = n }
        }
        endpoints = []string{fmt.Sprintf("http://127.0.0.1:%d", port)}
        if ma := os.Getenv("LOCAL_BOOTSTRAP_MULTIADDR"); ma != "" {
            peers = []string{ma}
        } else {
            // Leave peers empty if no local bootstrap multiaddr provided
            peers = []string{}
        }
    }

    return &ClientConfig{
        AppName:           appName,
        DatabaseName:      fmt.Sprintf("%s_db", appName),
        BootstrapPeers:    peers,
        DatabaseEndpoints: endpoints,
        ConnectTimeout:    time.Second * 30,
        RetryAttempts:     3,
        RetryDelay:        time.Second * 5,
    }
}

// isTruthyEnv returns true if the env var is set to a common truthy value.
func isTruthyEnv(key string) bool {
    v := os.Getenv(key)
    switch v {
    case "1", "true", "TRUE", "True", "yes", "YES", "on", "ON":
        return true
    default:
        return false
    }
}