mirror of
https://github.com/JuLi0n21/pwa-player.git
synced 2026-04-19 23:40:05 +00:00
137 lines
3.1 KiB
Go
137 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"log"
|
|
"time"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
type User struct {
|
|
UserID int `json:"id"`
|
|
Name string `json:"name"`
|
|
AvatarUrl string `json:"avatar_url"`
|
|
EndPoint string `json:"endpoint"`
|
|
Share bool `json:"share"`
|
|
Token `json:"-"`
|
|
}
|
|
|
|
type Token struct {
|
|
AccessToken string `json:"-"`
|
|
RefreshToken string `json:"-"`
|
|
ExpireDate time.Time `json:"-"`
|
|
}
|
|
|
|
var db *sql.DB
|
|
|
|
const (
|
|
layout = "2006-01-02 15:04:05.999999999-07:00"
|
|
)
|
|
|
|
func InitDB() {
|
|
var err error
|
|
db, err = sql.Open("sqlite3", "database.db")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
//users
|
|
u := `
|
|
CREATE TABLE IF NOT EXISTS users (
|
|
id INTEGER PRIMARY KEY NOT NULL,
|
|
name TEXT NOT NULL,
|
|
endpoint TEXT,
|
|
avatar_url TEXT,
|
|
access_token TEXT,
|
|
refresh_token TEXT,
|
|
expire_date TEXT
|
|
);`
|
|
|
|
_, err = db.Exec(u)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
//cookiejar
|
|
a := `
|
|
CREATE TABLE IF NOT EXISTS cookiejar (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
user_id INTEGER,
|
|
cookie TEXT NOT NULL UNIQUE,
|
|
FOREIGN KEY(user_id) REFERENCES users(id) On DELETE CASCADE
|
|
);`
|
|
|
|
_, err = db.Exec(a)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
}
|
|
|
|
func GetUserByCookie(cookie string) (*User, error) {
|
|
|
|
query := `
|
|
SELECT users.id, users.name, users.endpoint, users.avatar_url, users.access_token, users.refresh_token, users.expire_date
|
|
FROM users users
|
|
JOIN cookiejar co ON users.id = co.user_id
|
|
WHERE co.cookie = ?`
|
|
row := db.QueryRow(query, cookie)
|
|
|
|
var user User
|
|
var ExpireStr string
|
|
err := row.Scan(&user.UserID, &user.Name, &user.EndPoint, &user.AvatarUrl, &user.AccessToken, &user.RefreshToken, &ExpireStr)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return &User{}, err
|
|
}
|
|
|
|
user.ExpireDate, err = time.Parse(layout, ExpireStr)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return &User{}, err
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
func SaveUser(user User) error {
|
|
query := `INSERT INTO users (id, name, endpoint, avatar_url, access_token, refresh_token, expire_date) VALUES (?, ?, ?, ?, ?, ?, ?)
|
|
ON CONFLICT(id) DO UPDATE SET
|
|
name = excluded.name,
|
|
endpoint = excluded.endpoint,
|
|
avatar_url = excluded.avatar_url,
|
|
access_token = excluded.access_token,
|
|
refresh_token = excluded.refresh_token,
|
|
expire_date = excluded.expire_date;
|
|
`
|
|
|
|
_, err := db.Exec(query, user.UserID, user.Name, user.EndPoint, user.AvatarUrl, user.AccessToken, user.RefreshToken, user.ExpireDate)
|
|
return err
|
|
}
|
|
|
|
func SaveCookie(userID int, cookie string) error {
|
|
query := "INSERT INTO cookiejar (user_id, cookie) VALUES (?, ?)"
|
|
_, err := db.Exec(query, userID, cookie)
|
|
return err
|
|
}
|
|
|
|
func UpdateUserTokens(userID int, auth Token) error {
|
|
query := "UPDATE users SET auth_token = ?, refresh_token = ?, expire_date = ? WHERE id = ?"
|
|
_, err := db.Exec(query, auth.AccessToken, auth.RefreshToken, auth.ExpireDate, userID)
|
|
return err
|
|
}
|
|
|
|
func UpdateUserEndPoint(userID int, endPoint string) error {
|
|
query := "UPDATE users SET endpoint = ? WHERE id = ?"
|
|
_, err := db.Exec(query, endPoint, userID)
|
|
return err
|
|
}
|
|
|
|
func UpdateUserShare(userID int, sharing bool) error {
|
|
query := "UPDATE users SET sharing = ? WHERE id = ?"
|
|
_, err := db.Exec(query, sharing, userID)
|
|
return err
|
|
}
|