add migration

Author: yp05327

models/migrations/migrations.go

@@ -511,6 +511,8 @@ var migrations = []Migration{
 	NewMigration("Add git_size and lfs_size columns to repository table", v1_21.AddGitSizeAndLFSSizeToRepositoryTable),
 	// v264 -> v265
 	NewMigration("Add branch table", v1_21.AddBranchTable),
+	// v265 -> v266
+	NewMigration("Fix missing admin team unit records", v1_21.FixMissingAdminTeamUnitRecords),
 }
 
 // GetCurrentDBVersion returns the current db version

models/migrations/v1_21/v265.go

@@ -0,0 +1,113 @@
+// Copyright 2023 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package v1_21 //nolint
+
+import (
+	"code.gitea.io/gitea/modules/container"
+	"xorm.io/xorm"
+)
+
+func FixMissingAdminTeamUnitRecords(x *xorm.Engine) error {
+	type UnitType int
+	type AccessMode int
+
+	type Team struct {
+		ID         int64      `xorm:"pk autoincr"`
+		OrgID      int64      `xorm:"INDEX"`
+		AccessMode AccessMode `xorm:"'authorize'"`
+	}
+
+	type TeamUnit struct {
+		ID         int64    `xorm:"pk autoincr"`
+		OrgID      int64    `xorm:"INDEX"`
+		TeamID     int64    `xorm:"UNIQUE(s)"`
+		Type       UnitType `xorm:"UNIQUE(s)"`
+		AccessMode AccessMode
+	}
+
+	const (
+		// AccessModeRead read access
+		AccessModeRead = 1
+		// AccessModeAdmin admin access
+		AccessModeAdmin = 3
+
+		// Unit Type
+		TypeInvalid         UnitType = iota // 0 invalid
+		TypeCode                            // 1 code
+		TypeIssues                          // 2 issues
+		TypePullRequests                    // 3 PRs
+		TypeReleases                        // 4 Releases
+		TypeUncyclo                            // 5 Uncyclo
+		TypeExternalUncyclo                    // 6 ExternalUncyclo
+		TypeExternalTracker                 // 7 ExternalTracker
+		TypeProjects                        // 8 Kanban board
+		TypePackages                        // 9 Packages
+		TypeActions                         // 10 Actions
+	)
+
+	var AllRepoUnitTypes = []UnitType{
+		TypeCode,
+		TypeIssues,
+		TypePullRequests,
+		TypeReleases,
+		TypeUncyclo,
+		TypeExternalUncyclo,
+		TypeExternalTracker,
+		TypeProjects,
+		TypePackages,
+		TypeActions,
+	}
+
+	sess := x.NewSession()
+	defer sess.Close()
+
+	if err := sess.Begin(); err != nil {
+		return err
+	}
+
+	// find all admin teams
+	teams := make([]*Team, 0)
+	err := sess.Where("team.authorize == ?", AccessModeAdmin).Find(&teams)
+	if err != nil {
+		return err
+	}
+
+	for _, team := range teams {
+		// find all existing records
+		teamunits := make([]*TeamUnit, 0, len(AllRepoUnitTypes))
+		err := sess.Where("`team_unit`.team_id == ?", team.ID).Find(&teamunits)
+		if err != nil {
+			return err
+		}
+		existingUnitTypes := make(container.Set[UnitType], 0)
+		for _, tu := range teamunits {
+			if tu.Type > 0 {
+				existingUnitTypes.Add(tu.Type)
+			}
+		}
+
+		// insert or update records
+		for _, u := range AllRepoUnitTypes {
+			newTeamUnit := &TeamUnit{
+				OrgID:  team.OrgID,
+				TeamID: team.ID,
+				Type:   u,
+			}
+			// external unit should be read
+			if u == TypeExternalUncyclo || u == TypeExternalTracker {
+				newTeamUnit.AccessMode = AccessModeRead
+			} else {
+				newTeamUnit.AccessMode = AccessModeAdmin
+			}
+
+			if existingUnitTypes.Contains(u) {
+				sess.Cols("access_mode").Update(newTeamUnit)
+			} else {
+				sess.Insert(newTeamUnit)
+			}
+		}
+	}
+
+	return sess.Commit()
+}