This commit is contained in:
2025-02-19 15:01:38 +08:00
commit 584dbb81dc
10 changed files with 1340 additions and 0 deletions

8
backend/.gitignore vendored Normal file
View File

@@ -0,0 +1,8 @@
.idea
markdowns
*.db
manager_backend.tar
usermgmt_service
main
app

34
backend/Dockerfile Normal file
View File

@@ -0,0 +1,34 @@
# Use the Go image for building
FROM golang:1.21-alpine AS builder
WORKDIR /app
# Install dependencies
RUN apk add --no-cache gcc musl-dev sqlite sqlite-dev
# Copy source code and Go module files
COPY go.mod go.sum ./
RUN go mod download
COPY . .
# Build the application
RUN CGO_ENABLED=1 GOOS=linux go build -o app
# Use a minimal image for the final stage
FROM alpine:latest
WORKDIR /app
RUN apk add --no-cache gcc musl-dev sqlite sqlite-dev
# Copy the compiled app and necessary files from the builder stage
COPY --from=builder /app/app .
#COPY --from=builder /app/data /data
# Set environment variables
ENV DB_PATH=/data/admin.db
# Expose the application port
EXPOSE 8080
# Command to run the application
CMD ["./app"]

131
backend/auth/authService.go Normal file
View File

@@ -0,0 +1,131 @@
package auth
import (
"database/sql"
"errors"
_ "github.com/mattn/go-sqlite3"
"golang.org/x/crypto/bcrypt"
"time"
)
type Admin struct {
ID int `json:"id"`
Username string `json:"username"`
Password string `json:"-"` // 不返回密碼
IsActive bool `json:"is_active"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
type AuthService struct {
db *sql.DB
}
func NewAuthService(dbPath string) (*AuthService, error) {
db, err := sql.Open("sqlite3", dbPath)
if err != nil {
return nil, err
}
// 創建管理員表
_, err = db.Exec(`
CREATE TABLE IF NOT EXISTS admins (
id INTEGER PRIMARY KEY AUTOINCREMENT,
username TEXT UNIQUE NOT NULL,
password TEXT NOT NULL,
is_activate BOOLEAN NOT NULL,
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
)
`)
if err != nil {
return nil, err
}
return &AuthService{db: db}, nil
}
// 創建管理員
func (s *AuthService) CreateAdmin(username, password string) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
if err != nil {
return err
}
_, err = s.db.Exec(
"INSERT INTO admins (username, password) VALUES (?, ?)",
username, string(hashedPassword),
)
return err
}
// 驗證管理員
func (s *AuthService) ValidateAdmin(username, password string) (*Admin, error) {
admin := &Admin{}
err := s.db.QueryRow(
"SELECT id, username, password, is_active FROM admins WHERE username = ?",
username,
).Scan(&admin.ID, &admin.Username, &admin.Password, &admin.IsActive)
if !admin.IsActive {
return nil, errors.New("user is not a admin user")
}
if err != nil {
return nil, err
}
err = bcrypt.CompareHashAndPassword([]byte(admin.Password), []byte(password))
if err != nil {
return nil, err
}
return admin, nil
}
// 獲取所有管理員
func (s *AuthService) GetAdmins() ([]Admin, error) {
rows, err := s.db.Query("SELECT id, username, created_at, updated_at, is_active FROM admins")
if err != nil {
return nil, err
}
defer rows.Close()
var admins []Admin
for rows.Next() {
var admin Admin
err := rows.Scan(&admin.ID, &admin.Username, &admin.CreatedAt, &admin.UpdatedAt, &admin.IsActive)
if err != nil {
return nil, err
}
admins = append(admins, admin)
}
return admins, nil
}
// 更新管理員密碼
func (s *AuthService) UpdateAdminPassword(id string, newPassword string) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(newPassword), bcrypt.DefaultCost)
if err != nil {
return err
}
_, err = s.db.Exec(
"UPDATE admins SET password = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ?",
string(hashedPassword), id,
)
return err
}
func (s *AuthService) UpdateAdminActivate(id string, isActive bool) error {
query := `UPDATE admins SET is_active = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ?`
_, err := s.db.Exec(query, isActive, id)
return err
}
// 刪除管理員
func (s *AuthService) DeleteAdmin(id string) error {
_, err := s.db.Exec("DELETE FROM admins WHERE id = ?", id)
return err
}

View File

@@ -0,0 +1,63 @@
package auth
import (
"github.com/gin-gonic/gin"
"github.com/golang-jwt/jwt/v4"
"net/http"
"strings"
"time"
)
var JWTSecret = []byte("your-secret-key") // 在實際使用時應該從環境變量獲取
type Claims struct {
Username string `json:"username"`
jwt.StandardClaims
}
// 生成 JWT token
func GenerateToken(username string) (string, error) {
claims := Claims{
username,
jwt.StandardClaims{
ExpiresAt: time.Now().Add(24 * time.Hour).Unix(),
IssuedAt: time.Now().Unix(),
},
}
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
return token.SignedString(JWTSecret)
}
// JWT 認證中間件
func JWTAuthMiddleware() gin.HandlerFunc {
return func(c *gin.Context) {
authHeader := c.GetHeader("Authorization")
if authHeader == "" {
c.JSON(http.StatusUnauthorized, gin.H{"error": "Authorization header is required"})
c.Abort()
return
}
parts := strings.SplitN(authHeader, " ", 2)
if !(len(parts) == 2 && parts[0] == "Bearer") {
c.JSON(http.StatusUnauthorized, gin.H{"error": "Invalid authorization format"})
c.Abort()
return
}
claims := &Claims{}
token, err := jwt.ParseWithClaims(parts[1], claims, func(token *jwt.Token) (interface{}, error) {
return JWTSecret, nil
})
if err != nil || !token.Valid {
c.JSON(http.StatusUnauthorized, gin.H{"error": "Invalid or expired token"})
c.Abort()
return
}
c.Set("username", claims.Username)
c.Next()
}
}

View File

@@ -0,0 +1,18 @@
#version: '3.8'
services:
backend:
image: golang:1.21-alpine # 使用官方 Go 映像
ports:
- "8080:8080" # 映射端口
volumes:
- ./data:/data # 綁定數據文件夾
- ./:/app # 綁定代碼文件夾
- /var/run:/var/run
working_dir: /app # 設置工作目錄
environment:
- DB_PATH=/data/admin.db # 設定數據庫路徑環境變量
command: sh -c "
apk add --no-cache gcc musl-dev sqlite sqlite-dev &&
go mod download &&
CGO_ENABLED=1 GOOS=linux go build -o app &&
./app"

View File

@@ -0,0 +1,12 @@
version: '3.8'
services:
backend:
image: backend-backend-x86
restart: always
ports:
- "8080:8080" # Map port 8080
volumes:
- ./data:/data # Bind the remote server's data folder
- /var/run:/var/run
environment:
- DB_PATH=/data/admin.db # Set database path as an environment variable

43
backend/go.mod Normal file
View File

@@ -0,0 +1,43 @@
// go.mod
module user-management
go 1.21.0
toolchain go1.22.2
require (
github.com/gin-contrib/cors v1.7.3
github.com/gin-gonic/gin v1.10.0
github.com/golang-jwt/jwt/v4 v4.5.1
github.com/mattn/go-sqlite3 v1.14.24
golang.org/x/crypto v0.31.0
)
require (
github.com/bytedance/sonic v1.12.6 // indirect
github.com/bytedance/sonic/loader v0.2.1 // indirect
github.com/cloudwego/base64x v0.1.4 // indirect
github.com/cloudwego/iasm v0.2.0 // indirect
github.com/gabriel-vasile/mimetype v1.4.7 // indirect
github.com/gin-contrib/sse v0.1.0 // indirect
github.com/go-playground/locales v0.14.1 // indirect
github.com/go-playground/universal-translator v0.18.1 // indirect
github.com/go-playground/validator/v10 v10.23.0 // indirect
github.com/goccy/go-json v0.10.4 // indirect
github.com/json-iterator/go v1.1.12 // indirect
github.com/klauspost/cpuid/v2 v2.2.9 // indirect
github.com/kr/text v0.2.0 // indirect
github.com/leodido/go-urn v1.4.0 // indirect
github.com/mattn/go-isatty v0.0.20 // indirect
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
github.com/modern-go/reflect2 v1.0.2 // indirect
github.com/pelletier/go-toml/v2 v2.2.3 // indirect
github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
github.com/ugorji/go/codec v1.2.12 // indirect
golang.org/x/arch v0.12.0 // indirect
golang.org/x/net v0.33.0 // indirect
golang.org/x/sys v0.28.0 // indirect
golang.org/x/text v0.21.0 // indirect
google.golang.org/protobuf v1.36.1 // indirect
gopkg.in/yaml.v3 v3.0.1 // indirect
)

99
backend/go.sum Normal file
View File

@@ -0,0 +1,99 @@
github.com/bytedance/sonic v1.12.6 h1:/isNmCUF2x3Sh8RAp/4mh4ZGkcFAX/hLrzrK3AvpRzk=
github.com/bytedance/sonic v1.12.6/go.mod h1:B8Gt/XvtZ3Fqj+iSKMypzymZxw/FVwgIGKzMzT9r/rk=
github.com/bytedance/sonic/loader v0.1.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU=
github.com/bytedance/sonic/loader v0.2.1 h1:1GgorWTqf12TA8mma4DDSbaQigE2wOgQo7iCjjJv3+E=
github.com/bytedance/sonic/loader v0.2.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU=
github.com/cloudwego/base64x v0.1.4 h1:jwCgWpFanWmN8xoIUHa2rtzmkd5J2plF/dnLS6Xd/0Y=
github.com/cloudwego/base64x v0.1.4/go.mod h1:0zlkT4Wn5C6NdauXdJRhSKRlJvmclQ1hhJgA0rcu/8w=
github.com/cloudwego/iasm v0.2.0 h1:1KNIy1I1H9hNNFEEH3DVnI4UujN+1zjpuk6gwHLTssg=
github.com/cloudwego/iasm v0.2.0/go.mod h1:8rXZaNYT2n95jn+zTI1sDr+IgcD2GVs0nlbbQPiEFhY=
github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/gabriel-vasile/mimetype v1.4.7 h1:SKFKl7kD0RiPdbht0s7hFtjl489WcQ1VyPW8ZzUMYCA=
github.com/gabriel-vasile/mimetype v1.4.7/go.mod h1:GDlAgAyIRT27BhFl53XNAFtfjzOkLaF35JdEG0P7LtU=
github.com/gin-contrib/cors v1.7.3 h1:hV+a5xp8hwJoTw7OY+a70FsL8JkVVFTXw9EcfrYUdns=
github.com/gin-contrib/cors v1.7.3/go.mod h1:M3bcKZhxzsvI+rlRSkkxHyljJt1ESd93COUvemZ79j4=
github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
github.com/gin-gonic/gin v1.10.0 h1:nTuyha1TYqgedzytsKYqna+DfLos46nTv2ygFy86HFU=
github.com/gin-gonic/gin v1.10.0/go.mod h1:4PMNQiOhvDRa013RKVbsiNwoyezlm2rm0uX/T7kzp5Y=
github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=
github.com/go-playground/assert/v2 v2.2.0/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4=
github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA=
github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=
github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY=
github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
github.com/go-playground/validator/v10 v10.23.0 h1:/PwmTwZhS0dPkav3cdK9kV1FsAmrL8sThn8IHr/sO+o=
github.com/go-playground/validator/v10 v10.23.0/go.mod h1:dbuPbCMFw/DrkbEynArYaCwl3amGuJotoKCe95atGMM=
github.com/goccy/go-json v0.10.4 h1:JSwxQzIqKfmFX1swYPpUThQZp/Ka4wzJdK0LWVytLPM=
github.com/goccy/go-json v0.10.4/go.mod h1:oq7eo15ShAhp70Anwd5lgX2pLfOS3QCiwU/PULtXL6M=
github.com/golang-jwt/jwt/v4 v4.5.1 h1:JdqV9zKUdtaa9gdPlywC3aeoEsR681PlKC+4F5gQgeo=
github.com/golang-jwt/jwt/v4 v4.5.1/go.mod h1:m21LjoU+eqJr34lmDMbreY2eSTRJ1cv77w39/MY0Ch0=
github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU=
github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
github.com/klauspost/cpuid/v2 v2.2.9 h1:66ze0taIn2H33fBvCkXuv9BmCwDfafmiIVpKV9kKGuY=
github.com/klauspost/cpuid/v2 v2.2.9/go.mod h1:rqkxqrZ1EhYM9G+hXH7YdowN5R5RGN6NK4QwQ3WMXF8=
github.com/knz/go-libedit v1.10.1/go.mod h1:MZTVkCWyz0oBc7JOWP3wNAzd002ZbM/5hgShxwh4x8M=
github.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0=
github.com/kr/pretty v0.3.0/go.mod h1:640gp4NfQd8pI5XOwp5fnNeVWj67G7CFk/SaSQn7NBk=
github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
github.com/leodido/go-urn v1.4.0 h1:WT9HwE9SGECu3lg4d/dIA+jxlljEa1/ffXKmRjqdmIQ=
github.com/leodido/go-urn v1.4.0/go.mod h1:bvxc+MVxLKB4z00jd1z+Dvzr47oO32F/QSNjSBOlFxI=
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
github.com/mattn/go-sqlite3 v1.14.24 h1:tpSp2G2KyMnnQu99ngJ47EIkWVmliIizyZBfPrBWDRM=
github.com/mattn/go-sqlite3 v1.14.24/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
github.com/pelletier/go-toml/v2 v2.2.3 h1:YmeHyLY8mFWbdkNWwpr+qIL2bEqT0o95WSdkNHvL12M=
github.com/pelletier/go-toml/v2 v2.2.3/go.mod h1:MfCQTFTvCcUyyvvwm1+G6H/jORL20Xlb6rzQu9GuUkc=
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/rogpeppe/go-internal v1.8.0 h1:FCbCCtXNOY3UtUuHUYaghJg4y7Fd14rXifAYUAtL9R8=
github.com/rogpeppe/go-internal v1.8.0/go.mod h1:WmiCO8CzOY8rg0OYDC4/i/2WRWAB6poM+XZ2dLUbcbE=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
github.com/ugorji/go/codec v1.2.12 h1:9LC83zGrHhuUA9l16C9AHXAqEV/2wBQ4nkvumAE65EE=
github.com/ugorji/go/codec v1.2.12/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=
golang.org/x/arch v0.12.0 h1:UsYJhbzPYGsT0HbEdmYcqtCv8UNGvnaL561NnIUvaKg=
golang.org/x/arch v0.12.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys=
golang.org/x/crypto v0.31.0 h1:ihbySMvVjLAeSH1IbfcRTkD/iNscyz8rGzjF/E5hV6U=
golang.org/x/crypto v0.31.0/go.mod h1:kDsLvtWBEx7MV9tJOj9bnXsPbxwJQ6csT/x4KIN4Ssk=
golang.org/x/net v0.33.0 h1:74SYHlV8BIgHIFC/LrYkOGIwL19eTYXQ5wc6TBuO36I=
golang.org/x/net v0.33.0/go.mod h1:HXLR5J+9DxmrqMwG9qjGCxZ+zKXxBru04zlTvWlWuN4=
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.28.0 h1:Fksou7UEQUWlKvIdsqzJmUmCX3cZuD2+P3XyyzwMhlA=
golang.org/x/sys v0.28.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
golang.org/x/text v0.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo=
golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ=
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
google.golang.org/protobuf v1.36.1 h1:yBPeRvTftaleIgM3PZ/WBIZ7XM/eEYAaEyCwvyjq/gk=
google.golang.org/protobuf v1.36.1/go.mod h1:9fA7Ob0pmnwhb644+1+CVWFRbNajQ6iRojtC/QF5bRE=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
nullprogram.com/x/optparse v1.0.0/go.mod h1:KdyPE+Igbe0jQUrVfMqDMeJQIJZEuyV7pjYmp6pbG50=

413
backend/main.go Normal file

File diff suppressed because one or more lines are too long

519
backend/system/client.go Normal file
View File

@@ -0,0 +1,519 @@
// system/client.go
package system
import (
"bufio"
"encoding/json"
"fmt"
"net"
"strings"
)
type SystemClient struct {
socketPath string
}
type Command struct {
Action string `json:"action"`
Params map[string]interface{} `json:"params"`
}
type Response struct {
Success bool `json:"success"`
Data interface{} `json:"data"`
Error string `json:"error,omitempty"`
}
type User struct {
Username string `json:"username"`
UID string `json:"uid"`
GID string `json:"gid"`
HomeDir string `json:"home_dir"`
Shell string `json:"shell"`
}
type Group struct {
GroupName string `json:"groupName"`
}
type SambaConnection struct {
PID string `json:"pid"`
Username string `json:"username"`
Group string `json:"group"`
Machine string `json:"machine"`
Protocol string `json:"protocol"`
Folder string `json:"folder"`
}
func NewSystemClient(socketPath string) *SystemClient {
return &SystemClient{
socketPath: socketPath,
}
}
func (c *SystemClient) sendCommand(cmd Command) (*Response, error) {
// 連接到 Unix socket
conn, err := net.Dial("unix", c.socketPath)
if err != nil {
return nil, fmt.Errorf("failed to connect to socket: %v", err)
}
defer conn.Close()
// 發送命令
encoder := json.NewEncoder(conn)
if err := encoder.Encode(cmd); err != nil {
return nil, fmt.Errorf("failed to encode command: %v", err)
}
// 讀取響應
decoder := json.NewDecoder(conn)
var response Response
if err := decoder.Decode(&response); err != nil {
return nil, fmt.Errorf("failed to decode response: %v", err)
}
return &response, nil
}
func (c *SystemClient) GetUsers() ([]User, error) {
resp, err := c.sendCommand(Command{
Action: "get_users",
Params: nil,
})
if err != nil {
return nil, err
}
if !resp.Success {
return nil, fmt.Errorf(resp.Error)
}
var users []User
dataStr, ok := resp.Data.(string)
if !ok {
fmt.Println("Error: Data is not a string")
return users, nil
}
scanner := bufio.NewScanner(strings.NewReader(dataStr))
for scanner.Scan() {
line := scanner.Text()
parts := strings.Split(line, ":")
if len(parts) < 7 {
continue // Skip malformed lines
}
username := parts[0]
uid := parts[2]
gid := parts[3]
homeDir := parts[5]
shell := parts[6]
user := User{
Username: username,
UID: uid,
GID: gid,
HomeDir: homeDir,
Shell: shell,
}
users = append(users, user)
}
if err := scanner.Err(); err != nil {
fmt.Println("Error reading command output:", err)
return users, nil
}
// Marshal to JSON and print
_, err = json.MarshalIndent(users, "", " ")
if err != nil {
fmt.Println("Error marshalling to JSON:", err)
return users, nil
}
return users, nil
}
func (c *SystemClient) GetGroups() ([]Group, error) {
resp, err := c.sendCommand(Command{
Action: "get_groups",
Params: nil,
})
if err != nil {
return nil, err
}
if !resp.Success {
return nil, fmt.Errorf(resp.Error)
}
var groups []Group
dataStr, ok := resp.Data.(string)
if !ok {
fmt.Println("Error: Data is not a string")
return groups, nil
}
scanner := bufio.NewScanner(strings.NewReader(dataStr))
for scanner.Scan() {
line := scanner.Text()
group := Group{
GroupName: line,
}
groups = append(groups, group)
}
if err := scanner.Err(); err != nil {
fmt.Println("Error reading command output:", err)
return groups, nil
}
// Marshal to JSON and print
_, err = json.MarshalIndent(groups, "", " ")
if err != nil {
fmt.Println("Error marshalling to JSON:", err)
return groups, nil
}
return groups, nil
}
func (c *SystemClient) GetUserGroups(username string) ([]Group, error) {
resp, err := c.sendCommand(Command{
Action: "get_user_groups",
Params: map[string]interface{}{
"username": username,
},
})
if err != nil {
return nil, err
}
if !resp.Success {
return nil, fmt.Errorf(resp.Error)
}
var groups []Group
dataStr, ok := resp.Data.([]interface{})
if !ok {
fmt.Println("Error: Data is not a string")
return groups, nil
}
for _, data := range dataStr {
str, ok := data.(string)
if !ok {
return nil, fmt.Errorf("data contains non-string element")
}
group := Group{
GroupName: str,
}
groups = append(groups, group)
}
// Marshal to JSON and print
_, err = json.MarshalIndent(groups, "", " ")
if err != nil {
fmt.Println("Error marshalling to JSON:", err)
return groups, nil
}
return groups, nil
}
func (c *SystemClient) AddUserToGroups(username string, groups []string) error {
// Loop through the groups and add the user to each group
for _, group := range groups {
// Implement logic to add the user to the group
resp, err := c.sendCommand(Command{
Action: "add_user_to_group",
Params: map[string]interface{}{
"username": username,
"group": group,
},
})
if err != nil {
return err
}
if !resp.Success {
return fmt.Errorf(resp.Error)
}
}
return nil
}
func (c *SystemClient) RemoveUserFromGroup(username, group string) error {
resp, err := c.sendCommand(Command{
Action: "remove_user_from_group",
Params: map[string]interface{}{
"username": username,
"group": group,
},
})
if err != nil {
return err
}
if !resp.Success {
return fmt.Errorf(resp.Error)
}
return nil
}
func (c *SystemClient) CreateUser(username, password string, group string, isAdmin bool, shell string) error {
resp, err := c.sendCommand(Command{
Action: "create_user",
Params: map[string]interface{}{
"username": username,
"password": password,
"group": group,
"is_admin": isAdmin,
"shell": shell,
},
})
if err != nil {
return err
}
if !resp.Success {
return fmt.Errorf(resp.Error)
}
return nil
}
func (c *SystemClient) ModifyUserPasswd(username string, password string) error {
resp, err := c.sendCommand(Command{
Action: "modify_user_passwd",
Params: map[string]interface{}{
"username": username,
"password": password,
},
})
if err != nil {
return err
}
if !resp.Success {
return fmt.Errorf(resp.Error)
}
return nil
}
func (c *SystemClient) DeleteUser(username string) error {
resp, err := c.sendCommand(Command{
Action: "delete_user",
Params: map[string]interface{}{
"username": username,
},
})
if err != nil {
return err
}
if !resp.Success {
return fmt.Errorf(resp.Error)
}
return nil
}
func (c *SystemClient) GetSambaConnections() ([]SambaConnection, error) {
resp, err := c.sendCommand(Command{
Action: "get_samba_status",
Params: nil,
})
if err != nil {
return nil, err
}
if !resp.Success {
return nil, fmt.Errorf(resp.Error)
}
var connections []SambaConnection
data, err := json.Marshal(resp.Data)
if err != nil {
return nil, err
}
if err := json.Unmarshal(data, &connections); err != nil {
return nil, err
}
return connections, nil
}
func (c *SystemClient) GetSambaGlobalSetting() (string, error) {
resp, err := c.sendCommand(Command{
Action: "get_samba_settings",
Params: nil,
})
if err != nil {
return "", err
}
if !resp.Success {
return "", fmt.Errorf(resp.Error)
}
settings_str, ok := resp.Data.(string)
if !ok {
fmt.Println("Error: Data is not a string")
return "", nil
}
return settings_str, nil
}
func (c *SystemClient) GetSambaSectionSetting() (string, error) {
resp, err := c.sendCommand(Command{
Action: "get_samba_settings",
Params: nil,
})
if err != nil {
return "", err
}
if !resp.Success {
return "", fmt.Errorf(resp.Error)
}
settings_str, ok := resp.Data.(string)
if !ok {
fmt.Println("Error: Data is not a string")
return "", nil
}
return settings_str, nil
}
func (c *SystemClient) UpdateSambaSectionSetting(fullSectionSettingsJSON string) error {
// Step 1: Get all current settings
settings, err := c.GetSambaSectionSetting()
if err != nil {
return fmt.Errorf("failed to fetch current Samba settings: %v", err)
}
// Step 2: Parse the settings JSON string into a map
var settingsMap map[string]interface{}
if err := json.Unmarshal([]byte(settings), &settingsMap); err != nil {
return fmt.Errorf("failed to parse Samba settings JSON: %v", err)
}
// Step 3: Parse the full new section settings JSON
var newSectionSettings map[string]interface{}
if err := json.Unmarshal([]byte(fullSectionSettingsJSON), &newSectionSettings); err != nil {
return fmt.Errorf("failed to parse new section settings JSON: %v", err)
}
// Step 4: Replace the entire "sections" field
settingsMap["sections"] = newSectionSettings
// Step 5: Convert updated settings back to JSON
updatedSettings, err := json.Marshal(settingsMap)
if err != nil {
return fmt.Errorf("failed to marshal updated settings to JSON: %v", err)
}
// Step 6: Send the updated settings using `update_section_setting`
resp, err := c.sendCommand(Command{
Action: "update_section_setting",
Params: map[string]interface{}{"config": string(updatedSettings)},
})
if err != nil {
return fmt.Errorf("failed to send updated settings: %v", err)
}
if !resp.Success {
return fmt.Errorf("failed to update section settings: %v", resp.Error)
}
return nil
}
func (c *SystemClient) UpdateSambaGlobalSetting(fullGlobalSettingsJSON string) error {
// Step 1: Get all current settings
settings, err := c.GetSambaGlobalSetting()
if err != nil {
return fmt.Errorf("failed to fetch current Samba settings: %v", err)
}
// Step 2: Parse the settings JSON string into a map
var settingsMap map[string]interface{}
if err := json.Unmarshal([]byte(settings), &settingsMap); err != nil {
return fmt.Errorf("failed to parse Samba settings JSON: %v", err)
}
// Step 3: Parse the full new global settings JSON
var newGlobalSettings map[string]interface{}
if err := json.Unmarshal([]byte(fullGlobalSettingsJSON), &newGlobalSettings); err != nil {
return fmt.Errorf("failed to parse new global settings JSON: %v", err)
}
// Step 4: Replace the entire "global" field
settingsMap["global"] = newGlobalSettings
// Step 5: Convert updated settings back to JSON
updatedSettings, err := json.Marshal(settingsMap)
if err != nil {
return fmt.Errorf("failed to marshal updated settings to JSON: %v", err)
}
// Step 6: Send the updated settings using `update_section_setting`
resp, err := c.sendCommand(Command{
Action: "update_section_setting",
Params: map[string]interface{}{"config": string(updatedSettings)},
})
if err != nil {
return fmt.Errorf("failed to send updated settings: %v", err)
}
if !resp.Success {
return fmt.Errorf("failed to update global settings: %v", resp.Error)
}
return nil
}
func (c *SystemClient) GetSambaServiceInfo() (string, error) {
resp, err := c.sendCommand(Command{
Action: "get_samba_service_info",
Params: nil,
})
if err != nil {
return "", err
}
if !resp.Success {
return "", fmt.Errorf(resp.Error)
}
dataStr, ok := resp.Data.(string)
if !ok {
fmt.Println("Error: Data is not a string")
return "", nil
}
return dataStr, nil
}
func (c *SystemClient) RestartSambaService() error {
resp, err := c.sendCommand(Command{
Action: "restart_samba_service",
Params: nil,
})
if err != nil {
return err
}
if !resp.Success {
return fmt.Errorf(resp.Error)
}
return nil
}