diff options
Diffstat (limited to 'tvix/nar-bridge/cmd')
-rw-r--r-- | tvix/nar-bridge/cmd/nar_bridge/import.go | 57 | ||||
-rw-r--r-- | tvix/nar-bridge/cmd/nar_bridge/main.go | 30 | ||||
-rw-r--r-- | tvix/nar-bridge/cmd/nar_bridge/serve.go | 60 |
3 files changed, 147 insertions, 0 deletions
diff --git a/tvix/nar-bridge/cmd/nar_bridge/import.go b/tvix/nar-bridge/cmd/nar_bridge/import.go new file mode 100644 index 000000000000..51b99c93a04b --- /dev/null +++ b/tvix/nar-bridge/cmd/nar_bridge/import.go @@ -0,0 +1,57 @@ +package main + +import ( + "context" + "fmt" + "io" + "os" + "os/signal" + + storev1pb "code.tvl.fyi/tvix/store/protos" + + "code.tvl.fyi/tvix/nar-bridge/pkg/reader" + log "github.com/sirupsen/logrus" +) + +type ImportCmd struct { + NarPath string `name:"nar-path" help:"A path to a NAR file"` +} + +// `help:"Read a NAR file and display some information"` + +func (cmd *ImportCmd) Run() error { + retcode := 0 + + defer func() { os.Exit(retcode) }() + + c := make(chan os.Signal, 1) + signal.Notify(c, os.Interrupt) + + go func() { + for range c { + log.Info("Received Signal, shutting down…") + os.Exit(1) + } + }() + + log.Infof("Reading %v...", cmd.NarPath) + + f, _ := os.Open(cmd.NarPath) + + r := reader.New(f) + + actualPathInfo, _ := r.Import( + context.Background(), + func(fileReader io.Reader) error { + return nil + }, + func(directory *storev1pb.Directory) error { + return nil + }, + ) + + fmt.Printf("Node: %+v\n", actualPathInfo.Node) + fmt.Printf("References: %+v\n", actualPathInfo.References) + fmt.Printf("Narinfo: %+v\n", actualPathInfo.Narinfo) + return nil +} diff --git a/tvix/nar-bridge/cmd/nar_bridge/main.go b/tvix/nar-bridge/cmd/nar_bridge/main.go new file mode 100644 index 000000000000..78ed3e272d69 --- /dev/null +++ b/tvix/nar-bridge/cmd/nar_bridge/main.go @@ -0,0 +1,30 @@ +package main + +import ( + "os" + + "github.com/alecthomas/kong" +) + +//nolint:gochecknoglobals +var cli struct { + // TODO: make log level configurable + Import ImportCmd `kong:"cmd,name='import',help='Import a local NAR file into a tvix-store'"` + Serve ServeCmd `kong:"cmd,name='serve',help='Expose a tvix-store RPC interface as NAR/NARInfo'"` +} + +func main() { + parser, err := kong.New(&cli) + if err != nil { + panic(err) + } + + ctx, err := parser.Parse(os.Args[1:]) + if err != nil { + panic(err) + } + // Call the Run() method of the selected parsed command. + err = ctx.Run() + + ctx.FatalIfErrorf(err) +} diff --git a/tvix/nar-bridge/cmd/nar_bridge/serve.go b/tvix/nar-bridge/cmd/nar_bridge/serve.go new file mode 100644 index 000000000000..6566257176d8 --- /dev/null +++ b/tvix/nar-bridge/cmd/nar_bridge/serve.go @@ -0,0 +1,60 @@ +package main + +import ( + "os" + "os/signal" + + "google.golang.org/grpc" + "google.golang.org/grpc/credentials/insecure" + + "code.tvl.fyi/tvix/nar-bridge/pkg/server" + storev1pb "code.tvl.fyi/tvix/store/protos" + log "github.com/sirupsen/logrus" +) + +type ServeCmd struct { + ListenAddr string `name:"listen-addr" help:"The address this service listens on" type:"string" default:"[::]:9000"` //nolint:lll + EnableAccessLog bool `name:"access-log" help:"Enable access logging" type:"bool" default:"true" negatable:""` //nolint:lll + StoreAddr string `name:"store-addr" help:"The address to the tvix-store RPC interface this will connect to"` +} + +// `help:"Expose a tvix-store RPC interface as NAR/NARInfo"` +func (cmd *ServeCmd) Run() error { + retcode := 0 + + defer func() { os.Exit(retcode) }() + + c := make(chan os.Signal, 1) + signal.Notify(c, os.Interrupt) + + go func() { + for range c { + log.Info("Received Signal, shutting down…") + //s.Close() + os.Exit(1) + } + }() + + // connect to tvix-store + log.Debugf("Dialing to %v", cmd.StoreAddr) + conn, err := grpc.Dial(cmd.StoreAddr, grpc.WithTransportCredentials(insecure.NewCredentials())) + if err != nil { + log.Fatalf("did not connect: %v", err) + } + defer conn.Close() + + log.Printf("Starting nar-bridge at %v", cmd.ListenAddr) + s := server.New( + storev1pb.NewDirectoryServiceClient(conn), + storev1pb.NewBlobServiceClient(conn), + storev1pb.NewPathInfoServiceClient(conn), + cmd.EnableAccessLog, + 30, + ) + + err = s.ListenAndServe(cmd.ListenAddr) + if err != nil { + log.Error("Server failed: %w", err) + } + return nil +} |