start/stop: allow temporarily removing a route

Author: derrickstolee

cmd/git-bundle-server/start.go

@@ -0,0 +1,32 @@
+package main
+
+import (
+	"errors"
+
+	"github.com/github/git-bundle-server/internal/core"
+)
+
+type Start struct{}
+
+func (Start) subcommand() string {
+	return "start"
+}
+
+func (Start) run(args []string) error {
+	if len(args) < 1 {
+		return errors.New("usage: git-bundle-server start <route>")
+	}
+
+	route := args[0]
+
+	// CreateRepository re-reigsters the route.
+	_, err := core.CreateRepository(route)
+	if err != nil {
+		return err
+	}
+
+	// Make sure we have the global schedule running.
+	SetCronSchedule()
+
+	return nil
+}

cmd/git-bundle-server/stop.go

@@ -0,0 +1,23 @@
+package main
+
+import (
+	"errors"
+
+	"github.com/github/git-bundle-server/internal/core"
+)
+
+type Stop struct{}
+
+func (Stop) subcommand() string {
+	return "stop"
+}
+
+func (Stop) run(args []string) error {
+	if len(args) < 1 {
+		return errors.New("usage: git-bundle-server stop <route>")
+	}
+
+	route := args[0]
+
+	return core.RemoveRoute(route)
+}

cmd/git-bundle-server/subcommand.go

@@ -8,6 +8,8 @@ type Subcommand interface {
 func all() []Subcommand {
 	return []Subcommand{
 		Init{},
+		Start{},
+		Stop{},
 		Update{},
 		UpdateAll{},
 	}

internal/core/repo.go

@@ -49,6 +49,22 @@ func CreateRepository(route string) (*Repository, error) {
 	return &repo, nil
 }
 
+func RemoveRoute(route string) error {
+	repos, err := GetRepositories()
+	if err != nil {
+		return fmt.Errorf("failed to parse routes file")
+	}
+
+	_, contains := repos[route]
+	if !contains {
+		return fmt.Errorf("route '%s' is not registered", route)
+	}
+
+	delete(repos, route)
+
+	return WriteRouteFile(repos)
+}
+
 func WriteRouteFile(repos map[string]Repository) error {
 	dir := bundleroot()
 	routefile := dir + "/routes"
@@ -59,7 +75,7 @@ func WriteRouteFile(repos map[string]Repository) error {
 		contents = contents + routes + "\n"
 	}
 
-	return os.WriteFile(routefile, []byte(contents), 0600)
+	return os.WriteFile(routefile, []byte(contents), 0o600)
 }
 
 func GetRepositories() (map[string]Repository, error) {
@@ -68,7 +84,7 @@ func GetRepositories() (map[string]Repository, error) {
 	dir := bundleroot()
 	routefile := dir + "/routes"
 
-	file, err := os.OpenFile(routefile, os.O_RDONLY|os.O_CREATE, 0600)
+	file, err := os.OpenFile(routefile, os.O_RDONLY|os.O_CREATE, 0o600)
 	if err != nil {
 		// Assume that the file doesn't exist?
 		return repos, nil