save

Author: hiifong

models/migrations/migrations.go

@@ -373,6 +373,7 @@ func prepareMigrationTasks() []*migration {
 
 		// Gitea 1.23.0-rc0 ends at migration ID number 311 (database version 312)
 		newMigration(312, "Add DeleteBranchAfterMerge to AutoMerge", v1_24.AddDeleteBranchAfterMergeForAutoMerge),
+		newMigration(314, "Add star_list table", v1_24.AddStarList),
 	}
 	return preparedMigrations
 }

models/migrations/v1_24/v314.go

@@ -0,0 +1,46 @@
+// Copyright 2025 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package v1_24 //nolint
+
+import (
+	"code.gitea.io/gitea/modules/timeutil"
+
+	"xorm.io/xorm"
+)
+
+type Star struct {
+	StarListID int64 `xorm:"UNIQUE(s)"`
+}
+
+type StarList struct {
+	ID   int64 `xorm:"pk autoincr"`
+	UID  int64 `xorm:"INDEX"`
+	Name string
+	Desc string
+
+	CreatedUnix timeutil.TimeStamp `xorm:"INDEX created"`
+	UpdatedUnix timeutil.TimeStamp `xorm:"INDEX updated"`
+}
+
+// TableName return database table name for xorm
+func (StarList) TableName() string {
+	return "star_list"
+}
+
+func AddStarList(x *xorm.Engine) error {
+	sess := x.NewSession()
+	defer sess.Close()
+	if err := sess.Begin(); err != nil {
+		return err
+	}
+	err := sess.Sync(new(Star))
+	if err != nil {
+		return err
+	}
+	err = sess.Sync(new(StarList))
+	if err != nil {
+		return err
+	}
+	return sess.Commit()
+}

models/repo/star.go

@@ -16,6 +16,7 @@ type Star struct {
 	ID          int64              `xorm:"pk autoincr"`
 	UID         int64              `xorm:"UNIQUE(s)"`
 	RepoID      int64              `xorm:"UNIQUE(s)"`
+	StarListID  int64              `xorm:"UNIQUE(s)"`
 	CreatedUnix timeutil.TimeStamp `xorm:"INDEX created"`
 }
 

models/repo/star_list.go

@@ -4,6 +4,8 @@
 package repo
 
 import (
+	"context"
+
 	"code.gitea.io/gitea/models/db"
 	"code.gitea.io/gitea/modules/timeutil"
 )
@@ -12,14 +14,54 @@ import (
 type StarList struct {
 	ID   int64 `xorm:"pk autoincr"`
 	UID  int64 `xorm:"INDEX"`
-	SID  int64 `xorm:"INDEX"`
 	Name string
 	Desc string
 
+	StarIDs []int64      `xorm:"-"`
+	Repos   []Repository `xorm:"-"`
+
 	CreatedUnix timeutil.TimeStamp `xorm:"INDEX created"`
 	UpdatedUnix timeutil.TimeStamp `xorm:"INDEX updated"`
 }
 
 func init() {
 	db.RegisterModel(new(StarList))
 }
+
+func (list *StarList) GetCount() int64 {
+	var star Star
+	count, err := db.GetEngine(context.Background()).Where("star_list_id = ?", list.ID).Count(star)
+	if err != nil {
+		return 0
+	}
+	return count
+}
+
+func (list *StarList) LoadStars() {
+	var star []Star
+	err := db.GetEngine(context.Background()).Where("star_list_id = ?", list.ID).Find(&star)
+	if err != nil {
+		return
+	}
+	for _, star := range star {
+		list.StarIDs = append(list.StarIDs, star.ID)
+	}
+}
+
+func GetStarListByUID(ctx context.Context, uid int64) ([]*StarList, error) {
+	var list []*StarList
+	err := db.GetEngine(ctx).Where("uid = ?", uid).Find(&list)
+	if err != nil {
+		return nil, err
+	}
+	return list, nil
+}
+
+func GetReposByStarListID(ctx context.Context, starListID int64) ([]*Repository, error) {
+	repos := make([]*Repository, 0, 100)
+	err := db.GetEngine(ctx).Where("star_list_id = ?", starListID).Find(&repos)
+	if err != nil {
+		return nil, err
+	}
+	return repos, nil
+}