1 files changed, 63 insertions, 5 deletions
diff --git a/cmd/mbf/main.go b/cmd/mbf/main.go
index 4f471d6a..dd5ee624 100644
--- a/cmd/mbf/main.go
+++ b/cmd/mbf/main.go
@@ -43,8 +43,9 @@ import (
 	"hakurei.app/message"
 	"hakurei.app/pkg"
 
-	"hakurei.app/cmd/mbf/internal/pkgserver"
-	"hakurei.app/cmd/mbf/internal/pkgserver/ui"
+	"hakurei.app/cmd/mbf/internal/ci"
+	"hakurei.app/cmd/mbf/internal/pkgsite"
+	"hakurei.app/cmd/mbf/internal/pkgsite/ui"
 )
 
 // builtin contains native and embedded [rosa.Artifact] registrations.
@@ -70,7 +71,7 @@ func main() {
 	log.SetPrefix("mbf: ")
 	msg := message.New(log.Default())
 
-	if os.Geteuid() == 0 {
+	if !poisonOpen && os.Geteuid() == 0 {
 		log.Fatal("this program must not run as root")
 	}
 
@@ -343,7 +344,7 @@ func main() {
 
 				var mux http.ServeMux
 				ui.Register(&mux)
-				if err = pkgserver.Register(ctx, &mux, r); err != nil {
+				if err = pkgsite.Register(ctx, &mux, r); err != nil {
 					return
 				}
 
@@ -504,7 +505,7 @@ func main() {
 	c.NewCommand(
 		"daemon",
 		"Service artifact IR with Rosa OS extensions",
-		func(args []string) error {
+		func([]string) error {
 			ul, err := net.ListenUnix("unix", &addr)
 			if err != nil {
 				return err
@@ -514,6 +515,63 @@ func main() {
 		},
 	)
 
+	_ci := c.New("ci", command.UsageInternal)
+	_ci.NewCommand(
+		"daemon",
+		"Service CI workload dispatched through the socket",
+		func([]string) error {
+			return cm.Do(func(cache *pkg.Cache) error {
+				return ci.Serve(ctx, msg, cache)
+			})
+		},
+	)
+	{
+		var flagOutput string
+		_ci.NewCommand(
+			"dist",
+			"Request distribution tarball for the specified source directory",
+			func(args []string) error {
+				if len(args) != 2 {
+					return errors.New("dist requires 2 arguments")
+				}
+
+				return ci.Path(cm.base).MakeDist(
+					ctx,
+					os.Stdout,
+					flagOutput,
+					args[0], args[1],
+				)
+			},
+		).Flag(
+			&flagOutput,
+			"o", command.StringFlag("."),
+			"Write the resulting distribution to the named directory",
+		)
+	}
+	{
+		var flagOutput string
+		_ci.NewCommand(
+			"race",
+			"Request distribution tarball (race) for the specified source directory",
+			func(args []string) error {
+				if len(args) != 2 {
+					return errors.New("race requires 2 arguments")
+				}
+
+				return ci.Path(cm.base).MakeDistRace(
+					ctx,
+					os.Stdout,
+					flagOutput,
+					args[0], args[1],
+				)
+			},
+		).Flag(
+			&flagOutput,
+			"o", command.StringFlag("."),
+			"Write the resulting distribution to the named directory",
+		)
+	}
+
 	c.NewCommand(
 		"keygen",
 		"Create keypair for local cache",