tokeninit.go RawHistoryFolder

1package server
2
3import (
4	"crypto/rand"
5	"encoding/hex"
6	"fmt"
7)
8
9// HandleNewInitialization checks if there is ANY user in the database.
10func HandleNewTokenInitialization(a *App) (string, error) {
11	// Check if there is any user in the database
12	query := "SELECT COUNT(*) FROM user"
13	var count int
14	err := a.DB.QueryRow(query).Scan(&count)
15	if err != nil {
16		return "", fmt.Errorf("failed to check for admin user: %v", err)
17	}
18
19	// If count is greater than 0, it means there is at least one user in the database
20	if count > 0 {
21		return "", fmt.Errorf("user already present, skipping token initialization")
22	}
23
24	// If we reach here, it means admin does not exist, so we generate a token
25	token, err := generateRandomToken(16) // 16 bytes
26	if err != nil {
27		return "", fmt.Errorf("failed to generate token: %v", err)
28	}
29
30	// url = config.Http.SubDomain + config.Http.TopDomain + config.Http.Port + config.Admin.Path
31
32	fmt.Printf("Login with the new token: %s\n", token)
33
34	// Store the token in the config file
35	a.Config.General.Token = token
36	err = a.Config.SaveConfig()
37	if err != nil {
38		return "", fmt.Errorf("failed to save config: %v", err)
39	}
40	return token, nil
41}
42
43// generateRandomToken generates a random token of the given byte length
44func generateRandomToken(length int) (string, error) {
45	bytes := make([]byte, length)
46	if _, err := rand.Read(bytes); err != nil {
47		return "", err
48	}
49	return hex.EncodeToString(bytes), nil
50}