database

type AdminTokenEntity struct {
	ID          int64     `datastore:"-"`
	TokenHash   string    `datastore:"token_hash"`
	Description string    `datastore:"description"`
	CreatedAt   time.Time `datastore:"created_at"`
	LastUsedAt  time.Time `datastore:"last_used_at"`
	IsActive    bool      `datastore:"is_active"`
}

type Article struct {
	ID          int       `json:"id"`
	FeedID      int       `json:"feed_id"`
	FeedTitle   string    `json:"feed_title,omitempty"`
	Title       string    `json:"title"`
	URL         string    `json:"url"`
	Content     string    `json:"content"`
	Description string    `json:"description"`
	Author      string    `json:"author"`
	PublishedAt time.Time `json:"published_at"`
	CreatedAt   time.Time `json:"created_at"`
	IsRead      bool      `json:"is_read"`
	IsStarred   bool      `json:"is_starred"`
}

type ArticleEntity struct {
	ID          int64     `datastore:"-"`
	FeedID      int64     `datastore:"feed_id"`
	Title       string    `datastore:"title"`
	URL         string    `datastore:"url"`
	Content     string    `datastore:"content,noindex"`
	Description string    `datastore:"description,noindex"`
	Author      string    `datastore:"author"`
	PublishedAt time.Time `datastore:"published_at"`
	CreatedAt   time.Time `datastore:"created_at"`
	IsRead      bool      `datastore:"is_read"`
	IsStarred   bool      `datastore:"is_starred"`
}

type ArticlePaginationResult struct {
	Articles   []Article
	NextCursor string // Empty string means no more results
}

type AuditLog struct {
	ID               int       `json:"id"`
	Timestamp        time.Time `json:"timestamp"`
	AdminUserID      int       `json:"admin_user_id"`
	AdminEmail       string    `json:"admin_email"`
	OperationType    string    `json:"operation_type"`
	TargetUserID     int       `json:"target_user_id"`
	TargetUserEmail  string    `json:"target_user_email"`
	OperationDetails string    `json:"operation_details"` // JSON string
	IPAddress        string    `json:"ip_address"`
	Result           string    `json:"result"` // "success" or "failure"
	ErrorMessage     string    `json:"error_message"`
}

type AuditLogEntity struct {
	ID               int64     `datastore:"-"`
	Timestamp        time.Time `datastore:"timestamp"`
	AdminUserID      int64     `datastore:"admin_user_id"`
	AdminEmail       string    `datastore:"admin_email"`
	OperationType    string    `datastore:"operation_type"`
	TargetUserID     int64     `datastore:"target_user_id"`
	TargetUserEmail  string    `datastore:"target_user_email"`
	OperationDetails string    `datastore:"operation_details,noindex"`
	IPAddress        string    `datastore:"ip_address"`
	Result           string    `datastore:"result"`
	ErrorMessage     string    `datastore:"error_message,noindex"`
}

type DB struct {
	*sql.DB
}

type Database interface {
	// User methods
	CreateUser(user *User) error
	GetUserByGoogleID(googleID string) (*User, error)
	GetUserByID(userID int) (*User, error)
	UpdateUserSubscription(userID int, status, subscriptionID string, lastPaymentDate, nextBillingDate time.Time) error
	IsUserSubscriptionActive(userID int) (bool, error)
	GetUserFeedCount(userID int) (int, error)
	UpdateUserMaxArticlesOnFeedAdd(userID int, maxArticles int) error

	// Admin methods
	SetUserAdmin(userID int, isAdmin bool) error
	GrantFreeMonths(userID int, months int) error
	GetUserByEmail(email string) (*User, error)

	// Feed methods
	AddFeed(feed *Feed) error
	UpdateFeed(feed *Feed) error
	UpdateFeedTracking(feedID int, lastChecked, lastHadNewContent time.Time, averageUpdateInterval int) error
	GetFeeds() ([]Feed, error)
	GetFeedByURL(url string) (*Feed, error)
	GetUserFeeds(userID int) ([]Feed, error)
	GetAllUserFeeds() ([]Feed, error)
	DeleteFeed(id int) error
	SubscribeUserToFeed(userID, feedID int) error
	UnsubscribeUserFromFeed(userID, feedID int) error

	// Article methods
	AddArticle(article *Article) error
	GetArticles(feedID int) ([]Article, error)
	FindArticleByURL(url string) (*Article, error)
	GetUserArticles(userID int) ([]Article, error)
	GetUserArticlesPaginated(userID int, limit int, cursor string, unreadOnly bool) (*ArticlePaginationResult, error)
	GetUserFeedArticles(userID, feedID int) ([]Article, error)

	// User article status methods
	GetUserArticleStatus(userID, articleID int) (*UserArticle, error)
	SetUserArticleStatus(userID, articleID int, isRead, isStarred bool) error
	BatchSetUserArticleStatus(userID int, articles []Article, isRead, isStarred bool) error
	MarkUserArticleRead(userID, articleID int, isRead bool) error
	ToggleUserArticleStar(userID, articleID int) error
	GetUserUnreadCounts(userID int) (map[int]int, error)
	CleanupOrphanedUserArticles(olderThanDays int) (int, error)

	// Session methods
	CreateSession(session *Session) error
	GetSession(sessionID string) (*Session, error)
	DeleteSession(sessionID string) error
	DeleteExpiredSessions() error

	// Audit log methods
	CreateAuditLog(log *AuditLog) error
	GetAuditLogs(limit, offset int, filters map[string]interface{}) ([]AuditLog, error)

	UpdateFeedLastFetch(feedID int, lastFetch time.Time) error
	Close() error
}

type DatastoreDB struct {
	// contains filtered or unexported fields
}

type Feed struct {
	ID                    int       `json:"id"`
	Title                 string    `json:"title"`
	URL                   string    `json:"url"`
	Description           string    `json:"description"`
	CreatedAt             time.Time `json:"created_at"`
	UpdatedAt             time.Time `json:"updated_at"`
	LastFetch             time.Time `json:"last_fetch"`
	LastChecked           time.Time `json:"last_checked"`            // When we last attempted to check for updates
	LastHadNewContent     time.Time `json:"last_had_new_content"`    // When we last found new articles
	AverageUpdateInterval int       `json:"average_update_interval"` // Average seconds between updates (0 = unknown)
}

type FeedEntity struct {
	ID                    int64     `datastore:"-"`
	Title                 string    `datastore:"title"`
	URL                   string    `datastore:"url"`
	Description           string    `datastore:"description"`
	CreatedAt             time.Time `datastore:"created_at"`
	UpdatedAt             time.Time `datastore:"updated_at"`
	LastFetch             time.Time `datastore:"last_fetch"`
	LastChecked           time.Time `datastore:"last_checked"`
	LastHadNewContent     time.Time `datastore:"last_had_new_content"`
	AverageUpdateInterval int       `datastore:"average_update_interval"`
}

type Session struct {
	ID        string    `json:"id"`
	UserID    int       `json:"user_id"`
	CreatedAt time.Time `json:"created_at"`
	ExpiresAt time.Time `json:"expires_at"`
}

type SessionEntity struct {
	ID        string    `datastore:"-"` // SessionID is the key
	UserID    int64     `datastore:"user_id"`
	CreatedAt time.Time `datastore:"created_at"`
	ExpiresAt time.Time `datastore:"expires_at"`
}

type User struct {
	ID                   int       `json:"id"`
	GoogleID             string    `json:"google_id"`
	Email                string    `json:"email"`
	Name                 string    `json:"name"`
	Avatar               string    `json:"avatar"`
	CreatedAt            time.Time `json:"created_at"`
	SubscriptionStatus   string    `json:"subscription_status"`      // 'trial', 'active', 'cancelled', 'expired', 'admin'
	SubscriptionID       string    `json:"subscription_id"`          // Stripe subscription ID
	TrialEndsAt          time.Time `json:"trial_ends_at"`            // When free trial expires
	LastPaymentDate      time.Time `json:"last_payment_date"`        // Last successful payment
	NextBillingDate      time.Time `json:"next_billing_date"`        // Next billing date for active subscriptions
	IsAdmin              bool      `json:"is_admin"`                 // Admin users bypass subscription limits
	FreeMonthsRemaining  int       `json:"free_months_remaining"`    // Additional free months granted
	MaxArticlesOnFeedAdd int       `json:"max_articles_on_feed_add"` // Max articles to import when adding a new feed (0 = unlimited)
}

type UserArticle struct {
	UserID    int  `json:"user_id"`
	ArticleID int  `json:"article_id"`
	IsRead    bool `json:"is_read"`
	IsStarred bool `json:"is_starred"`
}

type UserArticleEntity struct {
	UserID    int64 `datastore:"user_id"`
	ArticleID int64 `datastore:"article_id"`
	IsRead    bool  `datastore:"is_read"`
	IsStarred bool  `datastore:"is_starred"`
}

type UserEntity struct {
	ID                   int64     `datastore:"-"`
	GoogleID             string    `datastore:"google_id"`
	Email                string    `datastore:"email"`
	Name                 string    `datastore:"name"`
	Avatar               string    `datastore:"avatar"`
	CreatedAt            time.Time `datastore:"created_at"`
	SubscriptionStatus   string    `datastore:"subscription_status"`
	SubscriptionID       string    `datastore:"subscription_id"`
	TrialEndsAt          time.Time `datastore:"trial_ends_at"`
	LastPaymentDate      time.Time `datastore:"last_payment_date"`
	NextBillingDate      time.Time `datastore:"next_billing_date"`
	IsAdmin              bool      `datastore:"is_admin"`
	FreeMonthsRemaining  int       `datastore:"free_months_remaining"`
	MaxArticlesOnFeedAdd int       `datastore:"max_articles_on_feed_add"`
}

type UserFeed struct {
	UserID int `json:"user_id"`
	FeedID int `json:"feed_id"`
}

type UserFeedEntity struct {
	UserID int64 `datastore:"user_id"`
	FeedID int64 `datastore:"feed_id"`
}