Matheus Sampaio Queiroga
ad65c0a9b0
- Removed old SQLite user schema and replaced it with a new structure that includes password handling and cookie management. - Updated user.go to implement password encryption and user creation logic. - Modified web templates to reflect the new server management system, including the removal of outdated templates and the addition of new server creation and listing functionalities. - Introduced error handling templates for better user feedback on bad requests and server errors. - Added a makefile for easier database setup and management with Docker. Signed-off-by: Matheus Sampaio Queiroga <srherobrine20@gmail.com>
77 lines
2.1 KiB
Go
77 lines
2.1 KiB
Go
package user
|
|
|
|
import (
|
|
"database/sql"
|
|
_ "embed"
|
|
|
|
"sirherobrine23.com.br/go-bds/bds/modules/datas/permission"
|
|
)
|
|
|
|
//go:embed sql/sqlite_create.sql
|
|
var sqliteTableCreate string
|
|
|
|
// Create table on start
|
|
func CreateSqliteTable(conn *sql.DB) error {
|
|
_, err := conn.Exec(sqliteTableCreate)
|
|
return err
|
|
}
|
|
|
|
func (p *Password) sqliteStorage(passwordHash string) error {
|
|
var count int
|
|
err := p.DB.QueryRow("SELECT count(*) FROM password WHERE user_id == $1 LIMIT 1", p.ID).Scan(&count)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if count == 1 {
|
|
_, err = p.DB.Exec("UPDATE password SET password = $2 WHERE user_id = $1", p.ID, passwordHash)
|
|
return err
|
|
}
|
|
|
|
_, err = p.DB.Exec("INSERT INTO password (user_id, password) VALUES ($1, $2)", p.ID, passwordHash)
|
|
return err
|
|
}
|
|
|
|
func (s *UserSearch) processSqliteRow(row *sql.Row) (*User, error) {
|
|
if row.Err() == nil {
|
|
user := &User{}
|
|
if err := row.Scan(&user.ID, &user.Name, &user.Username, &user.Permission); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
row = s.DB.QueryRow("SELECT password FROM password WHERE user_id = $1", user.ID)
|
|
if row.Err() == nil {
|
|
user.Password = &Password{DB: s.DB, Driver: s.Driver}
|
|
if err := row.Scan(&user.Password.PasswordHash); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return user, nil
|
|
}
|
|
return nil, row.Err()
|
|
}
|
|
|
|
func (s *UserSearch) sqliteByID(id int64) (*User, error) {
|
|
return s.processSqliteRow(s.DB.QueryRow("SELECT id, \"name\", username, permission FROM user WHERE id = $1", id))
|
|
}
|
|
|
|
func (s *UserSearch) sqliteUsername(username string) (*User, error) {
|
|
return s.processSqliteRow(s.DB.QueryRow("SELECT id, \"name\", username, permission FROM user WHERE username = $1", username))
|
|
}
|
|
|
|
func (s *UserSearch) sqliteCreate(name, username, email, passwordHASH string) (*User, error) {
|
|
res, err := s.DB.Exec("INSERT INTO user(\"name\", username, email, permission) VALUES ($1, $2, $3, $4)", name, username, email, permission.Unknown)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
userID, err := res.LastInsertId()
|
|
if err != nil {
|
|
return nil, err
|
|
} else if _, err = s.DB.Exec("INSERT INTO password(user_id, password) VALUES ($1, $2)", userID, passwordHASH); err != nil {
|
|
return nil, err
|
|
}
|
|
return s.ByID(userID)
|
|
}
|