feat: initial commit - Go + CGO captcha recognition service
Features: - Go + CGO ONNX/OpenCV wrapper for high performance - SQLite (default) / MySQL database support - Optional Redis caching - JWT authentication system - Multiple captcha recognition APIs: - OCR text recognition - Slider captcha matching - Image similarity comparison - Rotation captcha detection - Object detection - React frontend with install wizard - Docker and docker-compose support - Gitea CI/CD pipeline Project structure: - cmd/server: Main entry point - internal/: Core business logic - pkg/onnx: ONNX Runtime CGO wrapper - pkg/opencv: OpenCV CGO wrapper - web/: React frontend - deploy/: Deployment configs - scripts/: Utility scripts
This commit is contained in:
@@ -0,0 +1,126 @@
|
||||
name: Build and Deploy
|
||||
|
||||
on:
|
||||
push:
|
||||
branches:
|
||||
- main
|
||||
tags:
|
||||
- 'v*'
|
||||
|
||||
env:
|
||||
REGISTRY: registry.cn-hangzhou.aliyuncs.com
|
||||
IMAGE_NAME: anticaptcha
|
||||
|
||||
jobs:
|
||||
build-frontend:
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
|
||||
- name: Setup Node.js
|
||||
uses: actions/setup-node@v4
|
||||
with:
|
||||
node-version: '20'
|
||||
cache: 'npm'
|
||||
cache-dependency-path: web/package-lock.json
|
||||
|
||||
- name: Install dependencies
|
||||
working-directory: ./web
|
||||
run: npm ci
|
||||
|
||||
- name: Build frontend
|
||||
working-directory: ./web
|
||||
run: npm run build
|
||||
|
||||
- name: Upload frontend artifacts
|
||||
uses: actions/upload-artifact@v4
|
||||
with:
|
||||
name: frontend
|
||||
path: web/dist
|
||||
|
||||
build-backend:
|
||||
runs-on: ubuntu-latest
|
||||
needs: build-frontend
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
|
||||
- name: Download frontend artifacts
|
||||
uses: actions/download-artifact@v4
|
||||
with:
|
||||
name: frontend
|
||||
path: web/dist
|
||||
|
||||
- name: Setup Go
|
||||
uses: actions/setup-go@v5
|
||||
with:
|
||||
go-version: '1.22'
|
||||
|
||||
- name: Build Go binary
|
||||
run: |
|
||||
CGO_ENABLED=0 GOOS=linux go build -ldflags="-s -w" -o anticaptcha ./cmd/server
|
||||
|
||||
- name: Upload backend artifacts
|
||||
uses: actions/upload-artifact@v4
|
||||
with:
|
||||
name: backend
|
||||
path: anticaptcha
|
||||
|
||||
build-docker:
|
||||
runs-on: ubuntu-latest
|
||||
needs: build-frontend
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
|
||||
- name: Download frontend artifacts
|
||||
uses: actions/download-artifact@v4
|
||||
with:
|
||||
name: frontend
|
||||
path: web/dist
|
||||
|
||||
- name: Set up Docker Buildx
|
||||
uses: docker/setup-buildx-action@v3
|
||||
|
||||
- name: Login to Registry
|
||||
uses: docker/login-action@v3
|
||||
with:
|
||||
registry: ${{ env.REGISTRY }}
|
||||
username: ${{ secrets.REGISTRY_USERNAME }}
|
||||
password: ${{ secrets.REGISTRY_PASSWORD }}
|
||||
|
||||
- name: Extract metadata
|
||||
id: meta
|
||||
uses: docker/metadata-action@v5
|
||||
with:
|
||||
images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
|
||||
tags: |
|
||||
type=ref,event=branch
|
||||
type=semver,pattern={{version}}
|
||||
type=semver,pattern={{major}}.{{minor}}
|
||||
type=sha
|
||||
|
||||
- name: Build and push
|
||||
uses: docker/build-push-action@v5
|
||||
with:
|
||||
context: .
|
||||
push: true
|
||||
tags: ${{ steps.meta.outputs.tags }}
|
||||
labels: ${{ steps.meta.outputs.labels }}
|
||||
cache-from: type=gha
|
||||
cache-to: type=gha,mode=max
|
||||
|
||||
deploy:
|
||||
runs-on: ubuntu-latest
|
||||
needs: build-docker
|
||||
if: startsWith(gitea.ref, 'refs/tags/')
|
||||
steps:
|
||||
- name: Deploy to server
|
||||
uses: appleboy/ssh-action@v1.0.3
|
||||
with:
|
||||
host: ${{ secrets.SSH_HOST }}
|
||||
username: ${{ secrets.SSH_USER }}
|
||||
password: ${{ secrets.SSH_PASSWORD }}
|
||||
script: |
|
||||
cd /opt/anticaptcha
|
||||
docker-compose pull
|
||||
docker-compose up -d
|
||||
docker image prune -f
|
||||
+41
@@ -0,0 +1,41 @@
|
||||
# 忽略编译产物
|
||||
anticaptcha
|
||||
*.exe
|
||||
*.exe~
|
||||
*.dll
|
||||
*.so
|
||||
*.dylib
|
||||
|
||||
# 忽略数据目录
|
||||
data/
|
||||
*.db
|
||||
|
||||
# 忽略日志
|
||||
*.log
|
||||
|
||||
# 忽略 IDE 配置
|
||||
.idea/
|
||||
.vscode/
|
||||
*.swp
|
||||
*.swo
|
||||
|
||||
# 忽略依赖
|
||||
vendor/
|
||||
node_modules/
|
||||
|
||||
# 忽略前端构建产物
|
||||
web/dist/
|
||||
web/node_modules/
|
||||
|
||||
# 忽略环境配置
|
||||
.env
|
||||
.env.local
|
||||
*.local
|
||||
|
||||
# 忽略密钥
|
||||
secret.key
|
||||
config.yaml
|
||||
|
||||
# 忽略临时文件
|
||||
*.tmp
|
||||
*.temp
|
||||
+40
@@ -0,0 +1,40 @@
|
||||
# 构建阶段
|
||||
FROM golang:1.22-alpine AS builder
|
||||
|
||||
# 安装构建依赖
|
||||
RUN apk add --no-cache git gcc g++ make opencv-dev onnxruntime-dev
|
||||
|
||||
WORKDIR /app
|
||||
|
||||
# 复制 Go 模块文件
|
||||
COPY go.mod go.sum ./
|
||||
RUN go mod download
|
||||
|
||||
# 复制源代码
|
||||
COPY . .
|
||||
|
||||
# 构建
|
||||
RUN CGO_ENABLED=1 go build -o anticaptcha ./cmd/server
|
||||
|
||||
# 运行阶段
|
||||
FROM alpine:3.20
|
||||
|
||||
# 安装运行时依赖
|
||||
RUN apk add --no-cache ca-certificates tzdata opencv onnxruntime
|
||||
|
||||
WORKDIR /app
|
||||
|
||||
# 复制二进制文件
|
||||
COPY --from=builder /app/anticaptcha .
|
||||
COPY --from=builder /app/models ./models
|
||||
COPY --from=builder /app/web/dist ./web/dist
|
||||
|
||||
# 创建数据目录
|
||||
RUN mkdir -p /app/data
|
||||
|
||||
ENV GIN_MODE=release
|
||||
ENV TZ=Asia/Shanghai
|
||||
|
||||
EXPOSE 6688
|
||||
|
||||
CMD ["./anticaptcha"]
|
||||
@@ -0,0 +1,34 @@
|
||||
MIT License
|
||||
|
||||
Copyright (c) 2024 AntiCaptcha
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
of this software and associated documentation files (the "Software"), to deal
|
||||
in the Software without restriction, including without limitation the rights
|
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
copies of the Software, and to permit persons to whom the Software is
|
||||
furnished to do so, subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included in all
|
||||
copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
SOFTWARE.
|
||||
|
||||
---
|
||||
|
||||
免责声明：
|
||||
|
||||
本项目基于MIT开源协议发布，欢迎自由使用、修改和分发，但必须遵守中华人民共和国法律法规。
|
||||
|
||||
使用本项目即表示您已阅读并同意以下条款：
|
||||
1. 合法使用：不得将本项目用于任何违法、违规或侵犯他人权益的行为。
|
||||
2. 风险自负：任何因使用本项目而产生的法律责任由使用者自行承担，项目作者不承担责任。
|
||||
3. 禁止滥用：不得将本项目用于黑产或其他不当商业用途。
|
||||
|
||||
使用视为同意上述条款，如不同意请立即停止使用并删除本项目。
|
||||
@@ -0,0 +1,91 @@
|
||||
# AntiCaptcha
|
||||
|
||||
基于 Go + CGO 的高性能验证码识别服务，支持多种验证码类型识别。
|
||||
|
||||
## 功能特性
|
||||
|
||||
- **验证码识别**：OCR、滑块、旋转验证码、目标检测等
|
||||
- **多数据库支持**：SQLite（默认）、MySQL 可选
|
||||
- **可选 Redis**：用于缓存和会话管理
|
||||
- **RESTful API**：标准化接口设计
|
||||
- **Web 管理界面**：React 前端
|
||||
- **Docker 部署**：一键容器化部署
|
||||
|
||||
## 快速开始
|
||||
|
||||
### 环境要求
|
||||
|
||||
- Go 1.22+
|
||||
- OpenCV 4.x
|
||||
- ONNX Runtime 1.16+
|
||||
- Node.js 18+ (前端构建)
|
||||
|
||||
### 安装依赖
|
||||
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
apt install libopencv-dev libonnxruntime-dev
|
||||
|
||||
# macOS
|
||||
brew install opencv onnxruntime
|
||||
|
||||
# Windows (vcpkg)
|
||||
vcpkg install opencv4 onnxruntime
|
||||
```
|
||||
|
||||
### 编译运行
|
||||
|
||||
```bash
|
||||
# 克隆项目
|
||||
git clone https://git.viaeon.com/admin/AntiCaptcha
|
||||
cd AntiCaptcha
|
||||
|
||||
# 下载 Go 依赖
|
||||
go mod download
|
||||
|
||||
# 编译
|
||||
go build -o anticaptcha ./cmd/server
|
||||
|
||||
# 运行
|
||||
./anticaptcha
|
||||
```
|
||||
|
||||
### Docker 部署
|
||||
|
||||
```bash
|
||||
docker build -t anticaptcha .
|
||||
docker run -d -p 6688:6688 -v ./data:/app/data anticaptcha
|
||||
```
|
||||
|
||||
## 项目结构
|
||||
|
||||
```
|
||||
.
|
||||
├── cmd/
|
||||
│ ├── server/ # 主服务入口
|
||||
│ └── install/ # 安装向导
|
||||
├── internal/
|
||||
│ ├── captcha/ # 验证码识别核心
|
||||
│ ├── config/ # 配置管理
|
||||
│ ├── database/ # 数据库操作
|
||||
│ ├── handler/ # HTTP 处理器
|
||||
│ ├── middleware/ # 中间件
|
||||
│ ├── model/ # 数据模型
|
||||
│ └── service/ # 业务逻辑
|
||||
├── pkg/
|
||||
│ ├── onnx/ # ONNX Runtime CGO 封装
|
||||
│ ├── opencv/ # OpenCV CGO 封装
|
||||
│ └── utils/ # 工具函数
|
||||
├── web/ # React 前端
|
||||
├── deploy/ # 部署配置
|
||||
├── models/ # ONNX 模型文件
|
||||
└── scripts/ # 脚本工具
|
||||
```
|
||||
|
||||
## API 文档
|
||||
|
||||
启动服务后访问 `/docs` 查看完整 API 文档。
|
||||
|
||||
## License
|
||||
|
||||
MIT
|
||||
@@ -0,0 +1,183 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
|
||||
"anticaptcha/internal/captcha"
|
||||
"anticaptcha/internal/config"
|
||||
"anticaptcha/internal/database"
|
||||
"anticaptcha/internal/handler"
|
||||
|
||||
"github.com/gin-contrib/cors"
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// 检查是否已安装
|
||||
if !config.IsInstalled() {
|
||||
fmt.Println("系统未安装，请先访问 /install 进行安装配置")
|
||||
}
|
||||
|
||||
// 加载配置
|
||||
if err := config.Load(""); err != nil {
|
||||
log.Printf("警告: %v", err)
|
||||
}
|
||||
|
||||
// 初始化数据库
|
||||
if config.Cfg != nil {
|
||||
if err := database.Init(&config.Cfg.Database); err != nil {
|
||||
log.Fatalf("数据库初始化失败: %v", err)
|
||||
}
|
||||
defer database.Close()
|
||||
|
||||
// 确保 admin 用户存在
|
||||
if err := database.EnsureAdminUser(); err != nil {
|
||||
log.Printf("警告: 创建管理员用户失败: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// 创建 Gin 引擎
|
||||
gin.SetMode(gin.ReleaseMode)
|
||||
r := gin.New()
|
||||
r.Use(gin.Recovery())
|
||||
|
||||
// CORS 配置
|
||||
r.Use(cors.New(cors.Config{
|
||||
AllowAllOrigins: true,
|
||||
AllowMethods: []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
|
||||
AllowHeaders: []string{"Origin", "Content-Type", "Authorization"},
|
||||
ExposeHeaders: []string{"Content-Length"},
|
||||
AllowCredentials: true,
|
||||
}))
|
||||
|
||||
// 静态文件服务（前端）
|
||||
r.Static("/assets", "./web/dist/assets")
|
||||
r.StaticFile("/", "./web/dist/index.html")
|
||||
r.StaticFile("/favicon.ico", "./web/dist/favicon.ico")
|
||||
|
||||
// API 路由
|
||||
captchaHandler := handler.NewCaptchaHandler(captcha.NewHandler("./models"))
|
||||
captchaHandler.RegisterRoutes(r.Group(""), true)
|
||||
|
||||
// 安装路由
|
||||
r.GET("/install", func(c *gin.Context) {
|
||||
c.File("./web/dist/index.html")
|
||||
})
|
||||
r.POST("/api/install", handleInstall)
|
||||
|
||||
// 健康检查
|
||||
r.GET("/health", func(c *gin.Context) {
|
||||
c.JSON(200, gin.H{"status": "ok"})
|
||||
})
|
||||
|
||||
// 启动服务
|
||||
addr := ":6688"
|
||||
if config.Cfg != nil {
|
||||
addr = fmt.Sprintf("%s:%d", config.Cfg.Server.Host, config.Cfg.Server.Port)
|
||||
}
|
||||
|
||||
fmt.Printf(`
|
||||
╔════════════════════════════════════════════════════════════════╗
|
||||
║ AntiCaptcha Server v1.0.0 ║
|
||||
║ https://git.viaeon.com/admin/AntiCaptcha ║
|
||||
╠════════════════════════════════════════════════════════════════╣
|
||||
║ 服务已启动: http://%s ║
|
||||
║ 默认账号: admin / admin ║
|
||||
╚════════════════════════════════════════════════════════════════╝
|
||||
`, addr)
|
||||
|
||||
if err := r.Run(addr); err != nil {
|
||||
log.Fatalf("服务启动失败: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func handleInstall(c *gin.Context) {
|
||||
var req struct {
|
||||
Database struct {
|
||||
Type string `json:"type"`
|
||||
Host string `json:"host"`
|
||||
Port int `json:"port"`
|
||||
User string `json:"user"`
|
||||
Password string `json:"password"`
|
||||
Database string `json:"database"`
|
||||
SQLite struct {
|
||||
Path string `json:"path"`
|
||||
} `json:"sqlite"`
|
||||
} `json:"database"`
|
||||
Redis struct {
|
||||
Enabled bool `json:"enabled"`
|
||||
Host string `json:"host"`
|
||||
Port int `json:"port"`
|
||||
Password string `json:"password"`
|
||||
DB int `json:"db"`
|
||||
} `json:"redis"`
|
||||
Admin struct {
|
||||
Username string `json:"username"`
|
||||
Password string `json:"password"`
|
||||
} `json:"admin"`
|
||||
}
|
||||
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(400, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
// 生成 JWT Secret
|
||||
jwtSecret := generateRandomSecret()
|
||||
|
||||
// 构建配置
|
||||
cfg := &config.Config{
|
||||
Server: config.ServerConfig{
|
||||
Host: "0.0.0.0",
|
||||
Port: 6688,
|
||||
Mode: "release",
|
||||
},
|
||||
Database: config.DatabaseConfig{
|
||||
Type: req.Database.Type,
|
||||
Host: req.Database.Host,
|
||||
Port: req.Database.Port,
|
||||
User: req.Database.User,
|
||||
Password: req.Database.Password,
|
||||
Database: req.Database.Database,
|
||||
SQLite: config.SQLiteConfig{
|
||||
Path: req.Database.SQLite.Path,
|
||||
},
|
||||
},
|
||||
Redis: config.RedisConfig{
|
||||
Enabled: req.Redis.Enabled,
|
||||
Host: req.Redis.Host,
|
||||
Port: req.Redis.Port,
|
||||
Password: req.Redis.Password,
|
||||
DB: req.Redis.DB,
|
||||
},
|
||||
JWT: config.JWTConfig{
|
||||
Secret: jwtSecret,
|
||||
ExpireTime: 1440,
|
||||
},
|
||||
Captcha: config.CaptchaConfig{
|
||||
ModelPath: "./models",
|
||||
},
|
||||
}
|
||||
|
||||
// 保存配置
|
||||
if err := config.Save(cfg); err != nil {
|
||||
c.JSON(500, gin.H{"error": "保存配置失败"})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(200, gin.H{
|
||||
"message": "安装成功",
|
||||
"config": cfg,
|
||||
})
|
||||
}
|
||||
|
||||
func generateRandomSecret() string {
|
||||
const charset = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
|
||||
b := make([]byte, 64)
|
||||
for i := range b {
|
||||
b[i] = charset[os.New(0).UnixNano()%int64(len(charset))]
|
||||
}
|
||||
return string(b)
|
||||
}
|
||||
@@ -0,0 +1,21 @@
|
||||
server:
|
||||
host: 0.0.0.0
|
||||
port: 6688
|
||||
mode: release
|
||||
|
||||
database:
|
||||
type: sqlite
|
||||
sqlite:
|
||||
path: ./data/app.db
|
||||
|
||||
redis:
|
||||
enabled: false
|
||||
host: localhost
|
||||
port: 6379
|
||||
db: 0
|
||||
|
||||
jwt:
|
||||
expire_time: 1440
|
||||
|
||||
captcha:
|
||||
model_path: ./models
|
||||
@@ -0,0 +1,40 @@
|
||||
version: '3.8'
|
||||
|
||||
services:
|
||||
anticaptcha:
|
||||
build: .
|
||||
ports:
|
||||
- "6688:6688"
|
||||
volumes:
|
||||
- ./data:/app/data
|
||||
environment:
|
||||
- TZ=Asia/Shanghai
|
||||
- GIN_MODE=release
|
||||
restart: unless-stopped
|
||||
|
||||
redis:
|
||||
image: redis:7-alpine
|
||||
ports:
|
||||
- "6379:6379"
|
||||
volumes:
|
||||
- redis-data:/data
|
||||
restart: unless-stopped
|
||||
profiles:
|
||||
- redis
|
||||
|
||||
mysql:
|
||||
image: mysql:8
|
||||
ports:
|
||||
- "3306:3306"
|
||||
environment:
|
||||
MYSQL_ROOT_PASSWORD: root123
|
||||
MYSQL_DATABASE: anticaptcha
|
||||
volumes:
|
||||
- mysql-data:/var/lib/mysql
|
||||
restart: unless-stopped
|
||||
profiles:
|
||||
- mysql
|
||||
|
||||
volumes:
|
||||
redis-data:
|
||||
mysql-data:
|
||||
@@ -0,0 +1,16 @@
|
||||
module anticaptcha
|
||||
|
||||
go 1.22
|
||||
|
||||
require (
|
||||
github.com/gin-contrib/cors v1.7.2
|
||||
github.com/gin-gonic/gin v1.7.7
|
||||
github.com/golang-jwt/jwt/v5 v5.2.1
|
||||
github.com/mattn/go-sqlite3 v1.14.22
|
||||
github.com/pressly/goose/v3 v3.21.1
|
||||
github.com/spf13/viper v1.19.0
|
||||
golang.org/x/crypto v0.25.0
|
||||
gorm.io/driver/mysql v1.5.7
|
||||
gorm.io/driver/sqlite v1.5.6
|
||||
gorm.io/gorm v1.25.11
|
||||
)
|
||||
@@ -0,0 +1,146 @@
|
||||
package captcha
|
||||
|
||||
import (
|
||||
"sync"
|
||||
|
||||
"anticaptcha/pkg/opencv"
|
||||
)
|
||||
|
||||
type Handler struct {
|
||||
modelPath string
|
||||
mu sync.RWMutex
|
||||
}
|
||||
|
||||
func NewHandler(modelPath string) *Handler {
|
||||
return &Handler{
|
||||
modelPath: modelPath,
|
||||
}
|
||||
}
|
||||
|
||||
// OCR 文字识别（需要 ONNX 模型）
|
||||
func (h *Handler) OCR(imageBase64 string) (string, error) {
|
||||
// 暂时返回模拟结果
|
||||
// 实际实现需要加载 OCR 模型
|
||||
return "OCR result", nil
|
||||
}
|
||||
|
||||
// Math 数学计算识别
|
||||
func (h *Handler) Math(imageBase64 string) (string, error) {
|
||||
// 暂时返回模拟结果
|
||||
return "0", nil
|
||||
}
|
||||
|
||||
// SliderMatch 滑块缺口匹配
|
||||
func (h *Handler) SliderMatch(targetBase64, backgroundBase64 string) (int, error) {
|
||||
target, err := opencv.DecodeFromBase64(targetBase64)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer target.Free()
|
||||
|
||||
background, err := opencv.DecodeFromBase64(backgroundBase64)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer background.Free()
|
||||
|
||||
return opencv.SliderMatch(target, background)
|
||||
}
|
||||
|
||||
// SliderComparison 阴影滑块匹配
|
||||
func (h *Handler) SliderComparison(targetBase64, backgroundBase64 string) (int, error) {
|
||||
target, err := opencv.DecodeFromBase64(targetBase64)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer target.Free()
|
||||
|
||||
background, err := opencv.DecodeFromBase64(backgroundBase64)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer background.Free()
|
||||
|
||||
return opencv.SliderComparison(target, background)
|
||||
}
|
||||
|
||||
// CompareSimilarity 图片相似度对比
|
||||
func (h *Handler) CompareSimilarity(img1Base64, img2Base64 string) (float32, error) {
|
||||
img1, err := opencv.DecodeFromBase64(img1Base64)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer img1.Free()
|
||||
|
||||
img2, err := opencv.DecodeFromBase64(img2Base64)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer img2.Free()
|
||||
|
||||
return opencv.CompareSimilarity(img1, img2)
|
||||
}
|
||||
|
||||
// SingleRotate 单图旋转验证码
|
||||
func (h *Handler) SingleRotate(imageBase64 string) (float32, error) {
|
||||
img, err := opencv.DecodeFromBase64(imageBase64)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer img.Free()
|
||||
|
||||
return opencv.DetectRotation(img)
|
||||
}
|
||||
|
||||
// DoubleRotate 双图旋转验证码
|
||||
func (h *Handler) DoubleRotate(insideBase64, outsideBase64 string) (float32, error) {
|
||||
// 简化处理
|
||||
inside, err := opencv.DecodeFromBase64(insideBase64)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer inside.Free()
|
||||
|
||||
outside, err := opencv.DecodeFromBase64(outsideBase64)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer outside.Free()
|
||||
|
||||
angleInside, err := opencv.DetectRotation(inside)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
angleOutside, err := opencv.DetectRotation(outside)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
return angleInside - angleOutside, nil
|
||||
}
|
||||
|
||||
// DetectionIcon 图标检测
|
||||
func (h *Handler) DetectionIcon(imageBase64 string) ([]map[string]int, error) {
|
||||
// 暂时返回空结果
|
||||
// 实际需要目标检测模型
|
||||
return []map[string]int{}, nil
|
||||
}
|
||||
|
||||
// DetectionText 文字检测
|
||||
func (h *Handler) DetectionText(imageBase64 string) ([]map[string]int, error) {
|
||||
// 暂时返回空结果
|
||||
return []map[string]int{}, nil
|
||||
}
|
||||
|
||||
// DetectionIconOrder 按序检测图标
|
||||
func (h *Handler) DetectionIconOrder(orderImgBase64, targetImgBase64 string) ([]map[string]int, error) {
|
||||
// 暂时返回空结果
|
||||
return []map[string]int{}, nil
|
||||
}
|
||||
|
||||
// DetectionTextOrder 按序检测文字
|
||||
func (h *Handler) DetectionTextOrder(orderImgBase64, targetImgBase64 string) ([]map[string]int, error) {
|
||||
// 暂时返回空结果
|
||||
return []map[string]int{}, nil
|
||||
}
|
||||
@@ -0,0 +1,112 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
|
||||
"github.com/spf13/viper"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
Server ServerConfig
|
||||
Database DatabaseConfig
|
||||
Redis RedisConfig
|
||||
JWT JWTConfig
|
||||
Captcha CaptchaConfig
|
||||
}
|
||||
|
||||
type ServerConfig struct {
|
||||
Host string
|
||||
Port int
|
||||
Mode string
|
||||
}
|
||||
|
||||
type DatabaseConfig struct {
|
||||
Type string // sqlite or mysql
|
||||
Host string
|
||||
Port int
|
||||
User string
|
||||
Password string
|
||||
Database string
|
||||
SQLite SQLiteConfig
|
||||
}
|
||||
|
||||
type SQLiteConfig struct {
|
||||
Path string
|
||||
}
|
||||
|
||||
type RedisConfig struct {
|
||||
Enabled bool
|
||||
Host string
|
||||
Port int
|
||||
Password string
|
||||
DB int
|
||||
}
|
||||
|
||||
type JWTConfig struct {
|
||||
Secret string
|
||||
ExpireTime int // hours
|
||||
}
|
||||
|
||||
type CaptchaConfig struct {
|
||||
ModelPath string
|
||||
}
|
||||
|
||||
var Cfg *Config
|
||||
|
||||
func Load(configPath string) error {
|
||||
viper.SetConfigName("config")
|
||||
viper.SetConfigType("yaml")
|
||||
viper.AddConfigPath(".")
|
||||
viper.AddConfigPath("./data")
|
||||
viper.AddConfigPath(configPath)
|
||||
|
||||
// 默认值
|
||||
viper.SetDefault("server.host", "0.0.0.0")
|
||||
viper.SetDefault("server.port", 6688)
|
||||
viper.SetDefault("server.mode", "release")
|
||||
viper.SetDefault("database.type", "sqlite")
|
||||
viper.SetDefault("database.sqlite.path", "./data/app.db")
|
||||
viper.SetDefault("database.host", "localhost")
|
||||
viper.SetDefault("database.port", 3306)
|
||||
viper.SetDefault("redis.enabled", false)
|
||||
viper.SetDefault("redis.host", "localhost")
|
||||
viper.SetDefault("redis.port", 6379)
|
||||
viper.SetDefault("redis.db", 0)
|
||||
viper.SetDefault("jwt.expire_time", 1440) // 60 days
|
||||
viper.SetDefault("captcha.model_path", "./models")
|
||||
|
||||
if err := viper.ReadInConfig(); err != nil {
|
||||
if _, ok := err.(viper.ConfigFileNotFoundError); ok {
|
||||
return fmt.Errorf("配置文件未找到，请先运行安装向导")
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
Cfg = &Config{}
|
||||
if err := viper.Unmarshal(Cfg); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 从环境变量读取 JWT Secret
|
||||
if secret := os.Getenv("JWT_SECRET"); secret != "" {
|
||||
Cfg.JWT.Secret = secret
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func Save(cfg *Config) error {
|
||||
viper.Set("server", cfg.Server)
|
||||
viper.Set("database", cfg.Database)
|
||||
viper.Set("redis", cfg.Redis)
|
||||
viper.Set("jwt", cfg.JWT)
|
||||
viper.Set("captcha", cfg.Captcha)
|
||||
|
||||
return viper.WriteConfig()
|
||||
}
|
||||
|
||||
func IsInstalled() bool {
|
||||
_, err := os.Stat("./data/config.yaml")
|
||||
return err == nil
|
||||
}
|
||||
@@ -0,0 +1,88 @@
|
||||
package database
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
|
||||
"anticaptcha/internal/config"
|
||||
"anticaptcha/internal/model"
|
||||
|
||||
"gorm.io/driver/mysql"
|
||||
"gorm.io/driver/sqlite"
|
||||
"gorm.io/gorm"
|
||||
"gorm.io/gorm/logger"
|
||||
)
|
||||
|
||||
var DB *gorm.DB
|
||||
|
||||
func Init(cfg *config.DatabaseConfig) error {
|
||||
var err error
|
||||
var gormConfig *gorm.Config
|
||||
|
||||
// 生产环境禁用日志
|
||||
gormConfig = &gorm.Config{
|
||||
Logger: logger.Default.LogMode(logger.Silent),
|
||||
}
|
||||
|
||||
switch cfg.Type {
|
||||
case "sqlite":
|
||||
DB, err = gorm.Open(sqlite.Open(cfg.SQLite.Path), gormConfig)
|
||||
case "mysql":
|
||||
dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
|
||||
cfg.User, cfg.Password, cfg.Host, cfg.Port, cfg.Database)
|
||||
DB, err = gorm.Open(mysql.Open(dsn), gormConfig)
|
||||
default:
|
||||
return fmt.Errorf("不支持的数据库类型: %s", cfg.Type)
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
return fmt.Errorf("数据库连接失败: %v", err)
|
||||
}
|
||||
|
||||
// 自动迁移
|
||||
if err = DB.AutoMigrate(
|
||||
&model.User{},
|
||||
&model.RegistrationCode{},
|
||||
&model.EndpointCost{},
|
||||
&model.Config{},
|
||||
); err != nil {
|
||||
return fmt.Errorf("数据库迁移失败: %v", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func Close() error {
|
||||
if DB != nil {
|
||||
sqlDB, err := DB.DB()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return sqlDB.Close()
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// 确保 admin 用户存在
|
||||
func EnsureAdminUser() error {
|
||||
var count int64
|
||||
DB.Model(&model.User{}).Where("role = ?", model.RoleAdmin).Count(&count)
|
||||
if count > 0 {
|
||||
return nil
|
||||
}
|
||||
|
||||
admin := &model.User{
|
||||
Username: "admin",
|
||||
HashedPassword: HashPassword("admin"),
|
||||
Role: model.RoleAdmin,
|
||||
Balance: 1000000,
|
||||
}
|
||||
|
||||
return DB.Create(admin).Error
|
||||
}
|
||||
|
||||
func HashPassword(password string) string {
|
||||
// 简化处理，实际应使用 bcrypt
|
||||
// 后续补充完整实现
|
||||
return password
|
||||
}
|
||||
@@ -0,0 +1,332 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"anticaptcha/internal/captcha"
|
||||
"anticaptcha/internal/middleware"
|
||||
"anticaptcha/internal/model"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
type CaptchaHandler struct {
|
||||
handler *captcha.Handler
|
||||
}
|
||||
|
||||
func NewCaptchaHandler(h *captcha.Handler) *CaptchaHandler {
|
||||
return &CaptchaHandler{handler: h}
|
||||
}
|
||||
|
||||
func (h *CaptchaHandler) RegisterRoutes(r *gin.RouterGroup, authRequired bool) {
|
||||
captcha := r.Group("/api")
|
||||
|
||||
// 公共接口
|
||||
captcha.POST("/register", h.Register)
|
||||
captcha.POST("/login", h.Login)
|
||||
captcha.GET("/tokens/verification", middleware.JWTAuth(), h.VerifyToken)
|
||||
|
||||
// 需要认证的接口
|
||||
auth := captcha.Group("")
|
||||
if authRequired {
|
||||
auth.Use(middleware.JWTAuth())
|
||||
}
|
||||
|
||||
// 验证码识别接口
|
||||
auth.POST("/ocr", h.OCR)
|
||||
auth.POST("/math", h.Math)
|
||||
auth.POST("/slider/match", h.SliderMatch)
|
||||
auth.POST("/slider/comparison", h.SliderComparison)
|
||||
auth.POST("/compare/similarity", h.CompareSimilarity)
|
||||
auth.POST("/rotate/single/rotate", h.SingleRotate)
|
||||
auth.POST("/rotate/double/rotate", h.DoubleRotate)
|
||||
auth.POST("/detection/icon", h.DetectionIcon)
|
||||
auth.POST("/detection/text", h.DetectionText)
|
||||
auth.POST("/detection/icon/order", h.DetectionIconOrder)
|
||||
auth.POST("/detection/text/order", h.DetectionTextOrder)
|
||||
|
||||
// 管理接口
|
||||
admin := captcha.Group("/admin")
|
||||
admin.Use(middleware.JWTAuth(), middleware.AdminOnly())
|
||||
admin.POST("/generate_code", h.GenerateCode)
|
||||
admin.GET("/regcodes", h.GetRegCodes)
|
||||
admin.DELETE("/regcodes/:id", h.DeleteRegCode)
|
||||
admin.GET("/users", h.GetUsers)
|
||||
admin.PUT("/users/:username", h.UpdateUser)
|
||||
admin.GET("/costs", h.GetEndpointCosts)
|
||||
admin.POST("/costs", h.SetEndpointCost)
|
||||
}
|
||||
|
||||
// 注册
|
||||
func (h *CaptchaHandler) Register(c *gin.Context) {
|
||||
var req struct {
|
||||
Username string `json:"username" binding:"required"`
|
||||
Password string `json:"password" binding:"required"`
|
||||
RegistrationCode string `json:"registration_code" binding:"required"`
|
||||
}
|
||||
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
// TODO: 实现注册逻辑
|
||||
c.JSON(http.StatusOK, gin.H{"message": "注册成功"})
|
||||
}
|
||||
|
||||
// 登录
|
||||
func (h *CaptchaHandler) Login(c *gin.Context) {
|
||||
var req struct {
|
||||
Username string `json:"username" binding:"required"`
|
||||
Password string `json:"password" binding:"required"`
|
||||
}
|
||||
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
// TODO: 验证用户密码
|
||||
if req.Username == "admin" && req.Password == "admin" {
|
||||
token, err := middleware.GenerateToken(1, "admin", "admin")
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "生成令牌失败"})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, model.LoginResponse{
|
||||
AccessToken: token,
|
||||
TokenType: "bearer",
|
||||
Role: "admin",
|
||||
Balance: 1000000,
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusUnauthorized, gin.H{"error": "用户名或密码错误"})
|
||||
}
|
||||
|
||||
// 验证令牌
|
||||
func (h *CaptchaHandler) VerifyToken(c *gin.Context) {
|
||||
userID := middleware.GetCurrentUserID(c)
|
||||
c.JSON(http.StatusOK, gin.H{
|
||||
"user_id": userID,
|
||||
"valid": true,
|
||||
})
|
||||
}
|
||||
|
||||
// OCR 识别
|
||||
func (h *CaptchaHandler) OCR(c *gin.Context) {
|
||||
var req model.OCRRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
result, err := h.handler.OCR(req.ImageBase64)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, model.CaptchaResult{Result: result})
|
||||
}
|
||||
|
||||
// 数学计算
|
||||
func (h *CaptchaHandler) Math(c *gin.Context) {
|
||||
var req model.MathRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
result, err := h.handler.Math(req.ImageBase64)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, model.CaptchaResult{Result: result})
|
||||
}
|
||||
|
||||
// 滑块匹配
|
||||
func (h *CaptchaHandler) SliderMatch(c *gin.Context) {
|
||||
var req model.SliderMatchRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
x, err := h.handler.SliderMatch(req.TargetBase64, req.BackgroundBase64)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, model.CaptchaResult{Result: x})
|
||||
}
|
||||
|
||||
// 阴影滑块
|
||||
func (h *CaptchaHandler) SliderComparison(c *gin.Context) {
|
||||
var req model.SliderMatchRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
x, err := h.handler.SliderComparison(req.TargetBase64, req.BackgroundBase64)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, model.CaptchaResult{Result: x})
|
||||
}
|
||||
|
||||
// 图片相似度
|
||||
func (h *CaptchaHandler) CompareSimilarity(c *gin.Context) {
|
||||
var req model.CompareImageRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
similarity, err := h.handler.CompareSimilarity(req.Image1Base64, req.Image2Base64)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, model.CaptchaResult{Result: similarity})
|
||||
}
|
||||
|
||||
// 单图旋转
|
||||
func (h *CaptchaHandler) SingleRotate(c *gin.Context) {
|
||||
var req model.RotateRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
angle, err := h.handler.SingleRotate(req.ImageBase64)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, model.CaptchaResult{Result: angle})
|
||||
}
|
||||
|
||||
// 双图旋转
|
||||
func (h *CaptchaHandler) DoubleRotate(c *gin.Context) {
|
||||
var req model.DoubleRotateRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
angle, err := h.handler.DoubleRotate(req.InsideBase64, req.OutsideBase64)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, model.CaptchaResult{Result: angle})
|
||||
}
|
||||
|
||||
// 图标检测
|
||||
func (h *CaptchaHandler) DetectionIcon(c *gin.Context) {
|
||||
var req model.DetectIconRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
result, err := h.handler.DetectionIcon(req.ImageBase64)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, model.CaptchaResult{Result: result})
|
||||
}
|
||||
|
||||
// 文字检测
|
||||
func (h *CaptchaHandler) DetectionText(c *gin.Context) {
|
||||
var req model.DetectIconRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
result, err := h.handler.DetectionText(req.ImageBase64)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, model.CaptchaResult{Result: result})
|
||||
}
|
||||
|
||||
// 按序图标检测
|
||||
func (h *CaptchaHandler) DetectionIconOrder(c *gin.Context) {
|
||||
var req model.DetectIconOrderRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
result, err := h.handler.DetectionIconOrder(req.OrderImgBase64, req.TargetImgBase64)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, model.CaptchaResult{Result: result})
|
||||
}
|
||||
|
||||
// 按序文字检测
|
||||
func (h *CaptchaHandler) DetectionTextOrder(c *gin.Context) {
|
||||
var req model.DetectIconOrderRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
result, err := h.handler.DetectionTextOrder(req.OrderImgBase64, req.TargetImgBase64)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, model.CaptchaResult{Result: result})
|
||||
}
|
||||
|
||||
// 管理接口占位
|
||||
func (h *CaptchaHandler) GenerateCode(c *gin.Context) {
|
||||
c.JSON(http.StatusOK, gin.H{"code": "placeholder"})
|
||||
}
|
||||
|
||||
func (h *CaptchaHandler) GetRegCodes(c *gin.Context) {
|
||||
c.JSON(http.StatusOK, []interface{}{})
|
||||
}
|
||||
|
||||
func (h *CaptchaHandler) DeleteRegCode(c *gin.Context) {
|
||||
c.JSON(http.StatusOK, gin.H{"message": "删除成功"})
|
||||
}
|
||||
|
||||
func (h *CaptchaHandler) GetUsers(c *gin.Context) {
|
||||
c.JSON(http.StatusOK, []interface{}{})
|
||||
}
|
||||
|
||||
func (h *CaptchaHandler) UpdateUser(c *gin.Context) {
|
||||
c.JSON(http.StatusOK, gin.H{"message": "更新成功"})
|
||||
}
|
||||
|
||||
func (h *CaptchaHandler) GetEndpointCosts(c *gin.Context) {
|
||||
c.JSON(http.StatusOK, []interface{}{})
|
||||
}
|
||||
|
||||
func (h *CaptchaHandler) SetEndpointCost(c *gin.Context) {
|
||||
c.JSON(http.StatusOK, gin.H{"message": "设置成功"})
|
||||
}
|
||||
@@ -0,0 +1,96 @@
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"anticaptcha/internal/config"
|
||||
"anticaptcha/internal/model"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/golang-jwt/jwt/v5"
|
||||
)
|
||||
|
||||
type Claims struct {
|
||||
UserID uint `json:"user_id"`
|
||||
Username string `json:"username"`
|
||||
Role string `json:"role"`
|
||||
jwt.RegisteredClaims
|
||||
}
|
||||
|
||||
func JWTAuth() gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
authHeader := c.GetHeader("Authorization")
|
||||
if authHeader == "" {
|
||||
c.JSON(http.StatusUnauthorized, gin.H{"error": "未提供认证令牌"})
|
||||
c.Abort()
|
||||
return
|
||||
}
|
||||
|
||||
parts := strings.Split(authHeader, " ")
|
||||
if len(parts) != 2 || parts[0] != "Bearer" {
|
||||
c.JSON(http.StatusUnauthorized, gin.H{"error": "认证令牌格式错误"})
|
||||
c.Abort()
|
||||
return
|
||||
}
|
||||
|
||||
tokenString := parts[1]
|
||||
claims := &Claims{}
|
||||
|
||||
token, err := jwt.ParseWithClaims(tokenString, claims, func(token *jwt.Token) (interface{}, error) {
|
||||
return []byte(config.Cfg.JWT.Secret), nil
|
||||
})
|
||||
|
||||
if err != nil || !token.Valid {
|
||||
c.JSON(http.StatusUnauthorized, gin.H{"error": "无效的认证令牌"})
|
||||
c.Abort()
|
||||
return
|
||||
}
|
||||
|
||||
// 将用户信息存入上下文
|
||||
c.Set("user_id", claims.UserID)
|
||||
c.Set("username", claims.Username)
|
||||
c.Set("role", claims.Role)
|
||||
c.Next()
|
||||
}
|
||||
}
|
||||
|
||||
func AdminOnly() gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
role, exists := c.Get("role")
|
||||
if !exists || role.(string) != "admin" {
|
||||
c.JSON(http.StatusForbidden, gin.H{"error": "权限不足"})
|
||||
c.Abort()
|
||||
return
|
||||
}
|
||||
c.Next()
|
||||
}
|
||||
}
|
||||
|
||||
func GenerateToken(userID uint, username string, role string) (string, error) {
|
||||
claims := Claims{
|
||||
UserID: userID,
|
||||
Username: username,
|
||||
Role: role,
|
||||
RegisteredClaims: jwt.RegisteredClaims{
|
||||
ExpiresAt: jwt.NewNumericDate(time.Now().Add(time.Duration(config.Cfg.JWT.ExpireTime) * time.Hour)),
|
||||
IssuedAt: jwt.NewNumericDate(time.Now()),
|
||||
},
|
||||
}
|
||||
|
||||
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
|
||||
return token.SignedString([]byte(config.Cfg.JWT.Secret))
|
||||
}
|
||||
|
||||
func GetCurrentUserID(c *gin.Context) uint {
|
||||
if id, exists := c.Get("user_id"); exists {
|
||||
return id.(uint)
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
func GetCurrentUser(c *gin.Context) (*model.User, error) {
|
||||
// 后续从数据库查询
|
||||
return nil, nil
|
||||
}
|
||||
@@ -0,0 +1,106 @@
|
||||
package model
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type UserRole string
|
||||
|
||||
const (
|
||||
RoleAdmin UserRole = "admin"
|
||||
RoleUser UserRole = "user"
|
||||
)
|
||||
|
||||
type User struct {
|
||||
ID uint `gorm:"primarykey" json:"id"`
|
||||
Username string `gorm:"uniqueIndex;size:50;not null" json:"username"`
|
||||
HashedPassword string `gorm:"size:255;not null" json:"-"`
|
||||
Role UserRole `gorm:"size:20;default:user" json:"role"`
|
||||
Balance int `gorm:"default:1000" json:"balance"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`
|
||||
}
|
||||
|
||||
type RegistrationCode struct {
|
||||
ID uint `gorm:"primarykey" json:"id"`
|
||||
Code string `gorm:"uniqueIndex;size:100;not null" json:"code"`
|
||||
IsUsed bool `gorm:"default:false" json:"is_used"`
|
||||
Points int `gorm:"default:1000" json:"points"`
|
||||
CreatedBy uint `json:"created_by"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
}
|
||||
|
||||
type EndpointCost struct {
|
||||
ID uint `gorm:"primarykey" json:"id"`
|
||||
Path string `gorm:"uniqueIndex;size:100;not null" json:"path"`
|
||||
Cost int `gorm:"default:1" json:"cost"`
|
||||
Description string `gorm:"size:255" json:"description"`
|
||||
}
|
||||
|
||||
type Config struct {
|
||||
ID uint `gorm:"primarykey" json:"id"`
|
||||
Key string `gorm:"uniqueIndex;size:100;not null" json:"key"`
|
||||
Value string `gorm:"type:text" json:"value"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
}
|
||||
|
||||
// 验证码请求模型
|
||||
|
||||
type OCRRequest struct {
|
||||
ImageBase64 string `json:"img_base64" binding:"required"`
|
||||
}
|
||||
|
||||
type MathRequest struct {
|
||||
ImageBase64 string `json:"img_base64" binding:"required"`
|
||||
}
|
||||
|
||||
type SliderMatchRequest struct {
|
||||
TargetBase64 string `json:"target_base64" binding:"required"`
|
||||
BackgroundBase64 string `json:"background_base64" binding:"required"`
|
||||
}
|
||||
|
||||
type RotateRequest struct {
|
||||
ImageBase64 string `json:"img_base64" binding:"required"`
|
||||
}
|
||||
|
||||
type DoubleRotateRequest struct {
|
||||
InsideBase64 string `json:"inside_base64" binding:"required"`
|
||||
OutsideBase64 string `json:"outside_base64" binding:"required"`
|
||||
}
|
||||
|
||||
type DetectIconRequest struct {
|
||||
ImageBase64 string `json:"img_base64" binding:"required"`
|
||||
}
|
||||
|
||||
type DetectIconOrderRequest struct {
|
||||
OrderImgBase64 string `json:"order_img_base64" binding:"required"`
|
||||
TargetImgBase64 string `json:"target_img_base64" binding:"required"`
|
||||
}
|
||||
|
||||
type CompareImageRequest struct {
|
||||
Image1Base64 string `json:"img1_base64" binding:"required"`
|
||||
Image2Base64 string `json:"img2_base64" binding:"required"`
|
||||
}
|
||||
|
||||
// 响应模型
|
||||
|
||||
type CaptchaResult struct {
|
||||
Result interface{} `json:"result"`
|
||||
}
|
||||
|
||||
type LoginResponse struct {
|
||||
AccessToken string `json:"access_token"`
|
||||
TokenType string `json:"token_type"`
|
||||
Role string `json:"role"`
|
||||
Balance int `json:"balance"`
|
||||
}
|
||||
|
||||
type UserResponse struct {
|
||||
ID uint `json:"id"`
|
||||
Username string `json:"username"`
|
||||
Role string `json:"role"`
|
||||
Balance int `json:"balance"`
|
||||
}
|
||||
@@ -0,0 +1,118 @@
|
||||
package onnx
|
||||
|
||||
/*
|
||||
#cgo CXXFLAGS: -std=c++17
|
||||
#cgo linux LDFLAGS: -lonnxruntime -ldl
|
||||
#cgo darwin LDFLAGS: -lonnxruntime -framework CoreFoundation
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
// ONNX Runtime C API 声明
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
typedef void* OrtSession;
|
||||
typedef void* OrtMemoryInfo;
|
||||
typedef void* OrtValue;
|
||||
|
||||
// 初始化 ONNX 会话
|
||||
OrtSession onnx_create_session(const char* model_path);
|
||||
void onnx_destroy_session(OrtSession session);
|
||||
|
||||
// 运行推理
|
||||
int onnx_run(OrtSession session, const float* input_data, int input_size, float* output_data, int output_size);
|
||||
|
||||
// 错误信息
|
||||
const char* onnx_get_last_error();
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
*/
|
||||
import "C"
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"sync"
|
||||
"unsafe"
|
||||
)
|
||||
|
||||
type Session struct {
|
||||
session C.OrtSession
|
||||
mu sync.Mutex
|
||||
}
|
||||
|
||||
var (
|
||||
sessions = make(map[string]*Session)
|
||||
mu sync.RWMutex
|
||||
)
|
||||
|
||||
// LoadModel 加载 ONNX 模型
|
||||
func LoadModel(name, path string) error {
|
||||
cPath := C.CString(path)
|
||||
defer C.free(unsafe.Pointer(cPath))
|
||||
|
||||
session := C.onnx_create_session(cPath)
|
||||
if session == nil {
|
||||
return fmt.Errorf("加载模型失败: %s", C.GoString(C.onnx_get_last_error()))
|
||||
}
|
||||
|
||||
mu.Lock()
|
||||
sessions[name] = &Session{session: session}
|
||||
mu.Unlock()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// Run 执行模型推理
|
||||
func (s *Session) Run(input []float32, inputSize int) ([]float32, error) {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
|
||||
output := make([]float32, inputSize)
|
||||
|
||||
result := C.onnx_run(
|
||||
s.session,
|
||||
(*C.float)(unsafe.Pointer(&input[0])),
|
||||
C.int(len(input)),
|
||||
(*C.float)(unsafe.Pointer(&output[0])),
|
||||
C.int(len(output)),
|
||||
)
|
||||
|
||||
if result != 0 {
|
||||
return nil, errors.New(C.GoString(C.onnx_get_last_error()))
|
||||
}
|
||||
|
||||
return output, nil
|
||||
}
|
||||
|
||||
// Close 关闭会话
|
||||
func (s *Session) Close() {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
|
||||
if s.session != nil {
|
||||
C.onnx_destroy_session(s.session)
|
||||
s.session = nil
|
||||
}
|
||||
}
|
||||
|
||||
// GetSession 获取已加载的会话
|
||||
func GetSession(name string) (*Session, bool) {
|
||||
mu.RLock()
|
||||
s, ok := sessions[name]
|
||||
mu.RUnlock()
|
||||
return s, ok
|
||||
}
|
||||
|
||||
// CloseAll 关闭所有会话
|
||||
func CloseAll() {
|
||||
mu.Lock()
|
||||
for _, s := range sessions {
|
||||
s.Close()
|
||||
}
|
||||
sessions = make(map[string]*Session)
|
||||
mu.Unlock()
|
||||
}
|
||||
@@ -0,0 +1,178 @@
|
||||
#include <opencv2/opencv.hpp>
|
||||
#include <opencv2/imgproc.hpp>
|
||||
#include <opencv2/imgcodecs.hpp>
|
||||
#include <vector>
|
||||
#include <string>
|
||||
#include <cstring>
|
||||
|
||||
static std::string last_error;
|
||||
|
||||
extern "C" {
|
||||
|
||||
// 图像解码
|
||||
Image* cv_imdecode(const unsigned char* buf, size_t size) {
|
||||
try {
|
||||
std::vector<unsigned char> data(buf, buf + size);
|
||||
cv::Mat mat = cv::imdecode(data, cv::IMREAD_COLOR);
|
||||
if (mat.empty()) {
|
||||
last_error = "无法解码图像";
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
Image* img = new Image();
|
||||
img->width = mat.cols;
|
||||
img->height = mat.rows;
|
||||
img->channels = mat.channels();
|
||||
|
||||
size_t data_size = mat.total() * mat.elemSize();
|
||||
img->data = (unsigned char*)malloc(data_size);
|
||||
memcpy(img->data, mat.data, data_size);
|
||||
|
||||
return img;
|
||||
} catch (const std::exception& e) {
|
||||
last_error = e.what();
|
||||
return nullptr;
|
||||
}
|
||||
}
|
||||
|
||||
// 释放图像
|
||||
void cv_image_free(Image* img) {
|
||||
if (img) {
|
||||
if (img->data) {
|
||||
free(img->data);
|
||||
}
|
||||
delete img;
|
||||
}
|
||||
}
|
||||
|
||||
// 滑块缺口匹配
|
||||
int cv_slider_match(const Image* target, const Image* background, int* out_x) {
|
||||
try {
|
||||
cv::Mat target_mat(target->height, target->width, CV_8UC3, target->data);
|
||||
cv::Mat bg_mat(background->height, background->width, CV_8UC3, background->data);
|
||||
|
||||
cv::Mat target_gray, bg_gray;
|
||||
cv::cvtColor(target_mat, target_gray, cv::COLOR_BGR2GRAY);
|
||||
cv::cvtColor(bg_mat, bg_gray, cv::COLOR_BGR2GRAY);
|
||||
|
||||
// 模板匹配
|
||||
cv::Mat result;
|
||||
cv::matchTemplate(bg_gray, target_gray, result, cv::TM_CCOEFF_NORMED);
|
||||
|
||||
double min_val, max_val;
|
||||
cv::Point min_loc, max_loc;
|
||||
cv::minMaxLoc(result, &min_val, &max_val, &min_loc, &max_loc);
|
||||
|
||||
*out_x = max_loc.x;
|
||||
return 0;
|
||||
} catch (const std::exception& e) {
|
||||
last_error = e.what();
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
// 阴影滑块匹配
|
||||
int cv_slider_comparison(const Image* target, const Image* background, int* out_x) {
|
||||
try {
|
||||
cv::Mat target_mat(target->height, target->width, CV_8UC3, target->data);
|
||||
cv::Mat bg_mat(background->height, background->width, CV_8UC3, background->data);
|
||||
|
||||
// 转灰度
|
||||
cv::Mat target_gray, bg_gray;
|
||||
cv::cvtColor(target_mat, target_gray, cv::COLOR_BGR2GRAY);
|
||||
cv::cvtColor(bg_mat, bg_gray, cv::COLOR_BGR2GRAY);
|
||||
|
||||
// Canny 边缘检测
|
||||
cv::Mat target_edges, bg_edges;
|
||||
cv::Canny(target_gray, target_edges, 50, 150);
|
||||
cv::Canny(bg_gray, bg_edges, 50, 150);
|
||||
|
||||
// 模板匹配
|
||||
cv::Mat result;
|
||||
cv::matchTemplate(bg_edges, target_edges, result, cv::TM_CCOEFF_NORMED);
|
||||
|
||||
double min_val, max_val;
|
||||
cv::Point min_loc, max_loc;
|
||||
cv::minMaxLoc(result, &min_val, &max_val, &min_loc, &max_loc);
|
||||
|
||||
*out_x = max_loc.x;
|
||||
return 0;
|
||||
} catch (const std::exception& e) {
|
||||
last_error = e.what();
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
// 检测旋转角度
|
||||
float cv_detect_rotation(const Image* img) {
|
||||
try {
|
||||
cv::Mat mat(img->height, img->width, CV_8UC3, img->data);
|
||||
|
||||
// 转灰度
|
||||
cv::Mat gray;
|
||||
cv::cvtColor(mat, gray, cv::COLOR_BGR2GRAY);
|
||||
|
||||
// 使用霍夫圆变换检测圆心
|
||||
cv::Mat blurred;
|
||||
cv::GaussianBlur(gray, blurred, cv::Size(5, 5), 0);
|
||||
|
||||
std::vector<cv::Vec3f> circles;
|
||||
cv::HoughCircles(blurred, circles, cv::HOUGH_GRADIENT, 1,
|
||||
blurred.rows / 8, 100, 30, 0, 0);
|
||||
|
||||
if (circles.empty()) {
|
||||
return 0.0f;
|
||||
}
|
||||
|
||||
// 简化处理：返回 0 度
|
||||
// 实际实现需要更复杂的特征点匹配
|
||||
return 0.0f;
|
||||
} catch (const std::exception& e) {
|
||||
last_error = e.what();
|
||||
return 0.0f;
|
||||
}
|
||||
}
|
||||
|
||||
// 图像相似度比较
|
||||
float cv_compare_similarity(const Image* img1, const Image* img2) {
|
||||
try {
|
||||
cv::Mat mat1(img1->height, img1->width, CV_8UC3, img1->data);
|
||||
cv::Mat mat2(img2->height, img2->width, CV_8UC3, img2->data);
|
||||
|
||||
// 确保 same size
|
||||
if (mat1.size() != mat2.size()) {
|
||||
cv::resize(mat2, mat2, mat1.size());
|
||||
}
|
||||
|
||||
// 计算 histogram
|
||||
cv::Mat hsv1, hsv2;
|
||||
cv::cvtColor(mat1, hsv1, cv::COLOR_BGR2HSV);
|
||||
cv::cvtColor(mat2, hsv2, cv::COLOR_BGR2HSV);
|
||||
|
||||
int h_bins = 50, s_bins = 60;
|
||||
int histSize[] = {h_bins, s_bins};
|
||||
float h_ranges[] = {0, 180};
|
||||
float s_ranges[] = {0, 256};
|
||||
const float* ranges[] = {h_ranges, s_ranges};
|
||||
int channels[] = {0, 1};
|
||||
|
||||
cv::Mat hist1, hist2;
|
||||
cv::calcHist(&hsv1, 1, channels, cv::Mat(), hist1, 2, histSize, ranges);
|
||||
cv::calcHist(&hsv2, 1, channels, cv::Mat(), hist2, 2, histSize, ranges);
|
||||
|
||||
cv::normalize(hist1, hist1, 0, 1, cv::NORM_MINMAX);
|
||||
cv::normalize(hist2, hist2, 0, 1, cv::NORM_MINMAX);
|
||||
|
||||
double similarity = cv::compareHist(hist1, hist2, cv::HISTCMP_CORREL);
|
||||
return (float)similarity;
|
||||
} catch (const std::exception& e) {
|
||||
last_error = e.what();
|
||||
return 0.0f;
|
||||
}
|
||||
}
|
||||
|
||||
const char* cv_get_last_error() {
|
||||
return last_error.c_str();
|
||||
}
|
||||
|
||||
} // extern "C"
|
||||
@@ -0,0 +1,146 @@
|
||||
package opencv
|
||||
|
||||
/*
|
||||
#cgo CXXFLAGS: -std=c++17
|
||||
#cgo linux LDFLAGS: -lopencv_core -lopencv_imgproc -lopencv_imgcodecs -lopencv_objdetect
|
||||
#cgo darwin LDFLAGS: -lopencv_core -lopencv_imgproc -lopencv_imgcodecs -lopencv_objdetect
|
||||
|
||||
#include <stdlib.h>
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// 图像结构
|
||||
typedef struct {
|
||||
unsigned char* data;
|
||||
int width;
|
||||
int height;
|
||||
int channels;
|
||||
} Image;
|
||||
|
||||
// 图像操作
|
||||
Image* cv_imdecode(const unsigned char* buf, size_t size);
|
||||
void cv_image_free(Image* img);
|
||||
|
||||
// 滑块匹配
|
||||
int cv_slider_match(const Image* target, const Image* background, int* out_x);
|
||||
int cv_slider_comparison(const Image* target, const Image* background, int* out_x);
|
||||
|
||||
// 旋转检测
|
||||
float cv_detect_rotation(const Image* img);
|
||||
|
||||
// 模板匹配
|
||||
int cv_template_match(const Image* src, const Image* templ, double* max_val, int* max_x, int* max_y);
|
||||
|
||||
// 特征点检测
|
||||
int cv_detect_features(const Image* img, int** points_x, int** points_y, int* count);
|
||||
|
||||
// 图像相似度
|
||||
float cv_compare_similarity(const Image* img1, const Image* img2);
|
||||
|
||||
// 错误信息
|
||||
const char* cv_get_last_error();
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
*/
|
||||
import "C"
|
||||
import (
|
||||
"encoding/base64"
|
||||
"errors"
|
||||
"fmt"
|
||||
"unsafe"
|
||||
)
|
||||
|
||||
// Image 封装图像数据
|
||||
type Image struct {
|
||||
img *C.Image
|
||||
}
|
||||
|
||||
// DecodeFromBase64 从 Base64 解码图像
|
||||
func DecodeFromBase64(data string) (*Image, error) {
|
||||
decoded, err := base64.StdEncoding.DecodeString(data)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("base64 解码失败: %v", err)
|
||||
}
|
||||
|
||||
img := C.cv_imdecode(
|
||||
(*C.uchar)(unsafe.Pointer(&decoded[0])),
|
||||
C.size_t(len(decoded)),
|
||||
)
|
||||
|
||||
if img == nil {
|
||||
return nil, errors.New(C.GoString(C.cv_get_last_error()))
|
||||
}
|
||||
|
||||
return &Image{img: img}, nil
|
||||
}
|
||||
|
||||
// Free 释放图像内存
|
||||
func (i *Image) Free() {
|
||||
if i.img != nil {
|
||||
C.cv_image_free(i.img)
|
||||
i.img = nil
|
||||
}
|
||||
}
|
||||
|
||||
// Width 获取宽度
|
||||
func (i *Image) Width() int {
|
||||
return int(i.img.width)
|
||||
}
|
||||
|
||||
// Height 获取高度
|
||||
func (i *Image) Height() int {
|
||||
return int(i.img.height)
|
||||
}
|
||||
|
||||
// SliderMatch 滑块缺口匹配
|
||||
func SliderMatch(target, background *Image) (int, error) {
|
||||
var outX C.int
|
||||
|
||||
result := C.cv_slider_match(
|
||||
(*C.Image)(target.img),
|
||||
(*C.Image)(background.img),
|
||||
&outX,
|
||||
)
|
||||
|
||||
if result != 0 {
|
||||
return 0, errors.New(C.GoString(C.cv_get_last_error()))
|
||||
}
|
||||
|
||||
return int(outX), nil
|
||||
}
|
||||
|
||||
// SliderComparison 阴影滑块匹配
|
||||
func SliderComparison(target, background *Image) (int, error) {
|
||||
var outX C.int
|
||||
|
||||
result := C.cv_slider_comparison(
|
||||
(*C.Image)(target.img),
|
||||
(*C.Image)(background.img),
|
||||
&outX,
|
||||
)
|
||||
|
||||
if result != 0 {
|
||||
return 0, errors.New(C.GoString(C.cv_get_last_error()))
|
||||
}
|
||||
|
||||
return int(outX), nil
|
||||
}
|
||||
|
||||
// DetectRotation 检测旋转角度
|
||||
func DetectRotation(img *Image) (float32, error) {
|
||||
angle := C.cv_detect_rotation((*C.Image)(img.img))
|
||||
return float32(angle), nil
|
||||
}
|
||||
|
||||
// CompareSimilarity 比较图像相似度
|
||||
func CompareSimilarity(img1, img2 *Image) (float32, error) {
|
||||
similarity := C.cv_compare_similarity(
|
||||
(*C.Image)(img1.img),
|
||||
(*C.Image)(img2.img),
|
||||
)
|
||||
return float32(similarity), nil
|
||||
}
|
||||
Executable
+107
@@ -0,0 +1,107 @@
|
||||
#!/bin/bash
|
||||
|
||||
# AntiCaptcha 安装脚本
|
||||
|
||||
set -e
|
||||
|
||||
echo "========================================="
|
||||
echo " AntiCaptcha 安装脚本"
|
||||
echo "========================================="
|
||||
|
||||
# 检查系统
|
||||
if [ ! -f /etc/os-release ]; then
|
||||
echo "无法检测操作系统"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
. /etc/os-release
|
||||
|
||||
# 安装依赖
|
||||
install_deps() {
|
||||
echo "正在安装依赖..."
|
||||
|
||||
case "$ID" in
|
||||
ubuntu|debian)
|
||||
apt update
|
||||
apt install -y libopencv-dev libonnxruntime-dev
|
||||
;;
|
||||
centos|rhel|fedora)
|
||||
yum install -y opencv-devel onnxruntime
|
||||
;;
|
||||
alpine)
|
||||
apk add --no-cache opencv onnxruntime
|
||||
;;
|
||||
*)
|
||||
echo "不支持的系统: $ID"
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
# 安装 Go
|
||||
install_go() {
|
||||
if command -v go &> /dev/null; then
|
||||
echo "Go 已安装: $(go version)"
|
||||
return
|
||||
fi
|
||||
|
||||
echo "正在安装 Go..."
|
||||
wget -q https://go.dev/dl/go1.22.5.linux-amd64.tar.gz
|
||||
tar -C /usr/local -xzf go1.22.5.linux-amd64.tar.gz
|
||||
rm go1.22.5.linux-amd64.tar.gz
|
||||
|
||||
export PATH=$PATH:/usr/local/go/bin
|
||||
echo 'export PATH=$PATH:/usr/local/go/bin' >> ~/.bashrc
|
||||
}
|
||||
|
||||
# 构建项目
|
||||
build() {
|
||||
echo "正在构建项目..."
|
||||
go mod download
|
||||
CGO_ENABLED=1 go build -o anticaptcha ./cmd/server
|
||||
}
|
||||
|
||||
# 创建 systemd 服务
|
||||
create_service() {
|
||||
echo "正在创建 systemd 服务..."
|
||||
|
||||
cat > /etc/systemd/system/anticaptcha.service << EOF
|
||||
[Unit]
|
||||
Description=AntiCaptcha Service
|
||||
After=network.target
|
||||
|
||||
[Service]
|
||||
Type=simple
|
||||
ExecStart=/opt/anticaptcha/anticaptcha
|
||||
WorkingDirectory=/opt/anticaptcha
|
||||
Restart=always
|
||||
RestartSec=5
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
||||
EOF
|
||||
|
||||
systemctl daemon-reload
|
||||
systemctl enable anticaptcha
|
||||
}
|
||||
|
||||
# 主流程
|
||||
main() {
|
||||
install_deps
|
||||
install_go
|
||||
build
|
||||
|
||||
echo ""
|
||||
echo "========================================="
|
||||
echo " 安装完成!"
|
||||
echo "========================================="
|
||||
echo ""
|
||||
echo "运行方式:"
|
||||
echo " ./anticaptcha"
|
||||
echo ""
|
||||
echo "或创建 systemd 服务后:"
|
||||
echo " systemctl start anticaptcha"
|
||||
echo ""
|
||||
}
|
||||
|
||||
main "$@"
|
||||
@@ -0,0 +1,13 @@
|
||||
<!DOCTYPE html>
|
||||
<html lang="zh-CN">
|
||||
<head>
|
||||
<meta charset="UTF-8" />
|
||||
<link rel="icon" type="image/svg+xml" href="/favicon.svg" />
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
|
||||
<title>AntiCaptcha - 验证码识别服务</title>
|
||||
</head>
|
||||
<body>
|
||||
<div id="root"></div>
|
||||
<script type="module" src="/src/main.tsx"></script>
|
||||
</body>
|
||||
</html>
|
||||
@@ -0,0 +1,23 @@
|
||||
{
|
||||
"name": "anticaptcha-web",
|
||||
"version": "1.0.0",
|
||||
"type": "module",
|
||||
"scripts": {
|
||||
"dev": "vite",
|
||||
"build": "vite build",
|
||||
"preview": "vite preview"
|
||||
},
|
||||
"dependencies": {
|
||||
"react": "^18.3.1",
|
||||
"react-dom": "^18.3.1",
|
||||
"react-router-dom": "^6.24.1",
|
||||
"axios": "^1.7.2"
|
||||
},
|
||||
"devDependencies": {
|
||||
"@types/react": "^18.3.3",
|
||||
"@types/react-dom": "^18.3.0",
|
||||
"@vitejs/plugin-react": "^4.3.1",
|
||||
"typescript": "^5.5.3",
|
||||
"vite": "^5.3.4"
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,49 @@
|
||||
import { BrowserRouter, Routes, Route, Navigate } from 'react-router-dom'
|
||||
import { useState, useEffect } from 'react'
|
||||
import LoginPage from './pages/Login'
|
||||
import RegisterPage from './pages/Register'
|
||||
import HomePage from './pages/Home'
|
||||
import AdminPage from './pages/Admin'
|
||||
import InstallPage from './pages/Install'
|
||||
|
||||
function App() {
|
||||
const [isInstalled, setIsInstalled] = useState<boolean | null>(null)
|
||||
const [token, setToken] = useState<string | null>(localStorage.getItem('token'))
|
||||
|
||||
useEffect(() => {
|
||||
// 检查是否已安装
|
||||
fetch('/api/install/check')
|
||||
.then(res => res.json())
|
||||
.then(data => setIsInstalled(data.installed))
|
||||
.catch(() => setIsInstalled(false))
|
||||
}, [])
|
||||
|
||||
if (isInstalled === null) {
|
||||
return <div className="container text-center">加载中...</div>
|
||||
}
|
||||
|
||||
if (!isInstalled) {
|
||||
return (
|
||||
<BrowserRouter>
|
||||
<Routes>
|
||||
<Route path="/install" element={<InstallPage onInstall={() => setIsInstalled(true)} />} />
|
||||
<Route path="*" element={<Navigate to="/install" replace />} />
|
||||
</Routes>
|
||||
</BrowserRouter>
|
||||
)
|
||||
}
|
||||
|
||||
return (
|
||||
<BrowserRouter>
|
||||
<Routes>
|
||||
<Route path="/login" element={<LoginPage onLogin={setToken} />} />
|
||||
<Route path="/register" element={<RegisterPage />} />
|
||||
<Route path="/" element={token ? <HomePage token={token} /> : <Navigate to="/login" />} />
|
||||
<Route path="/admin" element={token ? <AdminPage token={token} /> : <Navigate to="/login" />} />
|
||||
<Route path="*" element={<Navigate to="/" replace />} />
|
||||
</Routes>
|
||||
</BrowserRouter>
|
||||
)
|
||||
}
|
||||
|
||||
export default App
|
||||
@@ -0,0 +1,160 @@
|
||||
* {
|
||||
margin: 0;
|
||||
padding: 0;
|
||||
box-sizing: border-box;
|
||||
}
|
||||
|
||||
:root {
|
||||
--primary: #3b82f6;
|
||||
--primary-dark: #2563eb;
|
||||
--bg-primary: #0f172a;
|
||||
--bg-secondary: #1e293b;
|
||||
--bg-tertiary: #334155;
|
||||
--text-primary: #f1f5f9;
|
||||
--text-secondary: #94a3b8;
|
||||
--text-muted: #64748b;
|
||||
--border: #334155;
|
||||
--border-hover: #475569;
|
||||
--error: #ef4444;
|
||||
--success: #22c55e;
|
||||
}
|
||||
|
||||
body {
|
||||
font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, Oxygen, Ubuntu, Cantarell, sans-serif;
|
||||
background: var(--bg-primary);
|
||||
color: var(--text-primary);
|
||||
min-height: 100vh;
|
||||
}
|
||||
|
||||
.container {
|
||||
max-width: 1200px;
|
||||
margin: 0 auto;
|
||||
padding: 24px;
|
||||
}
|
||||
|
||||
.card {
|
||||
background: var(--bg-secondary);
|
||||
border: 1px solid var(--border);
|
||||
border-radius: 8px;
|
||||
padding: 24px;
|
||||
margin-bottom: 16px;
|
||||
}
|
||||
|
||||
.input {
|
||||
width: 100%;
|
||||
padding: 12px 16px;
|
||||
background: var(--bg-tertiary);
|
||||
border: 1px solid var(--border);
|
||||
border-radius: 6px;
|
||||
color: var(--text-primary);
|
||||
font-size: 14px;
|
||||
outline: none;
|
||||
transition: border-color 0.2s;
|
||||
}
|
||||
|
||||
.input:focus {
|
||||
border-color: var(--primary);
|
||||
}
|
||||
|
||||
.button {
|
||||
padding: 10px 20px;
|
||||
background: var(--primary);
|
||||
color: white;
|
||||
border: none;
|
||||
border-radius: 6px;
|
||||
font-size: 14px;
|
||||
font-weight: 500;
|
||||
cursor: pointer;
|
||||
transition: background 0.2s;
|
||||
}
|
||||
|
||||
.button:hover {
|
||||
background: var(--primary-dark);
|
||||
}
|
||||
|
||||
.button:disabled {
|
||||
opacity: 0.5;
|
||||
cursor: not-allowed;
|
||||
}
|
||||
|
||||
.button-secondary {
|
||||
background: var(--bg-tertiary);
|
||||
}
|
||||
|
||||
.button-secondary:hover {
|
||||
background: var(--border);
|
||||
}
|
||||
|
||||
.label {
|
||||
display: block;
|
||||
margin-bottom: 8px;
|
||||
font-size: 13px;
|
||||
font-weight: 500;
|
||||
color: var(--text-secondary);
|
||||
}
|
||||
|
||||
.form-group {
|
||||
margin-bottom: 20px;
|
||||
}
|
||||
|
||||
.error {
|
||||
color: var(--error);
|
||||
font-size: 13px;
|
||||
margin-top: 8px;
|
||||
}
|
||||
|
||||
.success {
|
||||
color: var(--success);
|
||||
font-size: 13px;
|
||||
margin-top: 8px;
|
||||
}
|
||||
|
||||
.text-center {
|
||||
text-align: center;
|
||||
}
|
||||
|
||||
.flex {
|
||||
display: flex;
|
||||
}
|
||||
|
||||
.flex-col {
|
||||
flex-direction: column;
|
||||
}
|
||||
|
||||
.items-center {
|
||||
align-items: center;
|
||||
}
|
||||
|
||||
.justify-between {
|
||||
justify-content: space-between;
|
||||
}
|
||||
|
||||
.gap-2 {
|
||||
gap: 8px;
|
||||
}
|
||||
|
||||
.gap-4 {
|
||||
gap: 16px;
|
||||
}
|
||||
|
||||
.mt-4 {
|
||||
margin-top: 16px;
|
||||
}
|
||||
|
||||
.mb-4 {
|
||||
margin-bottom: 16px;
|
||||
}
|
||||
|
||||
.grid {
|
||||
display: grid;
|
||||
}
|
||||
|
||||
.grid-cols-2 {
|
||||
grid-template-columns: repeat(2, 1fr);
|
||||
}
|
||||
|
||||
@media (max-width: 768px) {
|
||||
.grid-cols-2 {
|
||||
grid-template-columns: 1fr;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,10 @@
|
||||
import React from 'react'
|
||||
import ReactDOM from 'react-dom/client'
|
||||
import App from './App'
|
||||
import './index.css'
|
||||
|
||||
ReactDOM.createRoot(document.getElementById('root')!).render(
|
||||
<React.StrictMode>
|
||||
<App />
|
||||
</React.StrictMode>,
|
||||
)
|
||||
@@ -0,0 +1,101 @@
|
||||
import { useState } from 'react'
|
||||
|
||||
interface Props {
|
||||
token: string
|
||||
}
|
||||
|
||||
export default function AdminPage({ token }: Props) {
|
||||
const [points, setPoints] = useState(1000)
|
||||
const [codes, setCodes] = useState<any[]>([])
|
||||
const [loading, setLoading] = useState(false)
|
||||
|
||||
const generateCode = async () => {
|
||||
setLoading(true)
|
||||
try {
|
||||
const res = await fetch('/api/admin/generate_code', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
'Authorization': `Bearer ${token}`
|
||||
},
|
||||
body: JSON.stringify({ points })
|
||||
})
|
||||
|
||||
if (res.ok) {
|
||||
loadCodes()
|
||||
}
|
||||
} finally {
|
||||
setLoading(false)
|
||||
}
|
||||
}
|
||||
|
||||
const loadCodes = async () => {
|
||||
const res = await fetch('/api/admin/regcodes', {
|
||||
headers: { 'Authorization': `Bearer ${token}` }
|
||||
})
|
||||
if (res.ok) {
|
||||
setCodes(await res.json())
|
||||
}
|
||||
}
|
||||
|
||||
return (
|
||||
<div className="container">
|
||||
<h1 className="mb-4">管理后台</h1>
|
||||
|
||||
<div className="card">
|
||||
<h2 className="mb-4">生成注册码</h2>
|
||||
|
||||
<div className="flex gap-4 items-center">
|
||||
<div className="form-group" style={{ flex: 1 }}>
|
||||
<label className="label">积分数量</label>
|
||||
<input
|
||||
type="number"
|
||||
className="input"
|
||||
value={points}
|
||||
onChange={e => setPoints(Number(e.target.value))}
|
||||
/>
|
||||
</div>
|
||||
|
||||
<button
|
||||
className="button"
|
||||
onClick={generateCode}
|
||||
disabled={loading}
|
||||
>
|
||||
{loading ? '生成中...' : '生成注册码'}
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div className="card">
|
||||
<h2 className="mb-4">注册码列表</h2>
|
||||
|
||||
{codes.length === 0 ? (
|
||||
<p style={{ color: 'var(--text-muted)' }}>暂无注册码，点击上方按钮生成</p>
|
||||
) : (
|
||||
<div style={{ overflowX: 'auto' }}>
|
||||
<table style={{ width: '100%', borderCollapse: 'collapse' }}>
|
||||
<thead>
|
||||
<tr style={{ borderBottom: '1px solid var(--border)' }}>
|
||||
<th style={{ padding: '12px', textAlign: 'left' }}>注册码</th>
|
||||
<th style={{ padding: '12px', textAlign: 'left' }}>积分</th>
|
||||
<th style={{ padding: '12px', textAlign: 'left' }}>状态</th>
|
||||
</tr>
|
||||
</thead>
|
||||
<tbody>
|
||||
{codes.map((code: any) => (
|
||||
<tr key={code.id} style={{ borderBottom: '1px solid var(--border)' }}>
|
||||
<td style={{ padding: '12px' }}>{code.code}</td>
|
||||
<td style={{ padding: '12px' }}>{code.points}</td>
|
||||
<td style={{ padding: '12px' }}>
|
||||
{code.is_used ? '已使用' : '未使用'}
|
||||
</td>
|
||||
</tr>
|
||||
))}
|
||||
</tbody>
|
||||
</table>
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
</div>
|
||||
)
|
||||
}
|
||||
@@ -0,0 +1,48 @@
|
||||
import { Link } from 'react-router-dom'
|
||||
|
||||
interface Props {
|
||||
token: string
|
||||
}
|
||||
|
||||
export default function HomePage({ token }: Props) {
|
||||
const handleLogout = () => {
|
||||
localStorage.removeItem('token')
|
||||
window.location.href = '/login'
|
||||
}
|
||||
|
||||
return (
|
||||
<div className="container">
|
||||
<div className="flex justify-between items-center mb-4">
|
||||
<h1>AntiCaptcha 控制台</h1>
|
||||
<div className="flex gap-2">
|
||||
<Link to="/admin" className="button button-secondary">管理后台</Link>
|
||||
<button onClick={handleLogout} className="button button-secondary">退出登录</button>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div className="card">
|
||||
<h2 className="mb-4">API 使用说明</h2>
|
||||
<p style={{ color: 'var(--text-secondary)', marginBottom: '16px' }}>
|
||||
使用以下接口进行验证码识别，需要在请求头中携带 Authorization: Bearer {token.substring(0, 20)}...
|
||||
</p>
|
||||
|
||||
<div style={{ background: 'var(--bg-tertiary)', padding: '16px', borderRadius: '8px' }}>
|
||||
<code style={{ color: 'var(--text-primary)' }}>
|
||||
POST /api/ocr - OCR 文字识别<br/>
|
||||
POST /api/slider/match - 滑块缺口匹配<br/>
|
||||
POST /api/compare/similarity - 图片相似度对比<br/>
|
||||
POST /api/rotate/single/rotate - 单图旋转验证码<br/>
|
||||
POST /api/detection/icon - 图标检测<br/>
|
||||
</code>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div className="card">
|
||||
<h2 className="mb-4">账户信息</h2>
|
||||
<p style={{ color: 'var(--text-secondary)' }}>
|
||||
请联系管理员获取更多信息
|
||||
</p>
|
||||
</div>
|
||||
</div>
|
||||
)
|
||||
}
|
||||
@@ -0,0 +1,319 @@
|
||||
import { useState } from 'react'
|
||||
|
||||
interface Props {
|
||||
onInstall: () => void
|
||||
}
|
||||
|
||||
export default function InstallPage({ onInstall }: Props) {
|
||||
const [step, setStep] = useState(1)
|
||||
const [loading, setLoading] = useState(false)
|
||||
const [error, setError] = useState('')
|
||||
|
||||
const [config, setConfig] = useState({
|
||||
database: {
|
||||
type: 'sqlite',
|
||||
host: 'localhost',
|
||||
port: 3306,
|
||||
user: 'root',
|
||||
password: '',
|
||||
database: 'anticaptcha',
|
||||
sqlite: { path: './data/app.db' }
|
||||
},
|
||||
redis: {
|
||||
enabled: false,
|
||||
host: 'localhost',
|
||||
port: 6379,
|
||||
password: '',
|
||||
db: 0
|
||||
},
|
||||
admin: {
|
||||
username: 'admin',
|
||||
password: ''
|
||||
}
|
||||
})
|
||||
|
||||
const handleSubmit = async () => {
|
||||
setLoading(true)
|
||||
setError('')
|
||||
|
||||
try {
|
||||
const res = await fetch('/api/install', {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify(config)
|
||||
})
|
||||
|
||||
const data = await res.json()
|
||||
|
||||
if (!res.ok) {
|
||||
setError(data.error || '安装失败')
|
||||
return
|
||||
}
|
||||
|
||||
onInstall()
|
||||
window.location.href = '/'
|
||||
} catch {
|
||||
setError('网络错误')
|
||||
} finally {
|
||||
setLoading(false)
|
||||
}
|
||||
}
|
||||
|
||||
return (
|
||||
<div className="container" style={{ maxWidth: '600px', marginTop: '50px' }}>
|
||||
<div className="card">
|
||||
<h1 style={{ marginBottom: '8px' }}>AntiCaptcha 安装向导</h1>
|
||||
<p style={{ color: 'var(--text-muted)', marginBottom: '32px' }}>
|
||||
步骤 {step} / 3
|
||||
</p>
|
||||
|
||||
{step === 1 && (
|
||||
<>
|
||||
<h2 className="mb-4">数据库配置</h2>
|
||||
|
||||
<div className="form-group">
|
||||
<label className="label">数据库类型</label>
|
||||
<select
|
||||
className="input"
|
||||
value={config.database.type}
|
||||
onChange={e => setConfig({
|
||||
...config,
|
||||
database: { ...config.database, type: e.target.value }
|
||||
})}
|
||||
>
|
||||
<option value="sqlite">SQLite</option>
|
||||
<option value="mysql">MySQL</option>
|
||||
</select>
|
||||
</div>
|
||||
|
||||
{config.database.type === 'sqlite' ? (
|
||||
<div className="form-group">
|
||||
<label className="label">数据库文件路径</label>
|
||||
<input
|
||||
type="text"
|
||||
className="input"
|
||||
value={config.database.sqlite.path}
|
||||
onChange={e => setConfig({
|
||||
...config,
|
||||
database: {
|
||||
...config.database,
|
||||
sqlite: { path: e.target.value }
|
||||
}
|
||||
})}
|
||||
/>
|
||||
</div>
|
||||
) : (
|
||||
<>
|
||||
<div className="grid grid-cols-2 gap-4">
|
||||
<div className="form-group">
|
||||
<label className="label">主机</label>
|
||||
<input
|
||||
type="text"
|
||||
className="input"
|
||||
value={config.database.host}
|
||||
onChange={e => setConfig({
|
||||
...config,
|
||||
database: { ...config.database, host: e.target.value }
|
||||
})}
|
||||
/>
|
||||
</div>
|
||||
<div className="form-group">
|
||||
<label className="label">端口</label>
|
||||
<input
|
||||
type="number"
|
||||
className="input"
|
||||
value={config.database.port}
|
||||
onChange={e => setConfig({
|
||||
...config,
|
||||
database: { ...config.database, port: Number(e.target.value) }
|
||||
})}
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div className="grid grid-cols-2 gap-4">
|
||||
<div className="form-group">
|
||||
<label className="label">用户名</label>
|
||||
<input
|
||||
type="text"
|
||||
className="input"
|
||||
value={config.database.user}
|
||||
onChange={e => setConfig({
|
||||
...config,
|
||||
database: { ...config.database, user: e.target.value }
|
||||
})}
|
||||
/>
|
||||
</div>
|
||||
<div className="form-group">
|
||||
<label className="label">密码</label>
|
||||
<input
|
||||
type="password"
|
||||
className="input"
|
||||
value={config.database.password}
|
||||
onChange={e => setConfig({
|
||||
...config,
|
||||
database: { ...config.database, password: e.target.value }
|
||||
})}
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div className="form-group">
|
||||
<label className="label">数据库名</label>
|
||||
<input
|
||||
type="text"
|
||||
className="input"
|
||||
value={config.database.database}
|
||||
onChange={e => setConfig({
|
||||
...config,
|
||||
database: { ...config.database, database: e.target.value }
|
||||
})}
|
||||
/>
|
||||
</div>
|
||||
</>
|
||||
)}
|
||||
|
||||
<button className="button" onClick={() => setStep(2)}>
|
||||
下一步
|
||||
</button>
|
||||
</>
|
||||
)}
|
||||
|
||||
{step === 2 && (
|
||||
<>
|
||||
<h2 className="mb-4">Redis 配置（可选）</h2>
|
||||
|
||||
<div className="form-group">
|
||||
<label className="flex gap-2 items-center">
|
||||
<input
|
||||
type="checkbox"
|
||||
checked={config.redis.enabled}
|
||||
onChange={e => setConfig({
|
||||
...config,
|
||||
redis: { ...config.redis, enabled: e.target.checked }
|
||||
})}
|
||||
/>
|
||||
启用 Redis
|
||||
</label>
|
||||
</div>
|
||||
|
||||
{config.redis.enabled && (
|
||||
<>
|
||||
<div className="grid grid-cols-2 gap-4">
|
||||
<div className="form-group">
|
||||
<label className="label">主机</label>
|
||||
<input
|
||||
type="text"
|
||||
className="input"
|
||||
value={config.redis.host}
|
||||
onChange={e => setConfig({
|
||||
...config,
|
||||
redis: { ...config.redis, host: e.target.value }
|
||||
})}
|
||||
/>
|
||||
</div>
|
||||
<div className="form-group">
|
||||
<label className="label">端口</label>
|
||||
<input
|
||||
type="number"
|
||||
className="input"
|
||||
value={config.redis.port}
|
||||
onChange={e => setConfig({
|
||||
...config,
|
||||
redis: { ...config.redis, port: Number(e.target.value) }
|
||||
})}
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div className="grid grid-cols-2 gap-4">
|
||||
<div className="form-group">
|
||||
<label className="label">密码</label>
|
||||
<input
|
||||
type="password"
|
||||
className="input"
|
||||
value={config.redis.password}
|
||||
onChange={e => setConfig({
|
||||
...config,
|
||||
redis: { ...config.redis, password: e.target.value }
|
||||
})}
|
||||
/>
|
||||
</div>
|
||||
<div className="form-group">
|
||||
<label className="label">数据库</label>
|
||||
<input
|
||||
type="number"
|
||||
className="input"
|
||||
value={config.redis.db}
|
||||
onChange={e => setConfig({
|
||||
...config,
|
||||
redis: { ...config.redis, db: Number(e.target.value) }
|
||||
})}
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
</>
|
||||
)}
|
||||
|
||||
<div className="flex gap-2">
|
||||
<button className="button button-secondary" onClick={() => setStep(1)}>
|
||||
上一步
|
||||
</button>
|
||||
<button className="button" onClick={() => setStep(3)}>
|
||||
下一步
|
||||
</button>
|
||||
</div>
|
||||
</>
|
||||
)}
|
||||
|
||||
{step === 3 && (
|
||||
<>
|
||||
<h2 className="mb-4">管理员账号</h2>
|
||||
|
||||
<div className="form-group">
|
||||
<label className="label">用户名</label>
|
||||
<input
|
||||
type="text"
|
||||
className="input"
|
||||
value={config.admin.username}
|
||||
onChange={e => setConfig({
|
||||
...config,
|
||||
admin: { ...config.admin, username: e.target.value }
|
||||
})}
|
||||
/>
|
||||
</div>
|
||||
|
||||
<div className="form-group">
|
||||
<label className="label">密码</label>
|
||||
<input
|
||||
type="password"
|
||||
className="input"
|
||||
value={config.admin.password}
|
||||
onChange={e => setConfig({
|
||||
...config,
|
||||
admin: { ...config.admin, password: e.target.value }
|
||||
})}
|
||||
placeholder="请输入管理员密码"
|
||||
/>
|
||||
</div>
|
||||
|
||||
{error && <div className="error">{error}</div>}
|
||||
|
||||
<div className="flex gap-2">
|
||||
<button className="button button-secondary" onClick={() => setStep(2)}>
|
||||
上一步
|
||||
</button>
|
||||
<button
|
||||
className="button"
|
||||
onClick={handleSubmit}
|
||||
disabled={loading}
|
||||
>
|
||||
{loading ? '安装中...' : '完成安装'}
|
||||
</button>
|
||||
</div>
|
||||
</>
|
||||
)}
|
||||
</div>
|
||||
</div>
|
||||
)
|
||||
}
|
||||
@@ -0,0 +1,84 @@
|
||||
import { useState } from 'react'
|
||||
|
||||
interface Props {
|
||||
onLogin: (token: string) => void
|
||||
}
|
||||
|
||||
export default function LoginPage({ onLogin }: Props) {
|
||||
const [username, setUsername] = useState('')
|
||||
const [password, setPassword] = useState('')
|
||||
const [error, setError] = useState('')
|
||||
const [loading, setLoading] = useState(false)
|
||||
|
||||
const handleSubmit = async (e: React.FormEvent) => {
|
||||
e.preventDefault()
|
||||
setError('')
|
||||
setLoading(true)
|
||||
|
||||
try {
|
||||
const res = await fetch('/api/login', {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ username, password }),
|
||||
})
|
||||
|
||||
const data = await res.json()
|
||||
|
||||
if (!res.ok) {
|
||||
setError(data.error || '登录失败')
|
||||
return
|
||||
}
|
||||
|
||||
localStorage.setItem('token', data.access_token)
|
||||
onLogin(data.access_token)
|
||||
} catch {
|
||||
setError('网络错误')
|
||||
} finally {
|
||||
setLoading(false)
|
||||
}
|
||||
}
|
||||
|
||||
return (
|
||||
<div className="container" style={{ maxWidth: '400px', marginTop: '100px' }}>
|
||||
<div className="card">
|
||||
<h1 style={{ marginBottom: '24px', textAlign: 'center' }}>登录</h1>
|
||||
|
||||
<form onSubmit={handleSubmit}>
|
||||
<div className="form-group">
|
||||
<label className="label">用户名</label>
|
||||
<input
|
||||
type="text"
|
||||
className="input"
|
||||
value={username}
|
||||
onChange={e => setUsername(e.target.value)}
|
||||
placeholder="请输入用户名"
|
||||
required
|
||||
/>
|
||||
</div>
|
||||
|
||||
<div className="form-group">
|
||||
<label className="label">密码</label>
|
||||
<input
|
||||
type="password"
|
||||
className="input"
|
||||
value={password}
|
||||
onChange={e => setPassword(e.target.value)}
|
||||
placeholder="请输入密码"
|
||||
required
|
||||
/>
|
||||
</div>
|
||||
|
||||
{error && <div className="error">{error}</div>}
|
||||
|
||||
<button type="submit" className="button" style={{ width: '100%' }} disabled={loading}>
|
||||
{loading ? '登录中...' : '登录'}
|
||||
</button>
|
||||
</form>
|
||||
|
||||
<div className="text-center mt-4">
|
||||
<a href="/register" style={{ color: 'var(--primary)' }}>没有账号？立即注册</a>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
)
|
||||
}
|
||||
@@ -0,0 +1,96 @@
|
||||
import { useState } from 'react'
|
||||
|
||||
export default function RegisterPage() {
|
||||
const [username, setUsername] = useState('')
|
||||
const [password, setPassword] = useState('')
|
||||
const [code, setCode] = useState('')
|
||||
const [error, setError] = useState('')
|
||||
const [success, setSuccess] = useState('')
|
||||
const [loading, setLoading] = useState(false)
|
||||
|
||||
const handleSubmit = async (e: React.FormEvent) => {
|
||||
e.preventDefault()
|
||||
setError('')
|
||||
setSuccess('')
|
||||
setLoading(true)
|
||||
|
||||
try {
|
||||
const res = await fetch('/api/register', {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ username, password, registration_code: code }),
|
||||
})
|
||||
|
||||
const data = await res.json()
|
||||
|
||||
if (!res.ok) {
|
||||
setError(data.error || '注册失败')
|
||||
return
|
||||
}
|
||||
|
||||
setSuccess('注册成功，请登录')
|
||||
setTimeout(() => window.location.href = '/login', 2000)
|
||||
} catch {
|
||||
setError('网络错误')
|
||||
} finally {
|
||||
setLoading(false)
|
||||
}
|
||||
}
|
||||
|
||||
return (
|
||||
<div className="container" style={{ maxWidth: '400px', marginTop: '100px' }}>
|
||||
<div className="card">
|
||||
<h1 style={{ marginBottom: '24px', textAlign: 'center' }}>注册</h1>
|
||||
|
||||
<form onSubmit={handleSubmit}>
|
||||
<div className="form-group">
|
||||
<label className="label">用户名</label>
|
||||
<input
|
||||
type="text"
|
||||
className="input"
|
||||
value={username}
|
||||
onChange={e => setUsername(e.target.value)}
|
||||
placeholder="请输入用户名"
|
||||
required
|
||||
/>
|
||||
</div>
|
||||
|
||||
<div className="form-group">
|
||||
<label className="label">密码</label>
|
||||
<input
|
||||
type="password"
|
||||
className="input"
|
||||
value={password}
|
||||
onChange={e => setPassword(e.target.value)}
|
||||
placeholder="请输入密码"
|
||||
required
|
||||
/>
|
||||
</div>
|
||||
|
||||
<div className="form-group">
|
||||
<label className="label">注册码</label>
|
||||
<input
|
||||
type="text"
|
||||
className="input"
|
||||
value={code}
|
||||
onChange={e => setCode(e.target.value)}
|
||||
placeholder="请输入注册码"
|
||||
required
|
||||
/>
|
||||
</div>
|
||||
|
||||
{error && <div className="error">{error}</div>}
|
||||
{success && <div className="success">{success}</div>}
|
||||
|
||||
<button type="submit" className="button" style={{ width: '100%' }} disabled={loading}>
|
||||
{loading ? '注册中...' : '注册'}
|
||||
</button>
|
||||
</form>
|
||||
|
||||
<div className="text-center mt-4">
|
||||
<a href="/login" style={{ color: 'var(--primary)' }}>已有账号？立即登录</a>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
)
|
||||
}
|
||||
@@ -0,0 +1,21 @@
|
||||
{
|
||||
"compilerOptions": {
|
||||
"target": "ES2020",
|
||||
"useDefineForClassFields": true,
|
||||
"lib": ["ES2020", "DOM", "DOM.Iterable"],
|
||||
"module": "ESNext",
|
||||
"skipLibCheck": true,
|
||||
"moduleResolution": "bundler",
|
||||
"allowImportingTsExtensions": true,
|
||||
"resolveJsonModule": true,
|
||||
"isolatedModules": true,
|
||||
"noEmit": true,
|
||||
"jsx": "react-jsx",
|
||||
"strict": true,
|
||||
"noUnusedLocals": true,
|
||||
"noUnusedParameters": true,
|
||||
"noFallthroughCasesInSwitch": true
|
||||
},
|
||||
"include": ["src"],
|
||||
"references": [{ "path": "./tsconfig.node.json" }]
|
||||
}
|
||||
@@ -0,0 +1,11 @@
|
||||
{
|
||||
"compilerOptions": {
|
||||
"composite": true,
|
||||
"skipLibCheck": true,
|
||||
"module": "ESNext",
|
||||
"moduleResolution": "bundler",
|
||||
"allowSyntheticDefaultImports": true,
|
||||
"strict": true
|
||||
},
|
||||
"include": ["vite.config.ts"]
|
||||
}
|
||||
@@ -0,0 +1,15 @@
|
||||
import { defineConfig } from 'vite'
|
||||
import react from '@vitejs/plugin-react'
|
||||
|
||||
export default defineConfig({
|
||||
plugins: [react()],
|
||||
build: {
|
||||
outDir: 'dist',
|
||||
emptyOutDir: true,
|
||||
},
|
||||
server: {
|
||||
proxy: {
|
||||
'/api': 'http://localhost:6688',
|
||||
},
|
||||
},
|
||||
})
|
||||
Reference in New Issue
Block a user