mirror of
https://github.com/veops/oneterm.git
synced 2025-10-05 15:27:01 +08:00
96 lines
2.8 KiB
Go
96 lines
2.8 KiB
Go
package controller
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"golang.org/x/crypto/ssh"
|
|
|
|
"github.com/veops/oneterm/internal/acl"
|
|
"github.com/veops/oneterm/internal/model"
|
|
dbpkg "github.com/veops/oneterm/pkg/db"
|
|
"github.com/veops/oneterm/pkg/utils"
|
|
)
|
|
|
|
var (
|
|
publicKeyPreHooks = []preHook[*model.PublicKey]{
|
|
func(ctx *gin.Context, data *model.PublicKey) {
|
|
if _, comment, _, _, err := ssh.ParseAuthorizedKey([]byte(data.Pk)); err != nil {
|
|
ctx.AbortWithError(http.StatusBadRequest, &ApiError{Code: ErrWrongPk, Data: nil})
|
|
} else {
|
|
data.Pk = strings.TrimSpace(strings.TrimSuffix(data.Pk, comment))
|
|
}
|
|
},
|
|
func(ctx *gin.Context, data *model.PublicKey) {
|
|
data.Pk = utils.EncryptAES(data.Pk)
|
|
},
|
|
func(ctx *gin.Context, data *model.PublicKey) {
|
|
currentUser, _ := acl.GetSessionFromCtx(ctx)
|
|
data.Uid = currentUser.GetUid()
|
|
data.UserName = currentUser.GetUserName()
|
|
},
|
|
}
|
|
publicKeyPostHooks = []postHook[*model.PublicKey]{
|
|
func(ctx *gin.Context, data []*model.PublicKey) {
|
|
for _, d := range data {
|
|
d.Pk = utils.DecryptAES(d.Pk)
|
|
}
|
|
},
|
|
}
|
|
)
|
|
|
|
// CreatePublicKey godoc
|
|
//
|
|
// @Tags public_key
|
|
// @Param publicKey body model.PublicKey true "publicKey"
|
|
// @Success 200 {object} HttpResponse
|
|
// @Router /public_key [post]
|
|
func (c *Controller) CreatePublicKey(ctx *gin.Context) {
|
|
doCreate(ctx, false, &model.PublicKey{}, "", publicKeyPreHooks...)
|
|
}
|
|
|
|
// DeletePublicKey godoc
|
|
//
|
|
// @Tags public_key
|
|
// @Param id path int true "publicKey id"
|
|
// @Success 200 {object} HttpResponse
|
|
// @Router /public_key/:id [delete]
|
|
func (c *Controller) DeletePublicKey(ctx *gin.Context) {
|
|
doDelete(ctx, false, &model.PublicKey{}, "")
|
|
}
|
|
|
|
// UpdatePublicKey godoc
|
|
//
|
|
// @Tags public_key
|
|
// @Param id path int true "publicKey id"
|
|
// @Param publicKey body model.PublicKey true "publicKey"
|
|
// @Success 200 {object} HttpResponse
|
|
// @Router /public_key/:id [put]
|
|
func (c *Controller) UpdatePublicKey(ctx *gin.Context) {
|
|
doUpdate(ctx, false, &model.PublicKey{}, "", publicKeyPreHooks...)
|
|
}
|
|
|
|
// GetPublicKeys godoc
|
|
//
|
|
// @Tags public_key
|
|
// @Param page_index query int true "publicKey id"
|
|
// @Param page_size query int true "publicKey id"
|
|
// @Param search query string false "name or mac"
|
|
// @Param id query int false "publicKey id"
|
|
// @Param name query string false "publicKey name"
|
|
// @Success 200 {object} HttpResponse{data=ListData{list=[]model.PublicKey}}
|
|
// @Router /public_key [get]
|
|
func (c *Controller) GetPublicKeys(ctx *gin.Context) {
|
|
currentUser, _ := acl.GetSessionFromCtx(ctx)
|
|
|
|
db := dbpkg.DB.Model(&model.PublicKey{})
|
|
db = filterSearch(ctx, db, "name", "mac")
|
|
db = filterEqual(ctx, db, "id")
|
|
db = filterLike(ctx, db, "name")
|
|
|
|
db = db.Where("uid = ?", currentUser.Uid)
|
|
|
|
doGet(ctx, false, db, "", publicKeyPostHooks...)
|
|
}
|