main.go

package main

import (
	"fmt"
	"os"
	"os/signal"
	"syscall"

	"github.com/AminN77/morty/cmd"
	"github.com/AminN77/morty/internal/config"
	"github.com/AminN77/morty/internal/logger"
	"github.com/AminN77/morty/internal/raft"
	"github.com/AminN77/morty/internal/store"
	etcdraft "go.etcd.io/etcd/raft/v3"
)

func main() {
	cfg, err := config.Load(os.Args[1:])
	if err != nil {
		fmt.Fprintf(os.Stderr, "Config error: %v\n", err)
		fmt.Fprintln(os.Stderr, "Usage: morty <node-id> <port>")
		fmt.Fprintln(os.Stderr, "Or: morty --node-id 1 --port 6000 --peers 1=localhost:6000,2=localhost:6001,3=localhost:6002")
		os.Exit(1)
	}

	if err := logger.Init(cfg.NodeID, cfg.LogTarget, cfg.LogDir); err != nil {
		fmt.Fprintf(os.Stderr, "Failed to init logger: %v\n", err)
		os.Exit(1)
	}
	defer logger.Close()

	// Redirect etcd/raft internal logging to our file logger
	etcdraft.SetLogger(&logger.RaftLogger{})

	// Initialize store and raft node
	st := &store.Store{}
	peers := cfg.PeerIDs()
	peerAddrs := cfg.PeerAddrs

	raftNode, err := raft.NewNode(cfg.NodeID, peers, st, peerAddrs)
	if err != nil {
		fmt.Fprintf(os.Stderr, "Failed to create node: %v\n", err)
		os.Exit(1)
	}

	// Start HTTP server
	httpServer := raft.NewServer(raftNode, cfg.Port)
	if err := httpServer.Start(); err != nil {
		fmt.Fprintf(os.Stderr, "Failed to start server: %v\n", err)
		os.Exit(1)
	}
	defer httpServer.Stop()

	// Start Raft engine
	go raftNode.Run()
	defer raftNode.Stop()

	if cfg.Interactive {
		fmt.Printf("Node %d ready. Commands: set, get, delete, status, exit\n", cfg.NodeID)
		cmd.Loop(raftNode, st, httpServer)
		return
	}

	sigCh := make(chan os.Signal, 1)
	signal.Notify(sigCh, os.Interrupt, syscall.SIGTERM)
	<-sigCh
}