initial commit

This commit is contained in:
wt committed 2025-05-27 18:20:05 +07:00
commit 55dbc36b92
9 files changed
+396

No files matched your search

+2
View File
@@ -0,0 +1,2 @@
users.db
go.sum
+97
View File
@@ -0,0 +1,97 @@
package api
import (
"encoding/json"
"fmt"
"form/database"
"form/model"
"maps"
"net/http"
"time"
"github.com/golang-jwt/jwt/v5"
)
const defaultsecretkey = "secret"
func createJWT(secretKey string, data map[string]interface{}, ttl time.Duration) (string, error) {
claims := jwt.MapClaims{
"exp": time.Now().Add(ttl).Unix(),
"iat": time.Now().Unix(),
"nbf": time.Now().Unix(),
}
maps.Copy(claims, data)
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
signedToken, err := token.SignedString([]byte(secretKey))
if err != nil {
return "", nil
}
return signedToken, nil
}
func VerifyJWT(tokenString string, secretKey string) (map[string]interface{}, error) {
token, e := jwt.Parse(
tokenString,
func(token *jwt.Token) (interface{}, error) {
if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok {
return nil, fmt.Errorf("unexpexted signing method: %v", token.Header["alg"])
}
return []byte(secretKey), nil
},
)
if e != nil {
return nil, fmt.Errorf("token validation failed: %w", e)
}
if claims, ok := token.Claims.(jwt.MapClaims); ok && token.Valid {
result := make(map[string]interface{})
for key, value := range claims {
if key == "exp" || key == "iat" || key == "nbf" {
continue
}
result[key] = value
}
return result, nil
}
return nil, fmt.Errorf("invalid token")
}
func Token(w http.ResponseWriter, req *http.Request) {
w.Header().Set("Content-Type", "application/json")
var request model.TokenRequest
decoder := json.NewDecoder(req.Body)
decoder.Decode(&request)
dbuser, e := database.GetUserByUsername(request.Username)
if e != nil {
fmt.Fprintf(w, "{\"status\":403,\"error\":\"%v\"}", e)
return
}
if dbuser.Password == request.Password {
responseUser := model.UserResponse{
Id: dbuser.Id,
Username: dbuser.Username,
Email: dbuser.Email,
Group: dbuser.Group,
IsSuperuser: dbuser.IsSuperuser,
}
responseUserMarshal, _ := json.Marshal(responseUser)
responseUserUnmarshal := make(map[string]interface{})
json.Unmarshal(responseUserMarshal, &responseUserUnmarshal)
token, _ := createJWT(defaultsecretkey, responseUserUnmarshal, time.Duration(24*time.Hour))
fmt.Fprintf(w, "{\"token\":\"%s\"}", token)
return
}
fmt.Fprintf(w, "{\"status\":\"403\"}")
}
func VerifyToken(w http.ResponseWriter, req *http.Request) {
w.Header().Set("Content-Type", "application/json")
var request model.Token
decoder := json.NewDecoder(req.Body)
decoder.Decode(&request)
_, e := VerifyJWT(request.Token, defaultsecretkey)
if e != nil {
fmt.Fprintf(w, "{\"status\":403}")
return
}
fmt.Fprintf(w, "{\"status\":200}")
}
+89
View File
@@ -0,0 +1,89 @@
package api
import (
"encoding/json"
"fmt"
"form/database"
"form/model"
"net/http"
)
func GetAllUsers(w http.ResponseWriter, req *http.Request) {
w.Header().Set("Content-Type", "application/json")
var request model.GetAllUsersRequest
decoder := json.NewDecoder(req.Body)
decoder.Decode(&request)
claims, e := VerifyJWT(request.Token, defaultsecretkey)
if e != nil || claims["is_superuser"] != true {
fmt.Fprintf(w, "{\"status\":403}")
return
}
users, e := database.GetAllUsers()
if e != nil {
fmt.Fprintf(w, "{\"error\":\"%v\"}", e)
return
}
var responseUsers []model.UserResponse
for i := range users {
responseUsers = append(responseUsers, model.UserResponse{
Id: users[i].Id,
Username: users[i].Username,
Email: users[i].Email,
Group: users[i].Group,
IsSuperuser: users[i].IsSuperuser,
})
}
jsondata, _ := json.Marshal(responseUsers)
fmt.Fprint(w, string(jsondata))
}
func GetUser(w http.ResponseWriter, req *http.Request) {
w.Header().Set("Content-Type", "application/json")
var request model.Token
decoder := json.NewDecoder(req.Body)
decoder.Decode(&request)
claims, e := VerifyJWT(request.Token, defaultsecretkey)
if e != nil {
fmt.Fprintf(w, "{\"status\":403}")
return
}
user, e := database.GetUserByUsername(fmt.Sprintf("%v", claims["username"]))
if e != nil {
fmt.Fprintf(w, "{\"error\":\"%v\"}", e)
return
}
responseUser := model.UserResponse{
Id: user.Id,
Username: user.Username,
Email: user.Email,
Group: user.Group,
IsSuperuser: user.IsSuperuser,
}
jsondata, _ := json.Marshal(responseUser)
fmt.Fprint(w, string(jsondata))
}
func ChangePassword(w http.ResponseWriter, req *http.Request) {
w.Header().Set("Content-Type", "application/json")
}
func DelUser(w http.ResponseWriter, req *http.Request) {
w.Header().Set("Content-Type", "application/json")
}
func AddUser(w http.ResponseWriter, req *http.Request) {
w.Header().Set("Content-Type", "application/json")
}
func RegisterUser(w http.ResponseWriter, req *http.Request) {
w.Header().Set("Content-Type", "application/json")
}
func ChangeUser(w http.ResponseWriter, req *http.Request) {
w.Header().Set("Content-Type", "application/json")
}
+25
View File
@@ -0,0 +1,25 @@
package database
import (
"fmt"
"form/model"
"gorm.io/driver/sqlite"
"gorm.io/gorm"
)
var db *gorm.DB
func Init() error {
var err error
db, err = gorm.Open(sqlite.Open("users.db"), &gorm.Config{})
if err != nil {
fmt.Println("failed to connect database")
return err
}
return nil
}
func Migrate() {
db.AutoMigrate(&model.User{})
}
+85
View File
@@ -0,0 +1,85 @@
package database
import (
"encoding/json"
"errors"
"fmt"
"form/model"
"os"
"strings"
"gorm.io/gorm"
)
func CreateUser(user *model.User) error {
e := db.Where(model.User{
Username: user.Username,
}).FirstOrCreate(&user).Error
if e != nil {
return e
}
return nil
}
func GetUserById(id string) (*model.User, error) {
var user model.User
e := db.First(&user, "id = ?", id).Error
if e != nil {
if errors.Is(e, gorm.ErrRecordNotFound) {
return nil, fmt.Errorf("user not found")
}
return nil, e
}
return &user, nil
}
func GetUserByUsername(username string) (*model.User, error) {
var user model.User
e := db.First(&user, "username = ?", username).Error
if e != nil {
if errors.Is(e, gorm.ErrRecordNotFound) {
return nil, fmt.Errorf("user not found")
}
return nil, e
}
return &user, nil
}
func DeleteUser(id string) error {
var user model.User
e := db.First(&user, "id = ?", id).Error
if e != nil {
if errors.Is(e, gorm.ErrRecordNotFound) {
return fmt.Errorf("user not found")
}
return e
}
e = db.Delete(&user).Error
if e != nil {
if errors.Is(e, gorm.ErrRecordNotFound) {
return fmt.Errorf("user not found")
}
return e
}
return nil
}
func GetUsersFromFile(filename string) []model.User {
data, e := os.ReadFile(filename)
if e != nil {
panic(e)
}
decoder := json.NewDecoder(strings.NewReader(string(data)))
var res []model.User
decoder.Decode(&res)
return res
}
func GetAllUsers() ([]model.User, error) {
var users []model.User
e := db.Find(&users).Error
if e != nil {
return nil, e
}
return users, nil
}
+16
View File
@@ -0,0 +1,16 @@
module form
go 1.24.3
require (
github.com/golang-jwt/jwt/v5 v5.2.2
gorm.io/driver/sqlite v1.5.7
gorm.io/gorm v1.30.0
)
require (
github.com/jinzhu/inflection v1.0.0 // indirect
github.com/jinzhu/now v1.1.5 // indirect
github.com/mattn/go-sqlite3 v1.14.28 // indirect
golang.org/x/text v0.25.0 // indirect
)
+30
View File
@@ -0,0 +1,30 @@
package main
import (
"form/api"
"form/database"
"net/http"
)
func main() {
database.Init()
database.Migrate()
users := database.GetUsersFromFile("users.json")
for i := range users {
database.CreateUser(&users[i])
}
http.HandleFunc("/api/token", api.Token)
http.HandleFunc("/api/token/verify", api.VerifyToken)
http.HandleFunc("/api/access", api.VerifyToken)
http.HandleFunc("/api/register", api.RegisterUser)
http.HandleFunc("/api/users", api.GetAllUsers)
http.HandleFunc("/api/users/user", api.GetUser)
http.HandleFunc("/api/users/password", api.ChangePassword)
http.HandleFunc("/api/users/change", api.ChangeUser)
http.HandleFunc("/api/users/del", api.DelUser)
http.HandleFunc("/api/user/add", api.AddUser)
http.ListenAndServe(":8090", nil)
}
+34
View File
@@ -0,0 +1,34 @@
package model
import "gorm.io/gorm"
type User struct {
gorm.Model
Id string `json:"id"`
Username string `json:"username"`
Email string `json:"email"`
Password string `json:"password"`
Group string `json:"group"`
IsSuperuser bool `json:"is_superuser"`
}
type UserResponse struct {
Id string `json:"id"`
Username string `json:"username"`
Email string `json:"email"`
Group string `json:"group"`
IsSuperuser bool `json:"is_superuser"`
}
type TokenRequest struct {
Username string `json:"username"`
Password string `json:"password"`
}
type Token struct {
Token string `json:"token"`
}
type GetAllUsersRequest struct {
Token string `json:"token"`
}
+18
View File
@@ -0,0 +1,18 @@
[
{
"id": "0",
"username": "tola",
"email": "mail@example.com",
"password": "2808",
"group": "root",
"is_superuser": true
},
{
"id": "1",
"username": "lisa",
"email": "lisa@example.com",
"password": "asdf",
"group": "users",
"is_superuser": false
}
]