feat: 添加鉴权组件

auth/auth.go

@@ -1,5 +1,37 @@
 package auth
 
-func isRootUser(qqID string) bool {
+import (
+	"git.lxtend.com/qqbot/handler"
+	"git.lxtend.com/qqbot/model"
+	"git.lxtend.com/qqbot/sqlite3"
+)
+
+func init() {
+	authTableCreateSQL := `
+		CREATE TABLE IF NOT EXISTS auth (
+			id INTEGER PRIMARY KEY AUTOINCREMENT,
+			qqid TEXT UNIQUE,
+			role INTEGER
+		);
+	`
+	sqlite3.TryCreateTable(authTableCreateSQL)
+}
+
+func HasPermission(qqID int64, level int) bool {
+	auth, err := getAuth(qqID)
+	if err != nil {
 		return false
 	}
+	return auth.Role <= level
+}
+
+func TryExecHandler(msg model.Message, level int, handler handler.Handler) (reply model.Reply) {
+	if HasPermission(msg.UserId, level) {
+		return handler(msg)
+	}
+	return model.Reply{
+		ReplyMsg:       "权限验证失败",
+		ReferOriginMsg: true,
+		FromMsg:        msg,
+	}
+}

auth/model.go

@@ -0,0 +1,6 @@
+package auth
+
+type Auth struct {
+	Qqid string `json:"qqid" db:"qqid"`
+	Role int    `json:"role" db:"role"`
+}

auth/service.go

@@ -0,0 +1,24 @@
+package auth
+
+import (
+	"database/sql"
+
+	"git.lxtend.com/qqbot/constants"
+	"git.lxtend.com/qqbot/sqlite3"
+)
+
+func getAuth(qqID int64) (auth Auth, err error) {
+	tx, err := sqlite3.GetTran()
+	if err != nil {
+		return Auth{}, err
+	}
+	defer tx.Rollback()
+	err = tx.Get(&auth, "SELECT * FROM auth WHERE qqid = ?", qqID)
+	if err == sql.ErrNoRows {
+		_, err = tx.Exec("INSERT INTO auth(qqid, role) VALUES(?, ?)", qqID, constants.LEVEL_USER)
+	}
+	if err != nil {
+		return Auth{}, err
+	}
+	return auth, nil
+}

constants/auth.go

@@ -0,0 +1,8 @@
+package constants
+
+const (
+	LEVEL_ROOT   = 0
+	LEVEL_ADMIN  = 10
+	LEVEL_USER   = 20 //default
+	LEVEL_BANNED = 100
+)