package main import ( "html/template" "net/http" "os" "sync" "github.com/gin-contrib/multitemplate" "github.com/gin-gonic/gin" ) var mu sync.Mutex var indexPage = template.Must(template.New("index.html").Parse(`
System Version: {{.Version}}
{{.Message}}
`)) func runHttp() { gin.SetMode(gin.ReleaseMode) g := gin.Default() r := multitemplate.NewRenderer() r.Add("index.html", indexPage) r.Add("reboot.html", rebootPage) g.HTMLRender = r g.GET("/", renderIndex) g.POST("/address", handleAddress) g.POST("/system", handleSystem) g.POST("/application", handleApplication) g.GET("/reboot", handleReboot) g.GET("/restore", handleRestore) g.Run(":8988") } func handleAddress(c *gin.Context) { mu.Lock() defer mu.Unlock() addr := c.PostForm("address") addr2 := c.PostForm("address2") if err := setIpAddress(addr, addr2); err != nil { c.String(http.StatusBadRequest, "address: setIpAddress: %s", err.Error()) return } c.HTML(http.StatusOK, "reboot.html", gin.H{"Message": "IP address is updated."}) } func handleSystem(c *gin.Context) { mu.Lock() defer mu.Unlock() if err := saveFormFile(c, "image", "/tmp/system.zip"); err != nil { c.String(http.StatusBadRequest, "system: saveFormFile: %s", err.Error()) return } if err := updateSystem(); err != nil { c.String(http.StatusBadRequest, "system: updateSystem: %s", err.Error()) return } c.HTML(http.StatusOK, "reboot.html", gin.H{"Message": "System is updated."}) } func handleApplication(c *gin.Context) { mu.Lock() defer mu.Unlock() if err := saveFormFile(c, "image", "/tmp/application.zip"); err != nil { c.String(http.StatusBadRequest, "application: saveFormFile: %s", err.Error()) return } if err := updateApp(); err != nil { c.String(http.StatusBadRequest, "application: updateApplication: %s", err.Error()) return } c.HTML(http.StatusOK, "reboot.html", gin.H{"Message": "Application is updated."}) } func handleReboot(c *gin.Context) { reboot() c.String(http.StatusOK, "OK") } func handleRestore(c *gin.Context) { mu.Lock() defer mu.Unlock() restore() c.HTML(http.StatusOK, "reboot.html", gin.H{"Message": "Restoration is done."}) } func renderIndex(c *gin.Context) { device := "System" if len(os.Args) > 1 { device = os.Args[1] } c.HTML(http.StatusOK, "index.html", gin.H{ "Device": device, "IP": getIpAddress(), "IP2": getIpAddress2(), "Version": getVersion(), }) } func saveFormFile(c *gin.Context, name string, dst string) error { file, err := c.FormFile(name) if err != nil { return err } return c.SaveUploadedFile(file, dst) }