|
|
@@ -2,6 +2,7 @@ package tool
|
|
|
|
|
|
import (
|
|
|
"database/sql"
|
|
|
+ "log"
|
|
|
"regexp"
|
|
|
"strconv"
|
|
|
)
|
|
|
@@ -22,7 +23,7 @@ func Get_level(db *sql.DB, db_set map[string]string, ip string) []string {
|
|
|
|
|
|
stmt, err := db.Prepare(DB_change(db_set, "select data from user_set where id = ? and name = 'level'"))
|
|
|
if err != nil {
|
|
|
- return []string{"", "", ""}
|
|
|
+ log.Fatal(err)
|
|
|
}
|
|
|
defer stmt.Close()
|
|
|
|
|
|
@@ -31,13 +32,13 @@ func Get_level(db *sql.DB, db_set map[string]string, ip string) []string {
|
|
|
if err == sql.ErrNoRows {
|
|
|
level = "0"
|
|
|
} else {
|
|
|
- return []string{"", "", ""}
|
|
|
+ log.Fatal(err)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
stmt, err = db.Prepare(DB_change(db_set, "select data from user_set where id = ? and name = 'experience'"))
|
|
|
if err != nil {
|
|
|
- return []string{"", "", ""}
|
|
|
+ log.Fatal(err)
|
|
|
}
|
|
|
defer stmt.Close()
|
|
|
|
|
|
@@ -46,7 +47,7 @@ func Get_level(db *sql.DB, db_set map[string]string, ip string) []string {
|
|
|
if err == sql.ErrNoRows {
|
|
|
exp = "0"
|
|
|
} else {
|
|
|
- return []string{"", "", ""}
|
|
|
+ log.Fatal(err)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
@@ -56,13 +57,13 @@ func Get_level(db *sql.DB, db_set map[string]string, ip string) []string {
|
|
|
return []string{level, exp, max_exp}
|
|
|
}
|
|
|
|
|
|
-func Get_admin_auth(db *sql.DB, db_set map[string]string, ip string) string {
|
|
|
+func Get_user_auth(db *sql.DB, db_set map[string]string, ip string) string {
|
|
|
if !IP_or_user(ip) {
|
|
|
var auth string
|
|
|
|
|
|
stmt, err := db.Prepare(DB_change(db_set, "select data from user_set where id = ? and name = 'acl'"))
|
|
|
if err != nil {
|
|
|
- return ""
|
|
|
+ log.Fatal(err)
|
|
|
}
|
|
|
defer stmt.Close()
|
|
|
|
|
|
@@ -71,7 +72,7 @@ func Get_admin_auth(db *sql.DB, db_set map[string]string, ip string) string {
|
|
|
if err == sql.ErrNoRows {
|
|
|
auth = "user"
|
|
|
} else {
|
|
|
- return ""
|
|
|
+ log.Fatal(err)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
@@ -85,6 +86,35 @@ func Get_admin_auth(db *sql.DB, db_set map[string]string, ip string) string {
|
|
|
return ""
|
|
|
}
|
|
|
|
|
|
+func Get_auth_group_info(db *sql.DB, db_set map[string]string, auth string) map[string]bool {
|
|
|
+ stmt, err := db.Prepare(DB_change(db_set, "select name from alist where name = ?"))
|
|
|
+ if err != nil {
|
|
|
+ log.Fatal(err)
|
|
|
+ }
|
|
|
+ defer stmt.Close()
|
|
|
+
|
|
|
+ rows, err := stmt.Query(auth)
|
|
|
+ if err != nil {
|
|
|
+ log.Fatal(err)
|
|
|
+ }
|
|
|
+ defer rows.Close()
|
|
|
+
|
|
|
+ data_list := map[string]bool{}
|
|
|
+
|
|
|
+ for rows.Next() {
|
|
|
+ var name string
|
|
|
+
|
|
|
+ err := rows.Scan(&name)
|
|
|
+ if err != nil {
|
|
|
+ log.Fatal(err)
|
|
|
+ }
|
|
|
+
|
|
|
+ data_list[name] = true
|
|
|
+ }
|
|
|
+
|
|
|
+ return data_list
|
|
|
+}
|
|
|
+
|
|
|
func IP_preprocess(db *sql.DB, db_set map[string]string, ip string, my_ip string) []string {
|
|
|
var ip_view string
|
|
|
var user_name_view string
|
|
|
@@ -94,7 +124,7 @@ func IP_preprocess(db *sql.DB, db_set map[string]string, ip string, my_ip string
|
|
|
if err == sql.ErrNoRows {
|
|
|
ip_view = ""
|
|
|
} else {
|
|
|
- return []string{"", ""}
|
|
|
+ log.Fatal(err)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
@@ -103,11 +133,11 @@ func IP_preprocess(db *sql.DB, db_set map[string]string, ip string, my_ip string
|
|
|
if err == sql.ErrNoRows {
|
|
|
user_name_view = ""
|
|
|
} else {
|
|
|
- return []string{"", ""}
|
|
|
+ log.Fatal(err)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
- if Get_admin_auth(db, db_set, my_ip) != "" {
|
|
|
+ if Get_user_auth(db, db_set, my_ip) != "" {
|
|
|
ip_view = ""
|
|
|
user_name_view = ""
|
|
|
}
|
|
|
@@ -125,7 +155,7 @@ func IP_preprocess(db *sql.DB, db_set map[string]string, ip string, my_ip string
|
|
|
|
|
|
stmt, err := db.Prepare(DB_change(db_set, "select data from user_set where id = ? and name = 'sub_user_name'"))
|
|
|
if err != nil {
|
|
|
- return []string{"", ""}
|
|
|
+ log.Fatal(err)
|
|
|
}
|
|
|
defer stmt.Close()
|
|
|
|
|
|
@@ -134,7 +164,7 @@ func IP_preprocess(db *sql.DB, db_set map[string]string, ip string, my_ip string
|
|
|
if err == sql.ErrNoRows {
|
|
|
sub_user_name = Get_language(db, db_set, "member", false)
|
|
|
} else {
|
|
|
- return []string{"", ""}
|
|
|
+ log.Fatal(err)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
@@ -149,7 +179,7 @@ func IP_preprocess(db *sql.DB, db_set map[string]string, ip string, my_ip string
|
|
|
|
|
|
stmt, err := db.Prepare(DB_change(db_set, "select data from user_set where name = 'user_name' and id = ?"))
|
|
|
if err != nil {
|
|
|
- return []string{"", ""}
|
|
|
+ log.Fatal(err)
|
|
|
}
|
|
|
defer stmt.Close()
|
|
|
|
|
|
@@ -158,7 +188,7 @@ func IP_preprocess(db *sql.DB, db_set map[string]string, ip string, my_ip string
|
|
|
if err == sql.ErrNoRows {
|
|
|
user_name = ip
|
|
|
} else {
|
|
|
- return []string{"", ""}
|
|
|
+ log.Fatal(err)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
@@ -194,7 +224,7 @@ func IP_parser(db *sql.DB, db_set map[string]string, ip string, my_ip string) st
|
|
|
if err == sql.ErrNoRows {
|
|
|
user_name_level = ""
|
|
|
} else {
|
|
|
- return ""
|
|
|
+ log.Fatal(err)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
@@ -207,7 +237,7 @@ func IP_parser(db *sql.DB, db_set map[string]string, ip string, my_ip string) st
|
|
|
|
|
|
stmt, err := db.Prepare(DB_change(db_set, "select data from user_set where name = 'user_title' and id = ?"))
|
|
|
if err != nil {
|
|
|
- return ""
|
|
|
+ log.Fatal(err)
|
|
|
}
|
|
|
defer stmt.Close()
|
|
|
|
|
|
@@ -216,11 +246,11 @@ func IP_parser(db *sql.DB, db_set map[string]string, ip string, my_ip string) st
|
|
|
if err == sql.ErrNoRows {
|
|
|
user_title = ""
|
|
|
} else {
|
|
|
- return ""
|
|
|
+ log.Fatal(err)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
- if Get_admin_auth(db, db_set, ip) != "" {
|
|
|
+ if Get_user_auth(db, db_set, ip) != "" {
|
|
|
ip = "<b>" + ip + "</b>"
|
|
|
}
|
|
|
|