package web import ( "encoding/csv" "fmt" "net" "net/http" "strconv" "strings" "time" "daili/internal/domain" "daili/internal/store" ) type logsData struct { List []store.OpLog Pager Pager Q string Total int } func (s *Server) handleLogs(w http.ResponseWriter, r *http.Request) { q := r.URL.Query().Get("q") pageNo := s.queryInt(r, "page", 1) if pageNo < 1 { pageNo = 1 } list, total, err := s.db.ListLogs(q, pageSize, (pageNo-1)*pageSize) if err != nil { http.Error(w, "读取日志失败", http.StatusInternalServerError) return } pager := newPager(r, pageNo, pageSize, total) if pager.Page != pageNo { list, total, err = s.db.ListLogs(q, pageSize, (pager.Page-1)*pageSize) if err != nil { http.Error(w, "读取日志失败", http.StatusInternalServerError) return } } p := s.newPage(r, "操作日志", "logs") p.D = &logsData{List: list, Pager: pager, Q: q, Total: total} s.render(w, r, "logs", p) } type settingsData struct { WarnDays int ScreenRefresh int Users []store.User Backups []string DBPath string BackupDir string BackupKeep int BackupEnabled bool ConfigPath string ScreenPublic bool Addrs []string Port string Uptime string LogCount int Version string CustomerCount int PackageCount int } func humanDuration(d time.Duration) string { secs := int(d.Seconds()) days := secs / 86400 hours := (secs % 86400) / 3600 mins := (secs % 3600) / 60 switch { case days > 0: return fmt.Sprintf("%d 天 %d 小时", days, hours) case hours > 0: return fmt.Sprintf("%d 小时 %d 分钟", hours, mins) default: return fmt.Sprintf("%d 分钟", mins) } } func LocalAddrs() []string { out := []string{} ifaces, err := net.Interfaces() if err != nil { return out } for _, iface := range ifaces { if iface.Flags&net.FlagUp == 0 || iface.Flags&net.FlagLoopback != 0 { continue } addrs, err := iface.Addrs() if err != nil { continue } for _, a := range addrs { var ip net.IP switch v := a.(type) { case *net.IPNet: ip = v.IP case *net.IPAddr: ip = v.IP } if ip == nil { continue } ip4 := ip.To4() if ip4 == nil || ip4.IsLoopback() { continue } ipStr := ip4.String() if strings.HasPrefix(ipStr, "169.254.") { continue } out = append(out, ipStr) } } return out } func (s *Server) handleSettings(w http.ResponseWriter, r *http.Request) { users, _ := s.db.ListUsers() backups, _ := s.db.ListBackups() custN, _ := s.db.CustomerCount() pkgs, _ := s.db.ListPackages(true) logN, _ := s.db.LogCount() port := s.cfg.Addr if strings.HasPrefix(port, ":") { port = port[1:] } data := &settingsData{ WarnDays: s.warnDays(), ScreenRefresh: s.screenRefresh(), Users: users, Backups: backups, DBPath: s.db.Path, BackupDir: s.cfg.BackupDir, BackupKeep: s.cfg.BackupKeep, BackupEnabled: s.cfg.BackupEnabled, ConfigPath: s.cfg.ConfigPath, ScreenPublic: s.cfg.ScreenPublic, Addrs: LocalAddrs(), Port: port, Uptime: humanDuration(time.Since(s.started)), LogCount: logN, Version: s.cfg.Version, CustomerCount: custN, PackageCount: len(pkgs), } p := s.newPage(r, "设置", "settings") p.D = data s.render(w, r, "settings", p) } // handleUserColor 设置管理员(打手)颜色。 func (s *Server) handleUserColor(w http.ResponseWriter, r *http.Request) { id := pathID(r) color, err := store.NormalizeColor(r.PostFormValue("color")) if err != nil { s.redirect(w, r, "/settings", "颜色格式不正确") return } if err := s.db.SetUserColor(id, color); err != nil { s.redirect(w, r, "/settings", "保存颜色失败:"+err.Error()) return } name := "" if u, err := s.db.UserByID(id); err == nil { name = u.DisplayName if name == "" { name = u.Username } } s.logOp(r, "设置打手颜色", "user", id, name+" → "+color) s.redirect(w, r, "/settings", "已更新 "+name+" 的颜色") } func (s *Server) handleSettingsSave(w http.ResponseWriter, r *http.Request) { warn := s.formInt(r, "warn_days", 3) if warn < 0 { warn = 0 } if warn > 60 { warn = 60 } refresh := s.formInt(r, "screen_refresh", 30) if refresh < 5 { refresh = 5 } if refresh > 600 { refresh = 600 } _ = s.db.SetSetting("warn_days", strconv.Itoa(warn)) _ = s.db.SetSetting("screen_refresh", strconv.Itoa(refresh)) s.logOp(r, "修改设置", "settings", 0, fmt.Sprintf("预警 %d 天、大屏刷新 %d 秒", warn, refresh)) s.redirect(w, r, "/settings", "设置已保存") } func (s *Server) handlePasswordChange(w http.ResponseWriter, r *http.Request) { u := s.currentUser(r) old := r.PostFormValue("old_password") pw := r.PostFormValue("password") pw2 := r.PostFormValue("password2") if u == nil { http.Redirect(w, r, "/login", http.StatusFound) return } if _, err := s.db.Authenticate(u.Username, old); err != nil { s.redirect(w, r, "/settings", "原密码不正确") return } if len(pw) < 6 { s.redirect(w, r, "/settings", "新密码至少 6 位") return } if pw != pw2 { s.redirect(w, r, "/settings", "两次输入的新密码不一致") return } if err := s.db.UpdatePassword(u.ID, pw); err != nil { s.redirect(w, r, "/settings", "修改失败:"+err.Error()) return } s.logOp(r, "修改密码", "user", u.ID, "管理员 "+u.Username+" 修改了自己的密码") s.redirect(w, r, "/settings", "密码已更新") } func (s *Server) handleUserCreate(w http.ResponseWriter, r *http.Request) { username := strings.TrimSpace(r.PostFormValue("username")) display := strings.TrimSpace(r.PostFormValue("display_name")) pw := r.PostFormValue("password") switch { case len(username) < 3: s.redirect(w, r, "/settings", "用户名至少 3 个字符") return case len(pw) < 6: s.redirect(w, r, "/settings", "密码至少 6 位") return } if display == "" { display = username } u, err := s.db.CreateUser(username, display, pw) if err != nil { if err == store.ErrDuplicate { s.redirect(w, r, "/settings", "用户名已存在") return } s.redirect(w, r, "/settings", "创建失败:"+err.Error()) return } s.logOp(r, "新增管理员", "user", u.ID, "新增管理员 "+username) s.redirect(w, r, "/settings", "管理员 "+username+" 已创建") } func (s *Server) handleUserToggle(w http.ResponseWriter, r *http.Request) { id := pathID(r) me := s.currentUser(r) if me != nil && me.ID == id { s.redirect(w, r, "/settings", "不能停用自己的账户") return } target, err := s.db.UserByID(id) if err != nil { http.NotFound(w, r) return } if err := s.db.SetUserActive(id, !target.IsActive); err != nil { s.redirect(w, r, "/settings", "操作失败:"+err.Error()) return } action := "停用账户" if !target.IsActive { action = "启用账户" } s.logOp(r, action, "user", id, action+" "+target.Username) s.redirect(w, r, "/settings", action+"成功") } func (s *Server) handleBackupNow(w http.ResponseWriter, r *http.Request) { path, err := s.db.BackupNow(s.cfg.BackupKeep) if err != nil { s.redirect(w, r, "/settings", "备份失败:"+err.Error()) return } s.logOp(r, "备份数据", "backup", 0, "手动备份 "+path) s.redirect(w, r, "/settings", "已备份:"+path) } func (s *Server) handleExport(w http.ResponseWriter, r *http.Request) { kind := strings.TrimSuffix(r.PathValue("kind"), ".csv") var set *store.ExportRowSet var err error switch kind { case "customers": set, err = s.db.ExportCustomers() case "subscriptions": set, err = s.db.ExportSubscriptions(domain.Today(), s.warnDays()) case "logs": set, err = s.db.ExportLogs() default: http.NotFound(w, r) return } if err != nil { http.Error(w, "导出失败:"+err.Error(), http.StatusInternalServerError) return } w.Header().Set("Content-Type", "text/csv; charset=utf-8") w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=\"daili-%s-%s.csv\"", kind, time.Now().Format("20060102"))) _, _ = w.Write([]byte{0xEF, 0xBB, 0xBF}) cw := csv.NewWriter(w) _ = cw.Write(set.Header) for _, row := range set.Rows { _ = cw.Write(row) } cw.Flush() s.logOp(r, "导出数据", "export", 0, "导出 "+kind+"("+strconv.Itoa(len(set.Rows))+" 行)") }