fix: restore default admin password and add reset flag
Restored 'admin123' as the default password in db init and added a -reset-admin flag to main.go.
This commit is contained in:
@@ -1,16 +1,22 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"flag"
|
||||||
"log"
|
"log"
|
||||||
|
"os"
|
||||||
|
|
||||||
"llm-proxy/internal/config"
|
"llm-proxy/internal/config"
|
||||||
"llm-proxy/internal/db"
|
"llm-proxy/internal/db"
|
||||||
"llm-proxy/internal/server"
|
"llm-proxy/internal/server"
|
||||||
|
|
||||||
"github.com/joho/godotenv"
|
"github.com/joho/godotenv"
|
||||||
|
"golang.org/x/crypto/bcrypt"
|
||||||
)
|
)
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
|
resetAdmin := flag.Bool("reset-admin", false, "Reset admin password to admin123")
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
// Load environment variables
|
// Load environment variables
|
||||||
if err := godotenv.Load(); err != nil {
|
if err := godotenv.Load(); err != nil {
|
||||||
log.Println("No .env file found")
|
log.Println("No .env file found")
|
||||||
@@ -28,6 +34,16 @@ func main() {
|
|||||||
log.Fatalf("Failed to initialize database: %v", err)
|
log.Fatalf("Failed to initialize database: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if *resetAdmin {
|
||||||
|
hash, _ := bcrypt.GenerateFromPassword([]byte("admin123"), 12)
|
||||||
|
_, err = database.Exec("UPDATE users SET password_hash = ?, must_change_password = 1 WHERE username = 'admin'", string(hash))
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Failed to reset admin password: %v", err)
|
||||||
|
}
|
||||||
|
log.Println("Admin password has been reset to 'admin123'")
|
||||||
|
os.Exit(0)
|
||||||
|
}
|
||||||
|
|
||||||
// Initialize server
|
// Initialize server
|
||||||
s := server.NewServer(cfg, database)
|
s := server.NewServer(cfg, database)
|
||||||
|
|
||||||
|
|||||||
@@ -159,7 +159,7 @@ func (db *DB) RunMigrations() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if count == 0 {
|
if count == 0 {
|
||||||
hash, err := bcrypt.GenerateFromPassword([]byte("admin"), 12)
|
hash, err := bcrypt.GenerateFromPassword([]byte("admin123"), 12)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("failed to hash default password: %w", err)
|
return fmt.Errorf("failed to hash default password: %w", err)
|
||||||
}
|
}
|
||||||
@@ -167,7 +167,7 @@ func (db *DB) RunMigrations() error {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("failed to insert default admin: %w", err)
|
return fmt.Errorf("failed to insert default admin: %w", err)
|
||||||
}
|
}
|
||||||
log.Println("Created default admin user with password 'admin' (must change on first login)")
|
log.Println("Created default admin user with password 'admin123' (must change on first login)")
|
||||||
}
|
}
|
||||||
|
|
||||||
// Default client
|
// Default client
|
||||||
|
|||||||
Reference in New Issue
Block a user