forked from Shiloh/githaven
140 lines
3.7 KiB
Go
140 lines
3.7 KiB
Go
// 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 explore
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
|
|
"code.gitea.io/gitea/models"
|
|
"code.gitea.io/gitea/modules/base"
|
|
"code.gitea.io/gitea/modules/context"
|
|
code_indexer "code.gitea.io/gitea/modules/indexer/code"
|
|
"code.gitea.io/gitea/modules/setting"
|
|
)
|
|
|
|
const (
|
|
// tplExploreCode explore code page template
|
|
tplExploreCode base.TplName = "explore/code"
|
|
)
|
|
|
|
// Code render explore code page
|
|
func Code(ctx *context.Context) {
|
|
if !setting.Indexer.RepoIndexerEnabled {
|
|
ctx.Redirect(setting.AppSubURL+"/explore", 302)
|
|
return
|
|
}
|
|
|
|
ctx.Data["UsersIsDisabled"] = setting.Service.Explore.DisableUsersPage
|
|
ctx.Data["IsRepoIndexerEnabled"] = setting.Indexer.RepoIndexerEnabled
|
|
ctx.Data["Title"] = ctx.Tr("explore")
|
|
ctx.Data["PageIsExplore"] = true
|
|
ctx.Data["PageIsExploreCode"] = true
|
|
|
|
language := strings.TrimSpace(ctx.Form("l"))
|
|
keyword := strings.TrimSpace(ctx.Form("q"))
|
|
page := ctx.FormInt("page")
|
|
if page <= 0 {
|
|
page = 1
|
|
}
|
|
|
|
queryType := strings.TrimSpace(ctx.Form("t"))
|
|
isMatch := queryType == "match"
|
|
|
|
var (
|
|
repoIDs []int64
|
|
err error
|
|
isAdmin bool
|
|
)
|
|
if ctx.User != nil {
|
|
isAdmin = ctx.User.IsAdmin
|
|
}
|
|
|
|
// guest user or non-admin user
|
|
if ctx.User == nil || !isAdmin {
|
|
repoIDs, err = models.FindUserAccessibleRepoIDs(ctx.User)
|
|
if err != nil {
|
|
ctx.ServerError("SearchResults", err)
|
|
return
|
|
}
|
|
}
|
|
|
|
var (
|
|
total int
|
|
searchResults []*code_indexer.Result
|
|
searchResultLanguages []*code_indexer.SearchResultLanguages
|
|
)
|
|
|
|
// if non-admin login user, we need check UnitTypeCode at first
|
|
if ctx.User != nil && len(repoIDs) > 0 {
|
|
repoMaps, err := models.GetRepositoriesMapByIDs(repoIDs)
|
|
if err != nil {
|
|
ctx.ServerError("SearchResults", err)
|
|
return
|
|
}
|
|
|
|
var rightRepoMap = make(map[int64]*models.Repository, len(repoMaps))
|
|
repoIDs = make([]int64, 0, len(repoMaps))
|
|
for id, repo := range repoMaps {
|
|
if repo.CheckUnitUser(ctx.User, models.UnitTypeCode) {
|
|
rightRepoMap[id] = repo
|
|
repoIDs = append(repoIDs, id)
|
|
}
|
|
}
|
|
|
|
ctx.Data["RepoMaps"] = rightRepoMap
|
|
|
|
total, searchResults, searchResultLanguages, err = code_indexer.PerformSearch(repoIDs, language, keyword, page, setting.UI.RepoSearchPagingNum, isMatch)
|
|
if err != nil {
|
|
ctx.ServerError("SearchResults", err)
|
|
return
|
|
}
|
|
// if non-login user or isAdmin, no need to check UnitTypeCode
|
|
} else if (ctx.User == nil && len(repoIDs) > 0) || isAdmin {
|
|
total, searchResults, searchResultLanguages, err = code_indexer.PerformSearch(repoIDs, language, keyword, page, setting.UI.RepoSearchPagingNum, isMatch)
|
|
if err != nil {
|
|
ctx.ServerError("SearchResults", err)
|
|
return
|
|
}
|
|
|
|
var loadRepoIDs = make([]int64, 0, len(searchResults))
|
|
for _, result := range searchResults {
|
|
var find bool
|
|
for _, id := range loadRepoIDs {
|
|
if id == result.RepoID {
|
|
find = true
|
|
break
|
|
}
|
|
}
|
|
if !find {
|
|
loadRepoIDs = append(loadRepoIDs, result.RepoID)
|
|
}
|
|
}
|
|
|
|
repoMaps, err := models.GetRepositoriesMapByIDs(loadRepoIDs)
|
|
if err != nil {
|
|
ctx.ServerError("SearchResults", err)
|
|
return
|
|
}
|
|
|
|
ctx.Data["RepoMaps"] = repoMaps
|
|
}
|
|
|
|
ctx.Data["Keyword"] = keyword
|
|
ctx.Data["Language"] = language
|
|
ctx.Data["queryType"] = queryType
|
|
ctx.Data["SearchResults"] = searchResults
|
|
ctx.Data["SearchResultLanguages"] = searchResultLanguages
|
|
ctx.Data["RequireHighlightJS"] = true
|
|
ctx.Data["PageIsViewCode"] = true
|
|
|
|
pager := context.NewPagination(total, setting.UI.RepoSearchPagingNum, page, 5)
|
|
pager.SetDefaultParams(ctx)
|
|
pager.AddParam(ctx, "l", "Language")
|
|
ctx.Data["Page"] = pager
|
|
|
|
ctx.HTML(http.StatusOK, tplExploreCode)
|
|
}
|