feat: version 2

backend/store.go

@@ -2,17 +2,35 @@ package main
 
 import (
 	"fmt"
+	"os"
 	"time"
+
 	"github.com/jinzhu/gorm"
 	_ "github.com/jinzhu/gorm/dialects/sqlite"
-
-	"os"
 )
 
-type CodeURLMap struct {
-	Code      string    `gorm:"primary_key" json:"code"`
-	LURL      string    `json:"lurl" gorm:"column:lurl"`
+// User represents a registered account
+type User struct {
+	ID        uint      `gorm:"primary_key" json:"id"`
+	Username  string    `gorm:"unique_index;not null;size:32" json:"username"`
+	Password  string    `gorm:"not null" json:"-"`
 	CreatedAt time.Time `json:"created_at"`
+	UpdatedAt time.Time `json:"updated_at"`
+}
+
+// Link represents a shortened URL
+type Link struct {
+	ID             uint      `gorm:"primary_key" json:"id"`
+	UserID         *uint     `gorm:"index" json:"user_id"`
+	Code           string    `gorm:"unique_index;not null;size:32" json:"code"`
+	LongURL        string    `gorm:"not null;column:long_url" json:"long_url"`
+	IsCustom       bool      `gorm:"default:false" json:"is_custom"`
+	RequiresAuth   bool      `gorm:"default:false" json:"requires_auth"`
+	AccessUsername string    `gorm:"column:access_username;size:64" json:"access_username,omitempty"`
+	AccessPassword string    `gorm:"column:access_password" json:"-"`
+	ClickCount     int       `gorm:"default:0" json:"click_count"`
+	CreatedAt      time.Time `json:"created_at"`
+	UpdatedAt      time.Time `json:"updated_at"`
 }
 
 var db *gorm.DB
@@ -29,6 +47,8 @@ func init() {
 		panic(fmt.Sprintf("Failed to connect to database: %v", err))
 	}
 
-	// Auto-migrate database
-	db.AutoMigrate(&CodeURLMap{})
+	db.Exec("PRAGMA foreign_keys = ON")
+	db.Exec("PRAGMA journal_mode = WAL")
+
+	db.AutoMigrate(&User{}, &Link{})
 }