nws/cmd/exit/exit.go

package main

import (
	"github.com/asmogo/nws/config"
	"github.com/asmogo/nws/exit"
	"github.com/spf13/cobra"
	"log/slog"
)

var httpsPort int32
var httpTarget string

func main() {
	rootCmd := &cobra.Command{Use: "exit", Run: startExitNode}
	rootCmd.Flags().Int32VarP(&httpsPort, "port", "p", 0, "port for the https reverse proxy")
	rootCmd.Flags().StringVarP(&httpTarget, "target", "t", "", "target for the https reverse proxy (your local service)")
	err := rootCmd.Execute()
	if err != nil {
		panic(err)
	}
}
func startExitNode(cmd *cobra.Command, args []string) {

	// load the configuration
	// from the environment
	cfg, err := config.LoadConfig[config.ExitConfig]()
	if err != nil {
		panic(err)
	}

	// create a new gw server
	// and start it
	ctx := cmd.Context()
	exitNode := exit.NewExit(ctx, cfg)
	if httpsPort != 0 {
		slog.Info("starting exit node with https reverse proxy", "port", httpsPort)
		go func() {
			err = exitNode.StartReverseProxy(httpTarget, httpsPort)
			if err != nil {
				panic(err)
			}
		}()

	}
	exitNode.ListenAndServe(ctx)
}