kevin
96278fdf37
- model: ServerURL → Protocol/Host/ServerIPs/Port/Path 五个字段 - db: 自动迁移旧 server_url 列到新表结构 - ui: 配置窗口改为横向分组布局，保存失败时不再关闭窗口 - transport: 支持 TLS SNI + Host 头覆盖以连接 CDN 边缘节点 - session: CDN IP 列表连接失败自动切换到下一个 - 新增重置数据库按钮，一键清空所有配置
115 lines
3.5 KiB
Go
115 lines
3.5 KiB
Go
package db
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"time"
|
|
|
|
"lmvpn/internal/model"
|
|
)
|
|
|
|
// CreateProfile inserts a new server profile and returns its ID.
|
|
func (s *Store) CreateProfile(p *model.ServerProfile) (int64, error) {
|
|
res, err := s.db.Exec(
|
|
`INSERT INTO server_profiles
|
|
(name, protocol, host, server_ips, port, path,
|
|
username, auth_mode, routing_mode,
|
|
custom_cidrs, mtu_override, auto_connect)
|
|
VALUES (?,?,?,?,?,?,?,?,?,?,?,?)`,
|
|
p.Name, p.Protocol, p.Host, p.ServerIPs, p.Port, p.Path,
|
|
p.Username, p.AuthMode, p.RoutingMode,
|
|
p.CustomCIDRs, p.MTUOverride, p.AutoConnect,
|
|
)
|
|
if err != nil {
|
|
return 0, fmt.Errorf("insert profile: %w", err)
|
|
}
|
|
id, _ := res.LastInsertId()
|
|
p.ID = id
|
|
p.CreatedAt = time.Now()
|
|
return id, nil
|
|
}
|
|
|
|
// GetProfile returns a single profile by ID.
|
|
func (s *Store) GetProfile(id int64) (*model.ServerProfile, error) {
|
|
p := &model.ServerProfile{}
|
|
var last sql.NullTime
|
|
err := s.db.QueryRow(
|
|
`SELECT id, name, protocol, host, server_ips, port, path,
|
|
username, auth_mode, routing_mode,
|
|
custom_cidrs, mtu_override, auto_connect, created_at, last_connected_at
|
|
FROM server_profiles WHERE id = ?`, id,
|
|
).Scan(&p.ID, &p.Name, &p.Protocol, &p.Host, &p.ServerIPs, &p.Port, &p.Path,
|
|
&p.Username, &p.AuthMode, &p.RoutingMode, &p.CustomCIDRs, &p.MTUOverride,
|
|
&p.AutoConnect, &p.CreatedAt, &last)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("get profile %d: %w", id, err)
|
|
}
|
|
if last.Valid {
|
|
p.LastConnectedAt = &last.Time
|
|
}
|
|
return p, nil
|
|
}
|
|
|
|
// ListProfiles returns all saved profiles ordered by name.
|
|
func (s *Store) ListProfiles() ([]model.ServerProfile, error) {
|
|
rows, err := s.db.Query(
|
|
`SELECT id, name, protocol, host, server_ips, port, path,
|
|
username, auth_mode, routing_mode,
|
|
custom_cidrs, mtu_override, auto_connect, created_at, last_connected_at
|
|
FROM server_profiles ORDER BY name`)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("list profiles: %w", err)
|
|
}
|
|
defer rows.Close()
|
|
|
|
var out []model.ServerProfile
|
|
for rows.Next() {
|
|
var p model.ServerProfile
|
|
var last sql.NullTime
|
|
if err := rows.Scan(&p.ID, &p.Name, &p.Protocol, &p.Host, &p.ServerIPs,
|
|
&p.Port, &p.Path, &p.Username, &p.AuthMode, &p.RoutingMode,
|
|
&p.CustomCIDRs, &p.MTUOverride, &p.AutoConnect, &p.CreatedAt, &last); err != nil {
|
|
return nil, err
|
|
}
|
|
if last.Valid {
|
|
p.LastConnectedAt = &last.Time
|
|
}
|
|
out = append(out, p)
|
|
}
|
|
return out, rows.Err()
|
|
}
|
|
|
|
// UpdateProfile updates an existing profile.
|
|
func (s *Store) UpdateProfile(p *model.ServerProfile) error {
|
|
_, err := s.db.Exec(
|
|
`UPDATE server_profiles SET
|
|
name = ?, protocol = ?, host = ?, server_ips = ?, port = ?, path = ?,
|
|
username = ?, auth_mode = ?, routing_mode = ?,
|
|
custom_cidrs = ?, mtu_override = ?, auto_connect = ?
|
|
WHERE id = ?`,
|
|
p.Name, p.Protocol, p.Host, p.ServerIPs, p.Port, p.Path,
|
|
p.Username, p.AuthMode, p.RoutingMode,
|
|
p.CustomCIDRs, p.MTUOverride, p.AutoConnect, p.ID)
|
|
if err != nil {
|
|
return fmt.Errorf("update profile %d: %w", p.ID, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// DeleteProfile removes a profile by ID.
|
|
func (s *Store) DeleteProfile(id int64) error {
|
|
_, err := s.db.Exec(`DELETE FROM server_profiles WHERE id = ?`, id)
|
|
if err != nil {
|
|
return fmt.Errorf("delete profile %d: %w", id, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// TouchLastConnected records the connection timestamp for a profile.
|
|
func (s *Store) TouchLastConnected(id int64) error {
|
|
_, err := s.db.Exec(
|
|
`UPDATE server_profiles SET last_connected_at = ? WHERE id = ?`,
|
|
time.Now(), id)
|
|
return err
|
|
}
|