notatio/database.go

package main

import (
	"fmt"
	"path/filepath"
	"strings"
	"time"
)

// insertFileIntoDatabase inserts a file into the database with the provided details.
func insertFileIntoDatabase(username, filename string, creationTime int64, lastEdited int64, lastOpened int64) error {
	// Retrieve the user ID based on the username.
	userID, err := getUserUUID(username)
	if err != nil {
		return err
	}

	// Insert the file details into the database.
	_, err = db.Exec("INSERT INTO files (user_id, filename, creation_time, last_edited, last_opened) VALUES ($1, $2, $3, $4, $5)",
		userID, filename, time.Unix(creationTime, 0), time.Unix(lastEdited, 0), time.Unix(lastOpened, 0))
	return err
}

// insertUserIntoDatabase inserts a user into the database with the provided details.
func insertUserIntoDatabase(username, hashedPassword string, accountType string, userUUID string, name string, email string) error {
	// Insert the user details into the database.
	_, err := db.Exec("INSERT INTO users (username, password, accountType, uuid, name, email) VALUES ($1, $2, $3, $4, $5, $6)", username, hashedPassword, accountType, userUUID, name, email)
	return err
}

// deleteFileFromDatabase deletes a file from the database for the specified user and filename.
func deleteFileFromDatabase(username, filename string) error {
	// Retrieve the user UUID based on the username.
	userUUID, err := getUserUUID(username)
	if err != nil {
		return err
	}

	// Delete the file from the database.
	_, err = db.Exec("DELETE FROM files WHERE user_id = $1 AND filename = $2", userUUID, filename)
	return err
}

// UpdateEditedTimestamp updates the last_edited timestamp for a file in the database.
func UpdateEditedTimestamp(username, filename string) error {
	// Retrieve the user UUID based on the username.
	userUUID, err := getUserUUID(username)
	if err != nil {
		return err
	}

	// Update the last_edited timestamp for the file.
	_, err = db.Exec("UPDATE files SET last_edited = $1 WHERE user_id = $2 AND filename = $3", time.Now(), userUUID, filename)
	return err
}

// getName retrieves the name of a user based on their username.
func getName(username string) (string, error) {
	// Retrieve the user UUID based on the username.
	userUUID, err := getUserUUID(username)
	if err != nil {
		return "", err
	}

	// Retrieve the name of the user from the database.
	var name string
	err = db.QueryRow("SELECT name FROM users WHERE uuid = $1", userUUID).Scan(&name)
	if err != nil {
		return "", err
	}

	return name, nil
}

// updateFilename updates the filename for a file in the database.
func updateFilename(username, oldFilename, newFilename string) error {
	// Retrieve the user UUID based on the username.
	userUUID, err := getUserUUID(username)
	if err != nil {
		return err
	}

	// Check if the new filename already exists.
	var count int
	err = db.QueryRow("SELECT COUNT(*) FROM files WHERE user_id = $1 AND filename = $2", userUUID, newFilename).Scan(&count)
	if err != nil {
		return err
	}

	// If a file with the new filename already exists, append a number to the end.
	if count > 0 {
		countSuffix := 1
		updatedFilename := newFilename
		extension := filepath.Ext(newFilename)
		filenameWithoutExt := strings.TrimSuffix(newFilename, extension)

		// Keep incrementing the countSuffix until a unique filename is found.
		for count > 0 {
			updatedFilename = fmt.Sprintf("%s_%d%s", filenameWithoutExt, countSuffix, extension)

			// Check if the updatedFilename already exists.
			err = db.QueryRow("SELECT COUNT(*) FROM files WHERE user_id = $1 AND filename = $2", userUUID, updatedFilename).Scan(&count)
			if err != nil {
				return err
			}

			countSuffix++
		}

		newFilename = updatedFilename
	}

	// Update the filename in the database.
	_, err = db.Exec("UPDATE files SET filename = $1 WHERE user_id = $2 AND filename = $3", newFilename, userUUID, oldFilename)
	return err
}

// isUsernameTaken checks if a username is already taken.
func isUsernameTaken(username string) (bool, error) {
	username = strings.ToLower(username)
	userUUID, err := getUserUUID(username)
	if err != nil {
		// Username is not taken.
		fmt.Println("Username not taken:", username)
		return false, nil
	}

	// Username is taken.
	fmt.Println("Username taken by user with UUID:", userUUID)
	return true, nil
}