Add doctor for wrong label and issue_label data

Author: lunny

models/issue_label.go

@@ -779,3 +779,36 @@ func deleteLabelsByRepoID(sess Engine, repoID int64) error {
 	_, err := sess.Delete(&Label{RepoID: repoID})
 	return err
 }
+
+// CountWrongLabels counts wrong data in issue_label and label
+func CountWrongLabels() (int64, int64, error) {
+	wrongLabels, err := x.Where("repo_id > 0 and repo_id not in (select id from repository)").Count(new(Label))
+	if err != nil {
+		return 0, 0, err
+	}
+
+	wrongIssueLabels, err := x.Where("label_id IN (select id from label where repo_id > 0 and repo_id not in (select id from repository))").Count(new(IssueLabel))
+	if err != nil {
+		return 0, 0, err
+	}
+
+	return wrongLabels, wrongIssueLabels, nil
+}
+
+// FixWrongLabels fix wrong data in label
+func FixWrongLabels() (int64, error) {
+	res, err := x.Exec("DELETE FROM label WHERE repo_id > 0 and repo_id not in (select id from repository)")
+	if err != nil {
+		return 0, err
+	}
+	return res.RowsAffected()
+}
+
+// FixWrongIssueLables fix wrong label data
+func FixWrongIssueLables() (int64, error) {
+	res, err := x.Exec("DELETE FROM issue_label WHERE label_id in (select id FROM label WHERE repo_id > 0 and repo_id not in (select id from repository))")
+	if err != nil {
+		return 0, err
+	}
+	return res.RowsAffected()
+}

modules/doctor/label.go

@@ -0,0 +1,54 @@
+// Copyright 2021 The Gitea Authors. All rights reserved.
+// Use of this source code is governed by a MIT-style
+// license that can be found in the LICENSE file.
+
+package doctor
+
+import (
+	"code.gitea.io/gitea/models"
+	"code.gitea.io/gitea/modules/log"
+)
+
+func checkLabels(logger log.Logger, autofix bool) error {
+	lablesCnt, issueLabelsCnt, err := models.CountWrongLabels()
+	if err != nil {
+		logger.Critical("Error: %v whilst counting wrong labels and issue labels")
+		return err
+	}
+	var count int64
+	if issueLabelsCnt > 0 { // We should check issue_label before label
+		if autofix {
+			if count, err = models.FixWrongIssueLables(); err != nil {
+				logger.Critical("Error: %v whilst fixing wrong issue labels")
+				return err
+			}
+			logger.Info("%d issue labels missing repositories fixed", count)
+		} else {
+			logger.Warn("%d issue labels missing repositories exist", issueLabelsCnt)
+		}
+	}
+	if lablesCnt > 0 {
+		if autofix {
+			if count, err = models.FixWrongLabels(); err != nil {
+				logger.Critical("Error: %v whilst fixing wrong labels")
+				return err
+			}
+			logger.Info("%d labels missing repositories fixed", count)
+		} else {
+			logger.Warn("%d labels missing repositories exist", lablesCnt)
+		}
+	}
+	return nil
+}
+
+func init() {
+	Register(&Check{
+		Title:                      "Check wrong labels and issue_labels where repositories deleted",
+		Name:                       "labels",
+		IsDefault:                  true,
+		Run:                        checkLabels,
+		AbortIfFailed:              true,
+		SkipDatabaseInitialization: false,
+		Priority:                   1,
+	})
+}