server.go

package main

import (
	"fmt"
	"net/http"
	"os"

	"github.com/chengjoey/web-terminal/views"
	"github.com/gin-gonic/gin"
)

var (
	EnvGOPATH     = os.Getenv("GOPATH")
	TemplateFiles = fmt.Sprintf("%s/src/github.com/chengjoey/web-terminal/templates/*.html", EnvGOPATH)
	StaticFiles   = fmt.Sprintf("%s/src/github.com/chengjoey/web-terminal/templates/static", EnvGOPATH)
)

//&#23545;&#20135;&#29983;&#30340;&#20219;&#20309;error&#36827;&#34892;&#22788;&#29702;
func JSONAppErrorReporter() gin.HandlerFunc {
	return jsonAppErrorReporterT(gin.ErrorTypeAny)
}

func jsonAppErrorReporterT(errType gin.ErrorType) gin.HandlerFunc {
	return func(c *gin.Context) {
		c.Next()
		detectedErrors := c.Errors.ByType(errType)
		if len(detectedErrors) > 0 {
			err := detectedErrors[0].Err
			var parsedError *views.ApiError
			switch err.(type) {
			//&#22914;&#26524;&#20135;&#29983;&#30340;error&#26159;&#33258;&#23450;&#20041;&#30340;&#32467;&#26500;&#20307;,&#36716;&#25442;error,&#36820;&#22238;&#33258;&#23450;&#20041;&#30340;code&#21644;msg
			case *views.ApiError:
				parsedError = err.(*views.ApiError)
			default:
				parsedError = &views.ApiError{
					Code:    http.StatusInternalServerError,
					Message: err.Error(),
				}
			}
			c.IndentedJSON(parsedError.Code, parsedError)
			return
		}

	}
}

//&#35774;&#32622;&#25152;&#26377;&#36328;&#22495;&#35831;&#27714;
func CORSMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		c.Writer.Header().Set("Access-Control-Allow-Origin", "*")
		c.Writer.Header().Set("Access-Control-Allow-Credentials", "true")
		c.Writer.Header().Set("Access-Control-Allow-Headers", "Content-Type, Content-Length, Accept-Encoding, X-CSRF-Token, Authorization, accept, origin, Cache-Control, X-Requested-With")
		c.Writer.Header().Set("Access-Control-Allow-Methods", "POST, OPTIONS, GET, PUT")

		if c.Request.Method == "OPTIONS" {
			c.AbortWithStatus(204)
			return
		}

		c.Next()
	}
}

func main() {
	server := gin.Default()
	server.LoadHTMLGlob(TemplateFiles)
	server.Static("/static", StaticFiles)
	server.Use(JSONAppErrorReporter())
	server.Use(CORSMiddleware())
	server.GET("/", func(c *gin.Context) {
		c.HTML(http.StatusOK, "index.html", nil)
	})
	server.GET("/ws", views.ShellWs)
	server.Run("0.0.0.0:5001")
}