Initial commit

pkg/api/routes.go

@@ -0,0 +1,145 @@
+package api
+
+import (
+	"github.com/labstack/echo/v4"
+	"github.com/labstack/echo/v4/middleware"
+	"net/http"
+	"next-terminal/pkg/config"
+	"next-terminal/pkg/model"
+)
+
+const Token = "X-Auth-Token"
+
+func SetupRoutes() *echo.Echo {
+
+	// Echo instance
+	e := echo.New()
+
+	e.File("/", "web/build/index.html")
+	e.File("/logo.svg", "web/build/logo.svg")
+	e.File("/favicon.ico", "web/build/favicon.ico")
+	e.Static("/static", "web/build/static")
+
+	// Middleware
+	e.Use(middleware.Logger())
+	e.Use(middleware.Recover())
+	e.Use(middleware.CORSWithConfig(middleware.CORSConfig{
+		Skipper:      middleware.DefaultSkipper,
+		AllowOrigins: []string{"*"},
+		AllowMethods: []string{http.MethodGet, http.MethodHead, http.MethodPut, http.MethodPatch, http.MethodPost, http.MethodDelete},
+	}))
+	e.Use(Auth)
+
+	e.POST("/login", LoginEndpoint)
+
+	e.GET("/tunnel", TunEndpoint)
+	e.GET("/ssh", SSHEndpoint)
+
+	e.POST("/logout", LogoutEndpoint)
+	e.POST("/change-password", ChangePasswordEndpoint)
+	e.GET("/info", InfoEndpoint)
+
+	users := e.Group("/users")
+	{
+		users.POST("", UserCreateEndpoint)
+		users.GET("/paging", UserPagingEndpoint)
+		users.PUT("/:id", UserUpdateEndpoint)
+		users.DELETE("/:id", UserDeleteEndpoint)
+		users.GET("/:id", UserGetEndpoint)
+	}
+
+	assets := e.Group("/assets", Auth)
+	{
+		assets.GET("", AssetAllEndpoint)
+		assets.POST("", AssetCreateEndpoint)
+		assets.GET("/paging", AssetPagingEndpoint)
+		assets.POST("/:id/tcping", AssetTcpingEndpoint)
+		assets.PUT("/:id", AssetUpdateEndpoint)
+		assets.DELETE("/:id", AssetDeleteEndpoint)
+		assets.GET("/:id", AssetGetEndpoint)
+	}
+
+	commands := e.Group("/commands")
+	{
+		commands.GET("/paging", CommandPagingEndpoint)
+		commands.POST("", CommandCreateEndpoint)
+		commands.PUT("/:id", CommandUpdateEndpoint)
+		commands.DELETE("/:id", CommandDeleteEndpoint)
+		commands.GET("/:id", CommandGetEndpoint)
+	}
+
+	credentials := e.Group("/credentials")
+	{
+		credentials.GET("", CredentialAllEndpoint)
+		credentials.GET("/paging", CredentialPagingEndpoint)
+		credentials.POST("", CredentialCreateEndpoint)
+		credentials.PUT("/:id", CredentialUpdateEndpoint)
+		credentials.DELETE("/:id", CredentialDeleteEndpoint)
+		credentials.GET("/:id", CredentialGetEndpoint)
+	}
+
+	sessions := e.Group("/sessions")
+	{
+		sessions.POST("", SessionCreateEndpoint)
+		sessions.GET("/paging", SessionPagingEndpoint)
+		sessions.POST("/:id/content", SessionContentEndpoint)
+		sessions.POST("/:id/discontent", SessionDiscontentEndpoint)
+		sessions.POST("/:id/resize", SessionResizeEndpoint)
+		sessions.POST("/:id/upload", SessionUploadEndpoint)
+		sessions.GET("/:id/download", SessionDownloadEndpoint)
+		sessions.GET("/:id/ls", SessionLsEndpoint)
+		sessions.POST("/:id/mkdir", SessionMkDirEndpoint)
+		sessions.DELETE("/:id/rmdir", SessionRmDirEndpoint)
+		sessions.DELETE("/:id/rm", SessionRmEndpoint)
+		sessions.DELETE("/:id", SessionDeleteEndpoint)
+	}
+
+	e.GET("/properties", PropertyGetEndpoint)
+	e.PUT("/properties", PropertyUpdateEndpoint)
+
+	e.GET("/overview/status", OverviewStatusEndPoint)
+	e.GET("/overview/counter", OverviewCounterEndPoint)
+
+	return e
+}
+
+type H map[string]interface{}
+
+func Fail(c echo.Context, code int, message string) error {
+	return c.JSON(200, H{
+		"code":    code,
+		"message": message,
+	})
+}
+
+func Success(c echo.Context, data interface{}) error {
+	return c.JSON(200, H{
+		"code":    1,
+		"message": "success",
+		"data":    data,
+	})
+}
+
+func NotFound(c echo.Context, message string) error {
+	return c.JSON(200, H{
+		"code":    -1,
+		"message": message,
+	})
+}
+
+func GetToken(c echo.Context) string {
+	token := c.Request().Header.Get(Token)
+	if len(token) > 0 {
+		return token
+	}
+	return c.QueryParam(Token)
+}
+
+func GetCurrentAccount(c echo.Context) (model.User, bool) {
+	token := GetToken(c)
+	get, b := config.Cache.Get(token)
+	if b {
+		return get.(model.User), true
+	}
+	return model.User{}, false
+}