main.go

package main

import (
	"fmt"
	"net/http"
	"os"

	"github.com/Native-Planet/perigee/cli"
	"github.com/Native-Planet/perigee/handlers"
	"github.com/Native-Planet/perigee/logger"
	"github.com/Native-Planet/perigee/pontifex"
	"github.com/Native-Planet/perigee/roller"

	"github.com/spf13/cobra"
	"go.uber.org/zap"
)

var port string

var rootCmd = &cobra.Command{
	Use:   "perigee",
	Short: "Perigee: Azimuth light client and macro server",
}

var serveCmd = &cobra.Command{
	Use:   "serve",
	Short: "Start the HTTP server",
	RunE: func(cmd *cobra.Command, args []string) error {
		return runServer()
	},
}

func init() {
	// add flags to commands
	serveCmd.Flags().StringVarP(&port, "port", "p", "8080", "Port to run server on")
	cli.GetPointCmd.Flags().String("point", "", "Point for retrieval")

	cli.GetPendingCmd.Flags().String("point", "", "Pending transactions for a point")
	cli.GetPendingCmd.Flags().String("address", "", "Pending transactions for a address")

	cli.GetWalletCmd.Flags().String("point", "", "Azimuth point for wallet generation")
	cli.GetWalletCmd.Flags().String("passphrase", "", "Passphrase for wallet")
	cli.GetWalletCmd.Flags().String("master-ticket", "", "Master ticket for wallet generation")
	cli.GetWalletCmd.Flags().String("output-dir", "", "Directory to save wallet")
	cli.GetWalletCmd.Flags().Int("life", 0, "Custom life value for wallet generation")

	cli.GetKeyfileCmd.Flags().String("master-ticket", "", "Master ticket for keyfile generation")
	cli.GetKeyfileCmd.Flags().String("passphrase", "", "Passphrase for wallet")
	cli.GetKeyfileCmd.Flags().String("point", "", "Azimuth point for keyfile generation")
	cli.GetKeyfileCmd.Flags().String("output-dir", "", "Directory to save keyfile")
	cli.GetKeyfileCmd.Flags().Int("life", 0, "Custom life value for keyfile generation (optional)")

	cli.GetCodeCmd.Flags().String("master-ticket", "", "Master ticket for code generation")
	cli.GetCodeCmd.Flags().String("passphrase", "", "Passphrase for wallet")
	cli.GetCodeCmd.Flags().String("point", "", "Azimuth point for code generation")
	cli.GetCodeCmd.Flags().String("output-dir", "", "Directory to save code")
	cli.GetCodeCmd.Flags().Int("step", 0, "Custom step value for code generation (optional)")
	cli.GetCodeCmd.Flags().Int("life", 0, "Custom life value for code generation (optional)")

	cli.ModBreachCmd.Flags().String("point", "", "Point for breach")
	cli.ModBreachCmd.Flags().String("master-ticket", "", "Master ticket for wallet generation")
	cli.ModBreachCmd.Flags().String("passphrase", "", "Passphrase for wallet")
	cli.ModBreachCmd.Flags().String("seed", "", "64 character hex seed for network keys (only necessary if using eth private key to breach)")
	cli.ModBreachCmd.Flags().Duration("wait", 0, "Wait until breach processes (forever if no value, or specific time like '70m' or '1h20m')")

	cli.ModEscapeCmd.Flags().String("point", "", "Escaping point")
	cli.ModEscapeCmd.Flags().String("master-ticket", "", "Master ticket for wallet generation")
	cli.ModEscapeCmd.Flags().String("private-key", "", "Ethereum address private key")
	cli.ModEscapeCmd.Flags().String("passphrase", "", "Passphrase for wallet")
	cli.ModEscapeCmd.Flags().String("sponsor", "", "New sponsor")
	cli.ModEscapeCmd.Flags().Duration("wait", 0, "Wait until breach processes (forever if no value, or specific time like '70m' or '1h20m')")

	cli.ModCancelEscapeCmd.Flags().String("point", "", "Escaping point")
	cli.ModCancelEscapeCmd.Flags().String("master-ticket", "", "Master ticket for wallet generation")
	cli.ModCancelEscapeCmd.Flags().String("private-key", "", "Ethereum address private key")
	cli.ModCancelEscapeCmd.Flags().String("passphrase", "", "Passphrase for wallet")
	cli.ModCancelEscapeCmd.Flags().String("sponsor", "", "New sponsor")

	cli.ModAdoptCmd.Flags().String("point", "", "Point for breach")
	cli.ModAdoptCmd.Flags().String("master-ticket", "", "Master ticket for wallet generation")
	cli.ModAdoptCmd.Flags().String("private-key", "", "Ethereum address private key")
	cli.ModAdoptCmd.Flags().String("passphrase", "", "Passphrase for wallet")
	cli.ModAdoptCmd.Flags().String("adoptee", "", "Newly adopted point")
	cli.ModAdoptCmd.Flags().Duration("wait", 0, "Wait until breach processes (forever if no value, or specific time like '70m' or '1h20m')")

	rootCmd.AddCommand(serveCmd)
	rootCmd.AddCommand(cli.GetWalletCmd)
	rootCmd.AddCommand(cli.GetKeyfileCmd)
	rootCmd.AddCommand(cli.GetPointCmd)
	rootCmd.AddCommand(cli.GetPendingCmd)
	rootCmd.AddCommand(cli.GetCodeCmd)
	rootCmd.AddCommand(cli.ModBreachCmd)
	rootCmd.AddCommand(cli.ModEscapeCmd)
	rootCmd.AddCommand(cli.ModCancelEscapeCmd)
	rootCmd.AddCommand(cli.ModAdoptCmd)
}

func runServer() error {
	web, err := pontifex.NewHandler()
	if err != nil {
		return err
	}

	http.HandleFunc("/v1/get/wallet", handlers.GetWallet())
	http.HandleFunc("/v1/get/point", handlers.GetPoint())
	http.HandleFunc("/v1/get/pending", handlers.GetPending())
	http.HandleFunc("/v1/get/code", handlers.GetCode())
	http.HandleFunc("/v1/mod/breach", handlers.ModBreach())
	http.HandleFunc("/v1/mod/escape", handlers.ModEscape())
	http.HandleFunc("/v1/mod/cancel-escape", handlers.ModCancelEscape())
	http.HandleFunc("/v1/mod/adopt", handlers.ModAdopt())

	http.HandleFunc("/healthz", handlers.ReadinessProbe)
	http.HandleFunc("/readyz", handlers.LivenessProbe)

	http.Handle("/", web)

	zap.L().Info(fmt.Sprintf("Using roller endpoint %s", roller.RollerURL))
	addr := fmt.Sprintf(":%s", port)
	zap.L().Info(fmt.Sprintf("Starting server on %s...", addr))
	return http.ListenAndServe(addr, nil)
}

func main() {
	if err := logger.Init(); err != nil {
		panic(fmt.Sprintf("Failed to initialize logger: %v", err))
	}
	defer logger.Sync()

	if err := rootCmd.Execute(); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
}