78 lines
1.6 KiB
Go
78 lines
1.6 KiB
Go
/*
|
|
Copyright © 2025 Ronmi Ren <ronmi.ren@gmail.com
|
|
*/
|
|
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/url"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"git.ronmi.tw/ronmi/forgejo-pages/lib"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// serveCmd represents the serve command
|
|
var serveCmd = &cobra.Command{
|
|
Use: "serve",
|
|
Short: "Start the static page server.",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
// check flags
|
|
bind, _ := cmd.Flags().GetString("bind")
|
|
server, _ := cmd.Flags().GetString("server")
|
|
token, _ := cmd.Flags().GetString("token")
|
|
branch, _ := cmd.Flags().GetString("branch")
|
|
if bind == "" || server == "" || token == "" || branch == "" {
|
|
fmt.Println("bind, server token and branch are required")
|
|
return
|
|
}
|
|
serverUrl, err := url.Parse(server)
|
|
if err != nil {
|
|
fmt.Println("invalid server url: ", err)
|
|
return
|
|
}
|
|
f := &lib.Forgejo{
|
|
Server: *serverUrl,
|
|
Token: token,
|
|
Branch: branch,
|
|
}
|
|
|
|
s, err := lib.Run(bind, f)
|
|
if err != nil {
|
|
fmt.Println("cannot create server: ", err)
|
|
return
|
|
}
|
|
|
|
ctx, stop := signal.NotifyContext(
|
|
context.TODO(),
|
|
os.Interrupt,
|
|
syscall.SIGTERM,
|
|
os.Kill,
|
|
)
|
|
defer stop()
|
|
|
|
fmt.Println("starting server")
|
|
go func() {
|
|
<-ctx.Done()
|
|
stop()
|
|
ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
|
|
defer cancel()
|
|
s.Shutdown(ctx)
|
|
}()
|
|
s.ListenAndServe()
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(serveCmd)
|
|
|
|
f := serveCmd.Flags()
|
|
f.StringP("bind", "a", ":8080", "bind address")
|
|
f.StringP("server", "s", "", "Forgejo server address")
|
|
f.StringP("token", "k", "", "Forgejo api token")
|
|
f.StringP("branch", "b", "static-pages", "branch to use")
|
|
}
|