Initial commit

handler/user.go

@@ -0,0 +1,87 @@
+package handler
+
+import (
+	"Ratte/common/slices"
+	"fmt"
+	"github.com/Yuzuki616/Ratte-Interface/core"
+	"github.com/Yuzuki616/Ratte-Interface/panel"
+)
+
+func compareUserList(old, new []panel.UserInfo) (deleted []string, added []panel.UserInfo) {
+	tmp := map[string]struct{}{}
+	tmp2 := map[string]struct{}{}
+	for i := range old {
+		tmp[old[i].GetHashOrKey()] = struct{}{}
+	}
+	l := len(tmp)
+	for i := range new {
+		e := new[i].GetHashOrKey()
+		tmp[e] = struct{}{}
+		tmp2[e] = struct{}{}
+		if l != len(tmp) {
+			added = append(added, new[i])
+			l++
+		}
+	}
+	tmp = nil
+	l = len(tmp2)
+	for i := range old {
+		tmp2[old[i].GetHashOrKey()] = struct{}{}
+		if l != len(tmp2) {
+			deleted = append(deleted, old[i].Name)
+			l++
+		}
+	}
+	return deleted, added
+}
+
+func (h *Handler) PullUserHandle(users []panel.UserInfo) error {
+	del, add := compareUserList(h.userList, users)
+	cas := slices.RangeToNew[panel.UserInfo, core.UserInfo](add, func(_ int, v panel.UserInfo) core.UserInfo {
+		return core.UserInfo(v.UserInfo)
+	})
+	err := h.c.AddUsers(&core.AddUsersParams{
+		NodeName: h.nodeName,
+		Users:    cas,
+	})
+	if err != nil {
+		return fmt.Errorf("add user error: %w", err)
+	}
+	h.l.Infof("Added %d users", len(users))
+	err = h.c.DelUsers(&core.DelUsersParams{
+		NodeName: h.nodeName,
+		Users:    del,
+	})
+	if err != nil {
+		return fmt.Errorf("del user error: %w", err)
+	}
+	h.l.Infof("Deleted %d users", len(users))
+	h.userList = users
+	return nil
+}
+
+func (h *Handler) ReportUserHandle(id int) error {
+	var err error
+	req := &core.GetUserTrafficParams{NodeName: h.nodeName}
+	var users []panel.UserTrafficInfo
+	slices.Range(h.userList, func(_ int, v panel.UserInfo) bool {
+		req.Username = v.Name
+		rsp := h.c.GetUserTraffic(req)
+		if rsp.Err != nil {
+			err = rsp.Err
+			return true
+		}
+		if rsp.Up == 0 && rsp.Down == 0 {
+			return false
+		}
+		return false
+	})
+	err = h.p.ReportUserTraffic(&panel.ReportUserTrafficParams{
+		Id:    id,
+		Users: users,
+	})
+	if err != nil {
+		return fmt.Errorf("report user error: %w", err)
+	}
+	return nil
+}