add ChangePort() to common (#11)

Author: tigerinus

build/scripts/migration/service.d/gateway/migration.list

@@ -1,2 +1,2 @@
-LEGACY_WITHOUT_VERSION v0.3.6-alpha3
-v0.3.5 v0.3.6-alpha3
+LEGACY_WITHOUT_VERSION v0.3.6-alpha4
+v0.3.5 v0.3.6-alpha4

common/management.go

@@ -15,19 +15,21 @@ import (
 const (
 	ManagementURLFilename = "management.url"
 	StaticURLFilename     = "static.url"
-	APIPath               = "/v1/gateway/routes"
+	APIGatewayRoutes      = "/v1/gateway/routes"
+	APIGatewayPort        = "/v1/gateway/port"
 )
 
 type ManagementService interface {
 	CreateRoute(route *Route) error
+	ChangePort(request *ChangePortRequest) error
 }
 
 type managementService struct {
 	address string
 }
 
 func (m *managementService) CreateRoute(route *Route) error {
-	url := strings.TrimSuffix(m.address, "/") + "/" + strings.TrimPrefix(APIPath, "/")
+	url := strings.TrimSuffix(m.address, "/") + "/" + strings.TrimPrefix(APIGatewayRoutes, "/")
 	body, err := json.Marshal(route)
 	if err != nil {
 		return err
@@ -45,10 +47,35 @@ func (m *managementService) CreateRoute(route *Route) error {
 	return nil
 }
 
+func (m *managementService) ChangePort(request *ChangePortRequest) error {
+	url := strings.TrimSuffix(m.address, "/") + "/" + strings.TrimPrefix(APIGatewayPort, "/")
+	body, err := json.Marshal(request)
+	if err != nil {
+		return err
+	}
+
+	req, err := http.NewRequest(http.MethodPut, url, bytes.NewBuffer(body))
+	if err != nil {
+		return err
+	}
+
+	req.Header.Set("Content-Type", "application/json")
+	response, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return err
+	}
+
+	if response.StatusCode != http.StatusOK {
+		return errors.New("failed to change port (status code: " + fmt.Sprint(response.StatusCode) + ")")
+	}
+
+	return nil
+}
+
 func NewManagementService(RuntimePath string) (ManagementService, error) {
-	managementAddresFile := filepath.Join(RuntimePath, ManagementURLFilename)
+	managementAddressFile := filepath.Join(RuntimePath, ManagementURLFilename)
 
-	buf, err := ioutil.ReadFile(managementAddresFile)
+	buf, err := ioutil.ReadFile(managementAddressFile)
 	if err != nil {
 		return nil, err
 	}

go.mod

@@ -3,7 +3,7 @@ module github.com/IceWhaleTech/CasaOS-Gateway
 go 1.19
 
 require (
-	github.com/IceWhaleTech/CasaOS-Common v0.0.0-20220829003259-cf7e89c30d78
+	github.com/IceWhaleTech/CasaOS-Common v0.0.0-20220830042034-a4df79c84ce6
 	github.com/gin-contrib/gzip v0.0.6
 	github.com/gin-gonic/gin v1.8.1
 	github.com/go-playground/assert/v2 v2.0.1

go.sum

@@ -39,8 +39,8 @@ dmitri.shuralyov.com/gpu/mtl v0.0.0-20190408044501-666a987793e9/go.mod h1:H6x//7
 github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
 github.com/BurntSushi/toml v1.2.0 h1:Rt8g24XnyGTyglgET/PRUNlrUeu9F5L+7FilkXfZgs0=
 github.com/BurntSushi/xgb v0.0.0-20160522181843-27f122750802/go.mod h1:IVnqGOEym/WlBOVXweHU+Q+/VP0lqqI8lqeDx9IjBqo=
-github.com/IceWhaleTech/CasaOS-Common v0.0.0-20220829003259-cf7e89c30d78 h1:Cxy+bTjqpl0txx6gn2/D7O/E6G2SW5c6Ve7RXUin5K4=
-github.com/IceWhaleTech/CasaOS-Common v0.0.0-20220829003259-cf7e89c30d78/go.mod h1:5sqNKg5cEH7IUnCklLSTrVoGx1dMBhm9DFDsCYVPvPQ=
+github.com/IceWhaleTech/CasaOS-Common v0.0.0-20220830042034-a4df79c84ce6 h1:ICitt/XvD1oh3BTdQbzV95G48TFY/aiZr55iA8dMD8M=
+github.com/IceWhaleTech/CasaOS-Common v0.0.0-20220830042034-a4df79c84ce6/go.mod h1:5sqNKg5cEH7IUnCklLSTrVoGx1dMBhm9DFDsCYVPvPQ=
 github.com/andybalholm/brotli v1.0.1 h1:KqhlKozYbRtJvsPrrEeXcO+N2l6NYT5A2QAFmSULpEc=
 github.com/andybalholm/brotli v1.0.1/go.mod h1:loMXtMfwqflxFJPmdbJO0a3KNoPuLBgiu3qAvBg8x/Y=
 github.com/benbjohnson/clock v1.1.0/go.mod h1:J11/hYXuz8f4ySSvYwY0FKfm+ezbsZBKZxNJlLklBHA=

route/management_route.go

@@ -88,15 +88,15 @@ func (m *ManagementRoute) buildV1RouteGroup(v1Group *gin.RouterGroup) {
 			ctx.Status(http.StatusCreated)
 		})
 
-		v1GatewayGroup.GET("/port", jwt.JWT(), func(ctx *gin.Context) {
+		v1GatewayGroup.GET("/port", jwt.ExceptLocalhost(), func(ctx *gin.Context) {
 			ctx.JSON(http.StatusOK, model.Result{
 				Success: common_err.SUCCESS,
 				Message: common_err.GetMsg(common_err.SUCCESS),
 				Data:    m.management.GetGatewayPort(),
 			})
 		})
 
-		v1GatewayGroup.PUT("/port", jwt.JWT(), func(ctx *gin.Context) {
+		v1GatewayGroup.PUT("/port", jwt.ExceptLocalhost(), func(ctx *gin.Context) {
 			var request *common.ChangePortRequest
 
 			if err := ctx.ShouldBindJSON(&request); err != nil {