From ee341e70b9f0ec229e657b9bf447e02fd5dc173f Mon Sep 17 00:00:00 2001
From: zouap <zouap@pcl.ac.cn>
Date: Thu, 10 Mar 2022 09:27:10 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: zouap <zouap@pcl.ac.cn>
---
 routers/search.go | 68 ++++++++++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 65 insertions(+), 3 deletions(-)

diff --git a/routers/search.go b/routers/search.go
index 5e4fff836..6893d6914 100644
--- a/routers/search.go
+++ b/routers/search.go
@@ -98,19 +98,81 @@ func searchRepoByLabel(ctx *context.Context, Key string, Page int, PageSize int)
 	if SortBy == "" {
 		SortBy = "updated_unix.keyword"
 	}
+	PrivateTotal := ctx.QueryInt("PrivateTotal")
+	WebTotal := ctx.QueryInt("WebTotal")
 	ascending := ctx.QueryBool("Ascending")
+	from := (Page - 1) * PageSize
+	resultObj := &SearchRes{}
+	log.Info("WebTotal=" + fmt.Sprint(WebTotal))
+	log.Info("PrivateTotal=" + fmt.Sprint(PrivateTotal))
+	resultObj.Result = make([]map[string]interface{}, 0)
+
+	if from < PrivateTotal || from == 0 {
+		orderBy := models.SearchOrderByRecentUpdated
+		switch SortBy {
+		case "updated_unix.keyword":
+			orderBy = models.SearchOrderByRecentUpdated
+		case "num_stars":
+			orderBy = models.SearchOrderByStarsReverse
+		case "num_forks":
+			orderBy = models.SearchOrderByForksReverse
+		case "num_watches":
+			orderBy = models.SearchOrderByWatches
+		}
+		log.Info("actor is null?:" + fmt.Sprint(ctx.User == nil))
+		repos, count, err := models.SearchRepository(&models.SearchRepoOptions{
+			ListOptions: models.ListOptions{
+				Page:     Page,
+				PageSize: PageSize,
+			},
+			Actor:              ctx.User,
+			OrderBy:            orderBy,
+			Private:            true,
+			OnlyPrivate:        true,
+			TopicOnly:          true,
+			TopicName:          Key,
+			IncludeDescription: setting.UI.SearchRepoDescription,
+		})
+		if err != nil {
+			ctx.JSON(200, "")
+			return
+		}
+		resultObj.PrivateTotal = count
+		if repos.Len() > 0 {
+			log.Info("Query private repo number is:" + fmt.Sprint(repos.Len()))
+			makePrivateRepo(repos, resultObj, Key)
+		} else {
+			log.Info("not found private repo,keyword=" + Key)
+		}
+		if repos.Len() >= PageSize {
+			resultObj.Total = int64(WebTotal)
+			ctx.JSON(200, resultObj)
+			return
+		}
+	} else {
+		resultObj.PrivateTotal = int64(PrivateTotal)
+	}
+
+	from = from - PrivateTotal
+	if from < 0 {
+		from = 0
+	}
+	Size := PageSize - len(resultObj.Result)
+
 	log.Info("query searchRepoByLabel start")
 	if Key != "" {
 		boolQ := elastic.NewBoolQuery()
 		topicsQuery := elastic.NewMatchQuery("topics", Key)
 		boolQ.Should(topicsQuery)
 
-		res, err := client.Search("repository-es-index").Query(boolQ).SortBy(elastic.NewScoreSort(), elastic.NewFieldSort(SortBy).Order(ascending)).From((Page - 1) * PageSize).Size(PageSize).Highlight(queryHighlight("topics")).Do(ctx.Req.Context())
+		res, err := client.Search("repository-es-index").Query(boolQ).SortBy(elastic.NewScoreSort(), elastic.NewFieldSort(SortBy).Order(ascending)).From(from).Size(Size).Highlight(queryHighlight("topics")).Do(ctx.Req.Context())
 		if err == nil {
 			searchJson, _ := json.Marshal(res)
 			log.Info("searchJson=" + string(searchJson))
-			result := makeRepoResult(res, "", false)
-			ctx.JSON(200, result)
+			esresult := makeRepoResult(res, "", false)
+			resultObj.Total = resultObj.PrivateTotal + esresult.Total
+			resultObj.Result = append(resultObj.Result, esresult.Result...)
+			ctx.JSON(200, esresult)
 		} else {
 			log.Info("query es error," + err.Error())
 			ctx.JSON(200, "")