webui.go

 1package commands
 2
 3import (
 4	"fmt"
 5	"github.com/MichaelMure/git-bug/graphql2"
 6	"github.com/MichaelMure/git-bug/webui"
 7	"github.com/gorilla/mux"
 8	"github.com/phayes/freeport"
 9	"github.com/skratchdot/open-golang/open"
10	"github.com/spf13/cobra"
11	"github.com/vektah/gqlgen/handler"
12	"log"
13	"net/http"
14)
15
16var port int
17
18func runWebUI(cmd *cobra.Command, args []string) error {
19	if port == 0 {
20		var err error
21		port, err = freeport.GetFreePort()
22		if err != nil {
23			return err
24		}
25	}
26
27	addr := fmt.Sprintf("127.0.0.1:%d", port)
28	webUiAddr := fmt.Sprintf("http://%s", addr)
29
30	fmt.Printf("Web UI available at %s\n", webUiAddr)
31
32	router := mux.NewRouter()
33
34	// Routes
35	router.Path("/playground").Handler(handler.Playground("git-bug", "/graphql"))
36	router.Path("/graphql").Handler(graphql2.NewHandler(repo))
37	router.PathPrefix("/").Handler(http.FileServer(webui.WebUIAssets))
38
39	open.Run(webUiAddr)
40
41	log.Fatal(http.ListenAndServe(addr, router))
42
43	return nil
44}
45
46var webUICmd = &cobra.Command{
47	Use:   "webui",
48	Short: "Launch the web UI",
49	RunE:  runWebUI,
50}
51
52func init() {
53	RootCmd.AddCommand(webUICmd)
54	webUICmd.Flags().IntVarP(&port, "port", "p", 0, "Port to listen to")
55}