You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

repo.go 62KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140
  1. // Copyright 2014 The Gogs Authors. All rights reserved.
  2. // Copyright 2017 The Gitea Authors. All rights reserved.
  3. // Use of this source code is governed by a MIT-style
  4. // license that can be found in the LICENSE file.
  5. package models
  6. import (
  7. "context"
  8. "errors"
  9. "fmt"
  10. "html/template"
  11. "unicode/utf8"
  12. // Needed for jpeg support
  13. _ "image/jpeg"
  14. "io/ioutil"
  15. "net"
  16. "net/url"
  17. "os"
  18. "path"
  19. "path/filepath"
  20. "sort"
  21. "strconv"
  22. "strings"
  23. "time"
  24. "code.gitea.io/gitea/modules/log"
  25. "code.gitea.io/gitea/modules/markup"
  26. "code.gitea.io/gitea/modules/options"
  27. "code.gitea.io/gitea/modules/setting"
  28. "code.gitea.io/gitea/modules/storage"
  29. api "code.gitea.io/gitea/modules/structs"
  30. "code.gitea.io/gitea/modules/timeutil"
  31. "code.gitea.io/gitea/modules/util"
  32. "xorm.io/builder"
  33. )
  34. var (
  35. // ErrMirrorNotExist mirror does not exist error
  36. ErrMirrorNotExist = errors.New("Mirror does not exist")
  37. // ErrNameEmpty name is empty error
  38. ErrNameEmpty = errors.New("Name is empty")
  39. )
  40. var (
  41. // Gitignores contains the gitiginore files
  42. Gitignores []string
  43. // Licenses contains the license files
  44. Licenses []string
  45. // Readmes contains the readme files
  46. Readmes []string
  47. // LabelTemplates contains the label template files and the list of labels for each file
  48. LabelTemplates map[string]string
  49. // ItemsPerPage maximum items per page in forks, watchers and stars of a repo
  50. ItemsPerPage = 40
  51. )
  52. // loadRepoConfig loads the repository config
  53. func loadRepoConfig() {
  54. // Load .gitignore and license files and readme templates.
  55. types := []string{"gitignore", "license", "readme", "label"}
  56. typeFiles := make([][]string, 4)
  57. for i, t := range types {
  58. files, err := options.Dir(t)
  59. if err != nil {
  60. log.Fatal("Failed to get %s files: %v", t, err)
  61. }
  62. customPath := path.Join(setting.CustomPath, "options", t)
  63. isDir, err := util.IsDir(customPath)
  64. if err != nil {
  65. log.Fatal("Failed to get custom %s files: %v", t, err)
  66. }
  67. if isDir {
  68. customFiles, err := util.StatDir(customPath)
  69. if err != nil {
  70. log.Fatal("Failed to get custom %s files: %v", t, err)
  71. }
  72. for _, f := range customFiles {
  73. if !util.IsStringInSlice(f, files, true) {
  74. files = append(files, f)
  75. }
  76. }
  77. }
  78. typeFiles[i] = files
  79. }
  80. Gitignores = typeFiles[0]
  81. Licenses = typeFiles[1]
  82. Readmes = typeFiles[2]
  83. LabelTemplatesFiles := typeFiles[3]
  84. sort.Strings(Gitignores)
  85. sort.Strings(Licenses)
  86. sort.Strings(Readmes)
  87. sort.Strings(LabelTemplatesFiles)
  88. // Load label templates
  89. LabelTemplates = make(map[string]string)
  90. for _, templateFile := range LabelTemplatesFiles {
  91. labels, err := LoadLabelsFormatted(templateFile)
  92. if err != nil {
  93. log.Error("Failed to load labels: %v", err)
  94. }
  95. LabelTemplates[templateFile] = labels
  96. }
  97. // Filter out invalid names and promote preferred licenses.
  98. sortedLicenses := make([]string, 0, len(Licenses))
  99. for _, name := range setting.Repository.PreferredLicenses {
  100. if util.IsStringInSlice(name, Licenses, true) {
  101. sortedLicenses = append(sortedLicenses, name)
  102. }
  103. }
  104. for _, name := range Licenses {
  105. if !util.IsStringInSlice(name, setting.Repository.PreferredLicenses, true) {
  106. sortedLicenses = append(sortedLicenses, name)
  107. }
  108. }
  109. Licenses = sortedLicenses
  110. }
  111. // NewRepoContext creates a new repository context
  112. func NewRepoContext() {
  113. loadRepoConfig()
  114. loadUnitConfig()
  115. RemoveAllWithNotice("Clean up repository temporary data", filepath.Join(setting.AppDataPath, "tmp"))
  116. }
  117. // RepositoryStatus defines the status of repository
  118. type RepositoryStatus int
  119. // all kinds of RepositoryStatus
  120. const (
  121. RepositoryReady RepositoryStatus = iota // a normal repository
  122. RepositoryBeingMigrated // repository is migrating
  123. RepositoryPendingTransfer // repository pending in ownership transfer state
  124. )
  125. // TrustModelType defines the types of trust model for this repository
  126. type TrustModelType int
  127. // kinds of TrustModel
  128. const (
  129. DefaultTrustModel TrustModelType = iota // default trust model
  130. CommitterTrustModel
  131. CollaboratorTrustModel
  132. CollaboratorCommitterTrustModel
  133. )
  134. // String converts a TrustModelType to a string
  135. func (t TrustModelType) String() string {
  136. switch t {
  137. case DefaultTrustModel:
  138. return "default"
  139. case CommitterTrustModel:
  140. return "committer"
  141. case CollaboratorTrustModel:
  142. return "collaborator"
  143. case CollaboratorCommitterTrustModel:
  144. return "collaboratorcommitter"
  145. }
  146. return "default"
  147. }
  148. // ToTrustModel converts a string to a TrustModelType
  149. func ToTrustModel(model string) TrustModelType {
  150. switch strings.ToLower(strings.TrimSpace(model)) {
  151. case "default":
  152. return DefaultTrustModel
  153. case "collaborator":
  154. return CollaboratorTrustModel
  155. case "committer":
  156. return CommitterTrustModel
  157. case "collaboratorcommitter":
  158. return CollaboratorCommitterTrustModel
  159. }
  160. return DefaultTrustModel
  161. }
  162. // Repository represents a git repository.
  163. type Repository struct {
  164. ID int64 `xorm:"pk autoincr"`
  165. OwnerID int64 `xorm:"UNIQUE(s) index"`
  166. OwnerName string
  167. Owner *User `xorm:"-"`
  168. LowerName string `xorm:"UNIQUE(s) INDEX NOT NULL"`
  169. Name string `xorm:"INDEX NOT NULL"`
  170. Description string `xorm:"TEXT"`
  171. Website string `xorm:"VARCHAR(2048)"`
  172. OriginalServiceType api.GitServiceType `xorm:"index"`
  173. OriginalURL string `xorm:"VARCHAR(2048)"`
  174. DefaultBranch string
  175. NumWatches int
  176. NumStars int
  177. NumForks int
  178. NumIssues int
  179. NumClosedIssues int
  180. NumOpenIssues int `xorm:"-"`
  181. NumPulls int
  182. NumClosedPulls int
  183. NumOpenPulls int `xorm:"-"`
  184. NumMilestones int `xorm:"NOT NULL DEFAULT 0"`
  185. NumClosedMilestones int `xorm:"NOT NULL DEFAULT 0"`
  186. NumOpenMilestones int `xorm:"-"`
  187. NumProjects int `xorm:"NOT NULL DEFAULT 0"`
  188. NumClosedProjects int `xorm:"NOT NULL DEFAULT 0"`
  189. NumOpenProjects int `xorm:"-"`
  190. IsPrivate bool `xorm:"INDEX"`
  191. IsEmpty bool `xorm:"INDEX"`
  192. IsArchived bool `xorm:"INDEX"`
  193. IsMirror bool `xorm:"INDEX"`
  194. *Mirror `xorm:"-"`
  195. Status RepositoryStatus `xorm:"NOT NULL DEFAULT 0"`
  196. RenderingMetas map[string]string `xorm:"-"`
  197. DocumentRenderingMetas map[string]string `xorm:"-"`
  198. Units []*RepoUnit `xorm:"-"`
  199. PrimaryLanguage *LanguageStat `xorm:"-"`
  200. IsFork bool `xorm:"INDEX NOT NULL DEFAULT false"`
  201. ForkID int64 `xorm:"INDEX"`
  202. BaseRepo *Repository `xorm:"-"`
  203. IsTemplate bool `xorm:"INDEX NOT NULL DEFAULT false"`
  204. TemplateID int64 `xorm:"INDEX"`
  205. TemplateRepo *Repository `xorm:"-"`
  206. Size int64 `xorm:"NOT NULL DEFAULT 0"`
  207. CodeIndexerStatus *RepoIndexerStatus `xorm:"-"`
  208. StatsIndexerStatus *RepoIndexerStatus `xorm:"-"`
  209. IsFsckEnabled bool `xorm:"NOT NULL DEFAULT true"`
  210. CloseIssuesViaCommitInAnyBranch bool `xorm:"NOT NULL DEFAULT false"`
  211. Topics []string `xorm:"TEXT JSON"`
  212. TrustModel TrustModelType
  213. // Avatar: ID(10-20)-md5(32) - must fit into 64 symbols
  214. Avatar string `xorm:"VARCHAR(64)"`
  215. CreatedUnix timeutil.TimeStamp `xorm:"INDEX created"`
  216. UpdatedUnix timeutil.TimeStamp `xorm:"INDEX updated"`
  217. }
  218. // SanitizedOriginalURL returns a sanitized OriginalURL
  219. func (repo *Repository) SanitizedOriginalURL() string {
  220. if repo.OriginalURL == "" {
  221. return ""
  222. }
  223. return util.SanitizeURLCredentials(repo.OriginalURL, false)
  224. }
  225. // ColorFormat returns a colored string to represent this repo
  226. func (repo *Repository) ColorFormat(s fmt.State) {
  227. log.ColorFprintf(s, "%d:%s/%s",
  228. log.NewColoredIDValue(repo.ID),
  229. repo.OwnerName,
  230. repo.Name)
  231. }
  232. // IsBeingMigrated indicates that repository is being migrated
  233. func (repo *Repository) IsBeingMigrated() bool {
  234. return repo.Status == RepositoryBeingMigrated
  235. }
  236. // IsBeingCreated indicates that repository is being migrated or forked
  237. func (repo *Repository) IsBeingCreated() bool {
  238. return repo.IsBeingMigrated()
  239. }
  240. // AfterLoad is invoked from XORM after setting the values of all fields of this object.
  241. func (repo *Repository) AfterLoad() {
  242. // FIXME: use models migration to solve all at once.
  243. if len(repo.DefaultBranch) == 0 {
  244. repo.DefaultBranch = setting.Repository.DefaultBranch
  245. }
  246. repo.NumOpenIssues = repo.NumIssues - repo.NumClosedIssues
  247. repo.NumOpenPulls = repo.NumPulls - repo.NumClosedPulls
  248. repo.NumOpenMilestones = repo.NumMilestones - repo.NumClosedMilestones
  249. repo.NumOpenProjects = repo.NumProjects - repo.NumClosedProjects
  250. }
  251. // MustOwner always returns a valid *User object to avoid
  252. // conceptually impossible error handling.
  253. // It creates a fake object that contains error details
  254. // when error occurs.
  255. func (repo *Repository) MustOwner() *User {
  256. return repo.mustOwner(x)
  257. }
  258. // FullName returns the repository full name
  259. func (repo *Repository) FullName() string {
  260. return repo.OwnerName + "/" + repo.Name
  261. }
  262. // HTMLURL returns the repository HTML URL
  263. func (repo *Repository) HTMLURL() string {
  264. return setting.AppURL + repo.FullName()
  265. }
  266. // CommitLink make link to by commit full ID
  267. // note: won't check whether it's an right id
  268. func (repo *Repository) CommitLink(commitID string) (result string) {
  269. if commitID == "" || commitID == "0000000000000000000000000000000000000000" {
  270. result = ""
  271. } else {
  272. result = repo.HTMLURL() + "/commit/" + commitID
  273. }
  274. return
  275. }
  276. // APIURL returns the repository API URL
  277. func (repo *Repository) APIURL() string {
  278. return setting.AppURL + "api/v1/repos/" + repo.FullName()
  279. }
  280. // GetCommitsCountCacheKey returns cache key used for commits count caching.
  281. func (repo *Repository) GetCommitsCountCacheKey(contextName string, isRef bool) string {
  282. var prefix string
  283. if isRef {
  284. prefix = "ref"
  285. } else {
  286. prefix = "commit"
  287. }
  288. return fmt.Sprintf("commits-count-%d-%s-%s", repo.ID, prefix, contextName)
  289. }
  290. func (repo *Repository) getUnits(e Engine) (err error) {
  291. if repo.Units != nil {
  292. return nil
  293. }
  294. repo.Units, err = getUnitsByRepoID(e, repo.ID)
  295. log.Trace("repo.Units: %-+v", repo.Units)
  296. return err
  297. }
  298. // CheckUnitUser check whether user could visit the unit of this repository
  299. func (repo *Repository) CheckUnitUser(user *User, unitType UnitType) bool {
  300. return repo.checkUnitUser(x, user, unitType)
  301. }
  302. func (repo *Repository) checkUnitUser(e Engine, user *User, unitType UnitType) bool {
  303. if user.IsAdmin {
  304. return true
  305. }
  306. perm, err := getUserRepoPermission(e, repo, user)
  307. if err != nil {
  308. log.Error("getUserRepoPermission(): %v", err)
  309. return false
  310. }
  311. return perm.CanRead(unitType)
  312. }
  313. // UnitEnabled if this repository has the given unit enabled
  314. func (repo *Repository) UnitEnabled(tp UnitType) bool {
  315. if err := repo.getUnits(x); err != nil {
  316. log.Warn("Error loading repository (ID: %d) units: %s", repo.ID, err.Error())
  317. }
  318. for _, unit := range repo.Units {
  319. if unit.Type == tp {
  320. return true
  321. }
  322. }
  323. return false
  324. }
  325. // ErrUnitTypeNotExist represents a "UnitTypeNotExist" kind of error.
  326. type ErrUnitTypeNotExist struct {
  327. UT UnitType
  328. }
  329. // IsErrUnitTypeNotExist checks if an error is a ErrUnitNotExist.
  330. func IsErrUnitTypeNotExist(err error) bool {
  331. _, ok := err.(ErrUnitTypeNotExist)
  332. return ok
  333. }
  334. func (err ErrUnitTypeNotExist) Error() string {
  335. return fmt.Sprintf("Unit type does not exist: %s", err.UT.String())
  336. }
  337. // MustGetUnit always returns a RepoUnit object
  338. func (repo *Repository) MustGetUnit(tp UnitType) *RepoUnit {
  339. ru, err := repo.GetUnit(tp)
  340. if err == nil {
  341. return ru
  342. }
  343. if tp == UnitTypeExternalWiki {
  344. return &RepoUnit{
  345. Type: tp,
  346. Config: new(ExternalWikiConfig),
  347. }
  348. } else if tp == UnitTypeExternalTracker {
  349. return &RepoUnit{
  350. Type: tp,
  351. Config: new(ExternalTrackerConfig),
  352. }
  353. } else if tp == UnitTypePullRequests {
  354. return &RepoUnit{
  355. Type: tp,
  356. Config: new(PullRequestsConfig),
  357. }
  358. } else if tp == UnitTypeIssues {
  359. return &RepoUnit{
  360. Type: tp,
  361. Config: new(IssuesConfig),
  362. }
  363. }
  364. return &RepoUnit{
  365. Type: tp,
  366. Config: new(UnitConfig),
  367. }
  368. }
  369. // GetUnit returns a RepoUnit object
  370. func (repo *Repository) GetUnit(tp UnitType) (*RepoUnit, error) {
  371. return repo.getUnit(x, tp)
  372. }
  373. func (repo *Repository) getUnit(e Engine, tp UnitType) (*RepoUnit, error) {
  374. if err := repo.getUnits(e); err != nil {
  375. return nil, err
  376. }
  377. for _, unit := range repo.Units {
  378. if unit.Type == tp {
  379. return unit, nil
  380. }
  381. }
  382. return nil, ErrUnitTypeNotExist{tp}
  383. }
  384. func (repo *Repository) getOwner(e Engine) (err error) {
  385. if repo.Owner != nil {
  386. return nil
  387. }
  388. repo.Owner, err = getUserByID(e, repo.OwnerID)
  389. return err
  390. }
  391. // GetOwner returns the repository owner
  392. func (repo *Repository) GetOwner() error {
  393. return repo.getOwner(x)
  394. }
  395. func (repo *Repository) mustOwner(e Engine) *User {
  396. if err := repo.getOwner(e); err != nil {
  397. return &User{
  398. Name: "error",
  399. FullName: err.Error(),
  400. }
  401. }
  402. return repo.Owner
  403. }
  404. // ComposeMetas composes a map of metas for properly rendering issue links and external issue trackers.
  405. func (repo *Repository) ComposeMetas() map[string]string {
  406. if len(repo.RenderingMetas) == 0 {
  407. metas := map[string]string{
  408. "user": repo.OwnerName,
  409. "repo": repo.Name,
  410. "repoPath": repo.RepoPath(),
  411. "mode": "comment",
  412. }
  413. unit, err := repo.GetUnit(UnitTypeExternalTracker)
  414. if err == nil {
  415. metas["format"] = unit.ExternalTrackerConfig().ExternalTrackerFormat
  416. switch unit.ExternalTrackerConfig().ExternalTrackerStyle {
  417. case markup.IssueNameStyleAlphanumeric:
  418. metas["style"] = markup.IssueNameStyleAlphanumeric
  419. default:
  420. metas["style"] = markup.IssueNameStyleNumeric
  421. }
  422. }
  423. repo.MustOwner()
  424. if repo.Owner.IsOrganization() {
  425. teams := make([]string, 0, 5)
  426. _ = x.Table("team_repo").
  427. Join("INNER", "team", "team.id = team_repo.team_id").
  428. Where("team_repo.repo_id = ?", repo.ID).
  429. Select("team.lower_name").
  430. OrderBy("team.lower_name").
  431. Find(&teams)
  432. metas["teams"] = "," + strings.Join(teams, ",") + ","
  433. metas["org"] = strings.ToLower(repo.OwnerName)
  434. }
  435. repo.RenderingMetas = metas
  436. }
  437. return repo.RenderingMetas
  438. }
  439. // ComposeDocumentMetas composes a map of metas for properly rendering documents
  440. func (repo *Repository) ComposeDocumentMetas() map[string]string {
  441. if len(repo.DocumentRenderingMetas) == 0 {
  442. metas := map[string]string{}
  443. for k, v := range repo.ComposeMetas() {
  444. metas[k] = v
  445. }
  446. metas["mode"] = "document"
  447. repo.DocumentRenderingMetas = metas
  448. }
  449. return repo.DocumentRenderingMetas
  450. }
  451. // DeleteWiki removes the actual and local copy of repository wiki.
  452. func (repo *Repository) DeleteWiki() error {
  453. return repo.deleteWiki(x)
  454. }
  455. func (repo *Repository) deleteWiki(e Engine) error {
  456. wikiPaths := []string{repo.WikiPath()}
  457. for _, wikiPath := range wikiPaths {
  458. removeAllWithNotice(e, "Delete repository wiki", wikiPath)
  459. }
  460. _, err := e.Where("repo_id = ?", repo.ID).And("type = ?", UnitTypeWiki).Delete(new(RepoUnit))
  461. return err
  462. }
  463. func (repo *Repository) getAssignees(e Engine) (_ []*User, err error) {
  464. if err = repo.getOwner(e); err != nil {
  465. return nil, err
  466. }
  467. accesses := make([]*Access, 0, 10)
  468. if err = e.
  469. Where("repo_id = ? AND mode >= ?", repo.ID, AccessModeWrite).
  470. Find(&accesses); err != nil {
  471. return nil, err
  472. }
  473. // Leave a seat for owner itself to append later, but if owner is an organization
  474. // and just waste 1 unit is cheaper than re-allocate memory once.
  475. users := make([]*User, 0, len(accesses)+1)
  476. if len(accesses) > 0 {
  477. userIDs := make([]int64, len(accesses))
  478. for i := 0; i < len(accesses); i++ {
  479. userIDs[i] = accesses[i].UserID
  480. }
  481. if err = e.In("id", userIDs).Find(&users); err != nil {
  482. return nil, err
  483. }
  484. }
  485. if !repo.Owner.IsOrganization() {
  486. users = append(users, repo.Owner)
  487. }
  488. return users, nil
  489. }
  490. // GetAssignees returns all users that have write access and can be assigned to issues
  491. // of the repository,
  492. func (repo *Repository) GetAssignees() (_ []*User, err error) {
  493. return repo.getAssignees(x)
  494. }
  495. func (repo *Repository) getReviewers(e Engine, doerID, posterID int64) ([]*User, error) {
  496. // Get the owner of the repository - this often already pre-cached and if so saves complexity for the following queries
  497. if err := repo.getOwner(e); err != nil {
  498. return nil, err
  499. }
  500. var users []*User
  501. if repo.IsPrivate ||
  502. (repo.Owner.IsOrganization() && repo.Owner.Visibility == api.VisibleTypePrivate) {
  503. // This a private repository:
  504. // Anyone who can read the repository is a requestable reviewer
  505. if err := e.
  506. SQL("SELECT * FROM `user` WHERE id in (SELECT user_id FROM `access` WHERE repo_id = ? AND mode >= ? AND user_id NOT IN ( ?, ?)) ORDER BY name",
  507. repo.ID, AccessModeRead,
  508. doerID, posterID).
  509. Find(&users); err != nil {
  510. return nil, err
  511. }
  512. return users, nil
  513. }
  514. // This is a "public" repository:
  515. // Any user that has read access, is a watcher or organization member can be requested to review
  516. if err := e.
  517. SQL("SELECT * FROM `user` WHERE id IN ( "+
  518. "SELECT user_id FROM `access` WHERE repo_id = ? AND mode >= ? "+
  519. "UNION "+
  520. "SELECT user_id FROM `watch` WHERE repo_id = ? AND mode IN (?, ?) "+
  521. "UNION "+
  522. "SELECT uid AS user_id FROM `org_user` WHERE org_id = ? "+
  523. ") AND id NOT IN (?, ?) ORDER BY name",
  524. repo.ID, AccessModeRead,
  525. repo.ID, RepoWatchModeNormal, RepoWatchModeAuto,
  526. repo.OwnerID,
  527. doerID, posterID).
  528. Find(&users); err != nil {
  529. return nil, err
  530. }
  531. return users, nil
  532. }
  533. // GetReviewers get all users can be requested to review:
  534. // * for private repositories this returns all users that have read access or higher to the repository.
  535. // * for public repositories this returns all users that have read access or higher to the repository,
  536. // all repo watchers and all organization members.
  537. // TODO: may be we should have a busy choice for users to block review request to them.
  538. func (repo *Repository) GetReviewers(doerID, posterID int64) ([]*User, error) {
  539. return repo.getReviewers(x, doerID, posterID)
  540. }
  541. // GetReviewerTeams get all teams can be requested to review
  542. func (repo *Repository) GetReviewerTeams() ([]*Team, error) {
  543. if err := repo.GetOwner(); err != nil {
  544. return nil, err
  545. }
  546. if !repo.Owner.IsOrganization() {
  547. return nil, nil
  548. }
  549. teams, err := GetTeamsWithAccessToRepo(repo.OwnerID, repo.ID, AccessModeRead)
  550. if err != nil {
  551. return nil, err
  552. }
  553. return teams, err
  554. }
  555. // GetMilestoneByID returns the milestone belongs to repository by given ID.
  556. func (repo *Repository) GetMilestoneByID(milestoneID int64) (*Milestone, error) {
  557. return GetMilestoneByRepoID(repo.ID, milestoneID)
  558. }
  559. // IssueStats returns number of open and closed repository issues by given filter mode.
  560. func (repo *Repository) IssueStats(uid int64, filterMode int, isPull bool) (int64, int64) {
  561. return GetRepoIssueStats(repo.ID, uid, filterMode, isPull)
  562. }
  563. // GetMirror sets the repository mirror, returns an error upon failure
  564. func (repo *Repository) GetMirror() (err error) {
  565. repo.Mirror, err = GetMirrorByRepoID(repo.ID)
  566. return err
  567. }
  568. // GetBaseRepo populates repo.BaseRepo for a fork repository and
  569. // returns an error on failure (NOTE: no error is returned for
  570. // non-fork repositories, and BaseRepo will be left untouched)
  571. func (repo *Repository) GetBaseRepo() (err error) {
  572. return repo.getBaseRepo(x)
  573. }
  574. func (repo *Repository) getBaseRepo(e Engine) (err error) {
  575. if !repo.IsFork {
  576. return nil
  577. }
  578. repo.BaseRepo, err = getRepositoryByID(e, repo.ForkID)
  579. return err
  580. }
  581. // IsGenerated returns whether _this_ repository was generated from a template
  582. func (repo *Repository) IsGenerated() bool {
  583. return repo.TemplateID != 0
  584. }
  585. // GetTemplateRepo populates repo.TemplateRepo for a generated repository and
  586. // returns an error on failure (NOTE: no error is returned for
  587. // non-generated repositories, and TemplateRepo will be left untouched)
  588. func (repo *Repository) GetTemplateRepo() (err error) {
  589. return repo.getTemplateRepo(x)
  590. }
  591. func (repo *Repository) getTemplateRepo(e Engine) (err error) {
  592. if !repo.IsGenerated() {
  593. return nil
  594. }
  595. repo.TemplateRepo, err = getRepositoryByID(e, repo.TemplateID)
  596. return err
  597. }
  598. // RepoPath returns the repository path
  599. func (repo *Repository) RepoPath() string {
  600. return RepoPath(repo.OwnerName, repo.Name)
  601. }
  602. // GitConfigPath returns the path to a repository's git config/ directory
  603. func GitConfigPath(repoPath string) string {
  604. return filepath.Join(repoPath, "config")
  605. }
  606. // GitConfigPath returns the repository git config path
  607. func (repo *Repository) GitConfigPath() string {
  608. return GitConfigPath(repo.RepoPath())
  609. }
  610. // RelLink returns the repository relative link
  611. func (repo *Repository) RelLink() string {
  612. return "/" + repo.FullName()
  613. }
  614. // Link returns the repository link
  615. func (repo *Repository) Link() string {
  616. return setting.AppSubURL + "/" + repo.FullName()
  617. }
  618. // ComposeCompareURL returns the repository comparison URL
  619. func (repo *Repository) ComposeCompareURL(oldCommitID, newCommitID string) string {
  620. return fmt.Sprintf("%s/compare/%s...%s", repo.FullName(), oldCommitID, newCommitID)
  621. }
  622. // UpdateDefaultBranch updates the default branch
  623. func (repo *Repository) UpdateDefaultBranch() error {
  624. _, err := x.ID(repo.ID).Cols("default_branch").Update(repo)
  625. return err
  626. }
  627. // IsOwnedBy returns true when user owns this repository
  628. func (repo *Repository) IsOwnedBy(userID int64) bool {
  629. return repo.OwnerID == userID
  630. }
  631. func (repo *Repository) updateSize(e Engine) error {
  632. size, err := util.GetDirectorySize(repo.RepoPath())
  633. if err != nil {
  634. return fmt.Errorf("updateSize: %v", err)
  635. }
  636. objs, err := repo.GetLFSMetaObjects(-1, 0)
  637. if err != nil {
  638. return fmt.Errorf("updateSize: GetLFSMetaObjects: %v", err)
  639. }
  640. for _, obj := range objs {
  641. size += obj.Size
  642. }
  643. repo.Size = size
  644. _, err = e.ID(repo.ID).Cols("size").NoAutoTime().Update(repo)
  645. return err
  646. }
  647. // UpdateSize updates the repository size, calculating it using util.GetDirectorySize
  648. func (repo *Repository) UpdateSize(ctx DBContext) error {
  649. return repo.updateSize(ctx.e)
  650. }
  651. // CanUserFork returns true if specified user can fork repository.
  652. func (repo *Repository) CanUserFork(user *User) (bool, error) {
  653. if user == nil {
  654. return false, nil
  655. }
  656. if repo.OwnerID != user.ID && !user.HasForkedRepo(repo.ID) {
  657. return true, nil
  658. }
  659. if err := user.GetOwnedOrganizations(); err != nil {
  660. return false, err
  661. }
  662. for _, org := range user.OwnedOrgs {
  663. if repo.OwnerID != org.ID && !org.HasForkedRepo(repo.ID) {
  664. return true, nil
  665. }
  666. }
  667. return false, nil
  668. }
  669. // CanUserDelete returns true if user could delete the repository
  670. func (repo *Repository) CanUserDelete(user *User) (bool, error) {
  671. if user.IsAdmin || user.ID == repo.OwnerID {
  672. return true, nil
  673. }
  674. if err := repo.GetOwner(); err != nil {
  675. return false, err
  676. }
  677. if repo.Owner.IsOrganization() {
  678. isOwner, err := repo.Owner.IsOwnedBy(user.ID)
  679. if err != nil {
  680. return false, err
  681. } else if isOwner {
  682. return true, nil
  683. }
  684. }
  685. return false, nil
  686. }
  687. // CanEnablePulls returns true if repository meets the requirements of accepting pulls.
  688. func (repo *Repository) CanEnablePulls() bool {
  689. return !repo.IsMirror && !repo.IsEmpty
  690. }
  691. // AllowsPulls returns true if repository meets the requirements of accepting pulls and has them enabled.
  692. func (repo *Repository) AllowsPulls() bool {
  693. return repo.CanEnablePulls() && repo.UnitEnabled(UnitTypePullRequests)
  694. }
  695. // CanEnableEditor returns true if repository meets the requirements of web editor.
  696. func (repo *Repository) CanEnableEditor() bool {
  697. return !repo.IsMirror
  698. }
  699. // GetReaders returns all users that have explicit read access or higher to the repository.
  700. func (repo *Repository) GetReaders() (_ []*User, err error) {
  701. return repo.getUsersWithAccessMode(x, AccessModeRead)
  702. }
  703. // GetWriters returns all users that have write access to the repository.
  704. func (repo *Repository) GetWriters() (_ []*User, err error) {
  705. return repo.getUsersWithAccessMode(x, AccessModeWrite)
  706. }
  707. // IsReader returns true if user has explicit read access or higher to the repository.
  708. func (repo *Repository) IsReader(userID int64) (bool, error) {
  709. if repo.OwnerID == userID {
  710. return true, nil
  711. }
  712. return x.Where("repo_id = ? AND user_id = ? AND mode >= ?", repo.ID, userID, AccessModeRead).Get(&Access{})
  713. }
  714. // getUsersWithAccessMode returns users that have at least given access mode to the repository.
  715. func (repo *Repository) getUsersWithAccessMode(e Engine, mode AccessMode) (_ []*User, err error) {
  716. if err = repo.getOwner(e); err != nil {
  717. return nil, err
  718. }
  719. accesses := make([]*Access, 0, 10)
  720. if err = e.Where("repo_id = ? AND mode >= ?", repo.ID, mode).Find(&accesses); err != nil {
  721. return nil, err
  722. }
  723. // Leave a seat for owner itself to append later, but if owner is an organization
  724. // and just waste 1 unit is cheaper than re-allocate memory once.
  725. users := make([]*User, 0, len(accesses)+1)
  726. if len(accesses) > 0 {
  727. userIDs := make([]int64, len(accesses))
  728. for i := 0; i < len(accesses); i++ {
  729. userIDs[i] = accesses[i].UserID
  730. }
  731. if err = e.In("id", userIDs).Find(&users); err != nil {
  732. return nil, err
  733. }
  734. }
  735. if !repo.Owner.IsOrganization() {
  736. users = append(users, repo.Owner)
  737. }
  738. return users, nil
  739. }
  740. // DescriptionHTML does special handles to description and return HTML string.
  741. func (repo *Repository) DescriptionHTML() template.HTML {
  742. desc, err := markup.RenderDescriptionHTML([]byte(repo.Description), repo.HTMLURL(), repo.ComposeMetas())
  743. if err != nil {
  744. log.Error("Failed to render description for %s (ID: %d): %v", repo.Name, repo.ID, err)
  745. return template.HTML(markup.Sanitize(repo.Description))
  746. }
  747. return template.HTML(markup.Sanitize(string(desc)))
  748. }
  749. // ReadBy sets repo to be visited by given user.
  750. func (repo *Repository) ReadBy(userID int64) error {
  751. return setRepoNotificationStatusReadIfUnread(x, userID, repo.ID)
  752. }
  753. func isRepositoryExist(e Engine, u *User, repoName string) (bool, error) {
  754. has, err := e.Get(&Repository{
  755. OwnerID: u.ID,
  756. LowerName: strings.ToLower(repoName),
  757. })
  758. if err != nil {
  759. return false, err
  760. }
  761. isDir, err := util.IsDir(RepoPath(u.Name, repoName))
  762. return has && isDir, err
  763. }
  764. // IsRepositoryExist returns true if the repository with given name under user has already existed.
  765. func IsRepositoryExist(u *User, repoName string) (bool, error) {
  766. return isRepositoryExist(x, u, repoName)
  767. }
  768. // CloneLink represents different types of clone URLs of repository.
  769. type CloneLink struct {
  770. SSH string
  771. HTTPS string
  772. Git string
  773. }
  774. // ComposeHTTPSCloneURL returns HTTPS clone URL based on given owner and repository name.
  775. func ComposeHTTPSCloneURL(owner, repo string) string {
  776. return fmt.Sprintf("%s%s/%s.git", setting.AppURL, url.PathEscape(owner), url.PathEscape(repo))
  777. }
  778. func (repo *Repository) cloneLink(isWiki bool) *CloneLink {
  779. repoName := repo.Name
  780. if isWiki {
  781. repoName += ".wiki"
  782. }
  783. sshUser := setting.RunUser
  784. if setting.SSH.StartBuiltinServer {
  785. sshUser = setting.SSH.BuiltinServerUser
  786. }
  787. cl := new(CloneLink)
  788. // if we have a ipv6 literal we need to put brackets around it
  789. // for the git cloning to work.
  790. sshDomain := setting.SSH.Domain
  791. ip := net.ParseIP(setting.SSH.Domain)
  792. if ip != nil && ip.To4() == nil {
  793. sshDomain = "[" + setting.SSH.Domain + "]"
  794. }
  795. if setting.SSH.Port != 22 {
  796. cl.SSH = fmt.Sprintf("ssh://%s@%s/%s/%s.git", sshUser, net.JoinHostPort(setting.SSH.Domain, strconv.Itoa(setting.SSH.Port)), repo.OwnerName, repoName)
  797. } else if setting.Repository.UseCompatSSHURI {
  798. cl.SSH = fmt.Sprintf("ssh://%s@%s/%s/%s.git", sshUser, sshDomain, repo.OwnerName, repoName)
  799. } else {
  800. cl.SSH = fmt.Sprintf("%s@%s:%s/%s.git", sshUser, sshDomain, repo.OwnerName, repoName)
  801. }
  802. cl.HTTPS = ComposeHTTPSCloneURL(repo.OwnerName, repoName)
  803. return cl
  804. }
  805. // CloneLink returns clone URLs of repository.
  806. func (repo *Repository) CloneLink() (cl *CloneLink) {
  807. return repo.cloneLink(false)
  808. }
  809. // CheckCreateRepository check if could created a repository
  810. func CheckCreateRepository(doer, u *User, name string, overwriteOrAdopt bool) error {
  811. if !doer.CanCreateRepo() {
  812. return ErrReachLimitOfRepo{u.MaxRepoCreation}
  813. }
  814. if err := IsUsableRepoName(name); err != nil {
  815. return err
  816. }
  817. has, err := isRepositoryExist(x, u, name)
  818. if err != nil {
  819. return fmt.Errorf("IsRepositoryExist: %v", err)
  820. } else if has {
  821. return ErrRepoAlreadyExist{u.Name, name}
  822. }
  823. isExist, err := util.IsExist(RepoPath(u.Name, name))
  824. if err != nil {
  825. log.Error("Unable to check if %s exists. Error: %v", RepoPath(u.Name, name), err)
  826. return err
  827. }
  828. if !overwriteOrAdopt && isExist {
  829. return ErrRepoFilesAlreadyExist{u.Name, name}
  830. }
  831. return nil
  832. }
  833. // CreateRepoOptions contains the create repository options
  834. type CreateRepoOptions struct {
  835. Name string
  836. Description string
  837. OriginalURL string
  838. GitServiceType api.GitServiceType
  839. Gitignores string
  840. IssueLabels string
  841. License string
  842. Readme string
  843. DefaultBranch string
  844. IsPrivate bool
  845. IsMirror bool
  846. IsTemplate bool
  847. AutoInit bool
  848. Status RepositoryStatus
  849. TrustModel TrustModelType
  850. MirrorInterval string
  851. }
  852. // GetRepoInitFile returns repository init files
  853. func GetRepoInitFile(tp, name string) ([]byte, error) {
  854. cleanedName := strings.TrimLeft(path.Clean("/"+name), "/")
  855. relPath := path.Join("options", tp, cleanedName)
  856. // Use custom file when available.
  857. customPath := path.Join(setting.CustomPath, relPath)
  858. isFile, err := util.IsFile(customPath)
  859. if err != nil {
  860. log.Error("Unable to check if %s is a file. Error: %v", customPath, err)
  861. }
  862. if isFile {
  863. return ioutil.ReadFile(customPath)
  864. }
  865. switch tp {
  866. case "readme":
  867. return options.Readme(cleanedName)
  868. case "gitignore":
  869. return options.Gitignore(cleanedName)
  870. case "license":
  871. return options.License(cleanedName)
  872. case "label":
  873. return options.Labels(cleanedName)
  874. default:
  875. return []byte{}, fmt.Errorf("Invalid init file type")
  876. }
  877. }
  878. var (
  879. reservedRepoNames = []string{".", ".."}
  880. reservedRepoPatterns = []string{"*.git", "*.wiki"}
  881. )
  882. // IsUsableRepoName returns true when repository is usable
  883. func IsUsableRepoName(name string) error {
  884. if alphaDashDotPattern.MatchString(name) {
  885. // Note: usually this error is normally caught up earlier in the UI
  886. return ErrNameCharsNotAllowed{Name: name}
  887. }
  888. return isUsableName(reservedRepoNames, reservedRepoPatterns, name)
  889. }
  890. // CreateRepository creates a repository for the user/organization.
  891. func CreateRepository(ctx DBContext, doer, u *User, repo *Repository, overwriteOrAdopt bool) (err error) {
  892. if err = IsUsableRepoName(repo.Name); err != nil {
  893. return err
  894. }
  895. has, err := isRepositoryExist(ctx.e, u, repo.Name)
  896. if err != nil {
  897. return fmt.Errorf("IsRepositoryExist: %v", err)
  898. } else if has {
  899. return ErrRepoAlreadyExist{u.Name, repo.Name}
  900. }
  901. repoPath := RepoPath(u.Name, repo.Name)
  902. isExist, err := util.IsExist(repoPath)
  903. if err != nil {
  904. log.Error("Unable to check if %s exists. Error: %v", repoPath, err)
  905. return err
  906. }
  907. if !overwriteOrAdopt && isExist {
  908. log.Error("Files already exist in %s and we are not going to adopt or delete.", repoPath)
  909. return ErrRepoFilesAlreadyExist{
  910. Uname: u.Name,
  911. Name: repo.Name,
  912. }
  913. }
  914. if _, err = ctx.e.Insert(repo); err != nil {
  915. return err
  916. }
  917. if err = deleteRepoRedirect(ctx.e, u.ID, repo.Name); err != nil {
  918. return err
  919. }
  920. // insert units for repo
  921. units := make([]RepoUnit, 0, len(DefaultRepoUnits))
  922. for _, tp := range DefaultRepoUnits {
  923. if tp == UnitTypeIssues {
  924. units = append(units, RepoUnit{
  925. RepoID: repo.ID,
  926. Type: tp,
  927. Config: &IssuesConfig{
  928. EnableTimetracker: setting.Service.DefaultEnableTimetracking,
  929. AllowOnlyContributorsToTrackTime: setting.Service.DefaultAllowOnlyContributorsToTrackTime,
  930. EnableDependencies: setting.Service.DefaultEnableDependencies,
  931. },
  932. })
  933. } else if tp == UnitTypePullRequests {
  934. units = append(units, RepoUnit{
  935. RepoID: repo.ID,
  936. Type: tp,
  937. Config: &PullRequestsConfig{AllowMerge: true, AllowRebase: true, AllowRebaseMerge: true, AllowSquash: true},
  938. })
  939. } else {
  940. units = append(units, RepoUnit{
  941. RepoID: repo.ID,
  942. Type: tp,
  943. })
  944. }
  945. }
  946. if _, err = ctx.e.Insert(&units); err != nil {
  947. return err
  948. }
  949. // Remember visibility preference.
  950. u.LastRepoVisibility = repo.IsPrivate
  951. if err = updateUserCols(ctx.e, u, "last_repo_visibility"); err != nil {
  952. return fmt.Errorf("updateUser: %v", err)
  953. }
  954. if _, err = ctx.e.Incr("num_repos").ID(u.ID).Update(new(User)); err != nil {
  955. return fmt.Errorf("increment user total_repos: %v", err)
  956. }
  957. u.NumRepos++
  958. // Give access to all members in teams with access to all repositories.
  959. if u.IsOrganization() {
  960. if err := u.GetTeams(&SearchTeamOptions{}); err != nil {
  961. return fmt.Errorf("GetTeams: %v", err)
  962. }
  963. for _, t := range u.Teams {
  964. if t.IncludesAllRepositories {
  965. if err := t.addRepository(ctx.e, repo); err != nil {
  966. return fmt.Errorf("addRepository: %v", err)
  967. }
  968. }
  969. }
  970. if isAdmin, err := isUserRepoAdmin(ctx.e, repo, doer); err != nil {
  971. return fmt.Errorf("isUserRepoAdmin: %v", err)
  972. } else if !isAdmin {
  973. // Make creator repo admin if it wan't assigned automatically
  974. if err = repo.addCollaborator(ctx.e, doer); err != nil {
  975. return fmt.Errorf("AddCollaborator: %v", err)
  976. }
  977. if err = repo.changeCollaborationAccessMode(ctx.e, doer.ID, AccessModeAdmin); err != nil {
  978. return fmt.Errorf("ChangeCollaborationAccessMode: %v", err)
  979. }
  980. }
  981. } else if err = repo.recalculateAccesses(ctx.e); err != nil {
  982. // Organization automatically called this in addRepository method.
  983. return fmt.Errorf("recalculateAccesses: %v", err)
  984. }
  985. if setting.Service.AutoWatchNewRepos {
  986. if err = watchRepo(ctx.e, doer.ID, repo.ID, true); err != nil {
  987. return fmt.Errorf("watchRepo: %v", err)
  988. }
  989. }
  990. if err = copyDefaultWebhooksToRepo(ctx.e, repo.ID); err != nil {
  991. return fmt.Errorf("copyDefaultWebhooksToRepo: %v", err)
  992. }
  993. return nil
  994. }
  995. func countRepositories(userID int64, private bool) int64 {
  996. sess := x.Where("id > 0")
  997. if userID > 0 {
  998. sess.And("owner_id = ?", userID)
  999. }
  1000. if !private {
  1001. sess.And("is_private=?", false)
  1002. }
  1003. count, err := sess.Count(new(Repository))
  1004. if err != nil {
  1005. log.Error("countRepositories: %v", err)
  1006. }
  1007. return count
  1008. }
  1009. // CountRepositories returns number of repositories.
  1010. // Argument private only takes effect when it is false,
  1011. // set it true to count all repositories.
  1012. func CountRepositories(private bool) int64 {
  1013. return countRepositories(-1, private)
  1014. }
  1015. // CountUserRepositories returns number of repositories user owns.
  1016. // Argument private only takes effect when it is false,
  1017. // set it true to count all repositories.
  1018. func CountUserRepositories(userID int64, private bool) int64 {
  1019. return countRepositories(userID, private)
  1020. }
  1021. // RepoPath returns repository path by given user and repository name.
  1022. func RepoPath(userName, repoName string) string {
  1023. return filepath.Join(UserPath(userName), strings.ToLower(repoName)+".git")
  1024. }
  1025. // IncrementRepoForkNum increment repository fork number
  1026. func IncrementRepoForkNum(ctx DBContext, repoID int64) error {
  1027. _, err := ctx.e.Exec("UPDATE `repository` SET num_forks=num_forks+1 WHERE id=?", repoID)
  1028. return err
  1029. }
  1030. // ChangeRepositoryName changes all corresponding setting from old repository name to new one.
  1031. func ChangeRepositoryName(doer *User, repo *Repository, newRepoName string) (err error) {
  1032. oldRepoName := repo.Name
  1033. newRepoName = strings.ToLower(newRepoName)
  1034. if err = IsUsableRepoName(newRepoName); err != nil {
  1035. return err
  1036. }
  1037. if err := repo.GetOwner(); err != nil {
  1038. return err
  1039. }
  1040. has, err := IsRepositoryExist(repo.Owner, newRepoName)
  1041. if err != nil {
  1042. return fmt.Errorf("IsRepositoryExist: %v", err)
  1043. } else if has {
  1044. return ErrRepoAlreadyExist{repo.Owner.Name, newRepoName}
  1045. }
  1046. newRepoPath := RepoPath(repo.Owner.Name, newRepoName)
  1047. if err = os.Rename(repo.RepoPath(), newRepoPath); err != nil {
  1048. return fmt.Errorf("rename repository directory: %v", err)
  1049. }
  1050. wikiPath := repo.WikiPath()
  1051. isExist, err := util.IsExist(wikiPath)
  1052. if err != nil {
  1053. log.Error("Unable to check if %s exists. Error: %v", wikiPath, err)
  1054. return err
  1055. }
  1056. if isExist {
  1057. if err = os.Rename(wikiPath, WikiPath(repo.Owner.Name, newRepoName)); err != nil {
  1058. return fmt.Errorf("rename repository wiki: %v", err)
  1059. }
  1060. }
  1061. sess := x.NewSession()
  1062. defer sess.Close()
  1063. if err = sess.Begin(); err != nil {
  1064. return fmt.Errorf("sess.Begin: %v", err)
  1065. }
  1066. if err := newRepoRedirect(sess, repo.Owner.ID, repo.ID, oldRepoName, newRepoName); err != nil {
  1067. return err
  1068. }
  1069. return sess.Commit()
  1070. }
  1071. func getRepositoriesByForkID(e Engine, forkID int64) ([]*Repository, error) {
  1072. repos := make([]*Repository, 0, 10)
  1073. return repos, e.
  1074. Where("fork_id=?", forkID).
  1075. Find(&repos)
  1076. }
  1077. // GetRepositoriesByForkID returns all repositories with given fork ID.
  1078. func GetRepositoriesByForkID(forkID int64) ([]*Repository, error) {
  1079. return getRepositoriesByForkID(x, forkID)
  1080. }
  1081. func updateRepository(e Engine, repo *Repository, visibilityChanged bool) (err error) {
  1082. repo.LowerName = strings.ToLower(repo.Name)
  1083. if utf8.RuneCountInString(repo.Description) > 255 {
  1084. repo.Description = string([]rune(repo.Description)[:255])
  1085. }
  1086. if utf8.RuneCountInString(repo.Website) > 255 {
  1087. repo.Website = string([]rune(repo.Website)[:255])
  1088. }
  1089. if _, err = e.ID(repo.ID).AllCols().Update(repo); err != nil {
  1090. return fmt.Errorf("update: %v", err)
  1091. }
  1092. if err = repo.updateSize(e); err != nil {
  1093. log.Error("Failed to update size for repository: %v", err)
  1094. }
  1095. if visibilityChanged {
  1096. if err = repo.getOwner(e); err != nil {
  1097. return fmt.Errorf("getOwner: %v", err)
  1098. }
  1099. if repo.Owner.IsOrganization() {
  1100. // Organization repository need to recalculate access table when visibility is changed.
  1101. if err = repo.recalculateTeamAccesses(e, 0); err != nil {
  1102. return fmt.Errorf("recalculateTeamAccesses: %v", err)
  1103. }
  1104. }
  1105. // If repo has become private, we need to set its actions to private.
  1106. if repo.IsPrivate {
  1107. _, err = e.Where("repo_id = ?", repo.ID).Cols("is_private").Update(&Action{
  1108. IsPrivate: true,
  1109. })
  1110. if err != nil {
  1111. return err
  1112. }
  1113. }
  1114. // Create/Remove git-daemon-export-ok for git-daemon...
  1115. daemonExportFile := path.Join(repo.RepoPath(), `git-daemon-export-ok`)
  1116. isExist, err := util.IsExist(daemonExportFile)
  1117. if err != nil {
  1118. log.Error("Unable to check if %s exists. Error: %v", daemonExportFile, err)
  1119. return err
  1120. }
  1121. if repo.IsPrivate && isExist {
  1122. if err = util.Remove(daemonExportFile); err != nil {
  1123. log.Error("Failed to remove %s: %v", daemonExportFile, err)
  1124. }
  1125. } else if !repo.IsPrivate && !isExist {
  1126. if f, err := os.Create(daemonExportFile); err != nil {
  1127. log.Error("Failed to create %s: %v", daemonExportFile, err)
  1128. } else {
  1129. f.Close()
  1130. }
  1131. }
  1132. forkRepos, err := getRepositoriesByForkID(e, repo.ID)
  1133. if err != nil {
  1134. return fmt.Errorf("getRepositoriesByForkID: %v", err)
  1135. }
  1136. for i := range forkRepos {
  1137. forkRepos[i].IsPrivate = repo.IsPrivate || repo.Owner.Visibility == api.VisibleTypePrivate
  1138. if err = updateRepository(e, forkRepos[i], true); err != nil {
  1139. return fmt.Errorf("updateRepository[%d]: %v", forkRepos[i].ID, err)
  1140. }
  1141. }
  1142. }
  1143. return nil
  1144. }
  1145. // UpdateRepositoryCtx updates a repository with db context
  1146. func UpdateRepositoryCtx(ctx DBContext, repo *Repository, visibilityChanged bool) error {
  1147. return updateRepository(ctx.e, repo, visibilityChanged)
  1148. }
  1149. // UpdateRepository updates a repository
  1150. func UpdateRepository(repo *Repository, visibilityChanged bool) (err error) {
  1151. sess := x.NewSession()
  1152. defer sess.Close()
  1153. if err = sess.Begin(); err != nil {
  1154. return err
  1155. }
  1156. if err = updateRepository(sess, repo, visibilityChanged); err != nil {
  1157. return fmt.Errorf("updateRepository: %v", err)
  1158. }
  1159. return sess.Commit()
  1160. }
  1161. // UpdateRepositoryOwnerNames updates repository owner_names (this should only be used when the ownerName has changed case)
  1162. func UpdateRepositoryOwnerNames(ownerID int64, ownerName string) error {
  1163. if ownerID == 0 {
  1164. return nil
  1165. }
  1166. sess := x.NewSession()
  1167. defer sess.Close()
  1168. if err := sess.Begin(); err != nil {
  1169. return err
  1170. }
  1171. if _, err := sess.Where("owner_id = ?", ownerID).Cols("owner_name").Update(&Repository{
  1172. OwnerName: ownerName,
  1173. }); err != nil {
  1174. return err
  1175. }
  1176. return sess.Commit()
  1177. }
  1178. // UpdateRepositoryUpdatedTime updates a repository's updated time
  1179. func UpdateRepositoryUpdatedTime(repoID int64, updateTime time.Time) error {
  1180. _, err := x.Exec("UPDATE repository SET updated_unix = ? WHERE id = ?", updateTime.Unix(), repoID)
  1181. return err
  1182. }
  1183. // UpdateRepositoryUnits updates a repository's units
  1184. func UpdateRepositoryUnits(repo *Repository, units []RepoUnit, deleteUnitTypes []UnitType) (err error) {
  1185. sess := x.NewSession()
  1186. defer sess.Close()
  1187. if err = sess.Begin(); err != nil {
  1188. return err
  1189. }
  1190. // Delete existing settings of units before adding again
  1191. for _, u := range units {
  1192. deleteUnitTypes = append(deleteUnitTypes, u.Type)
  1193. }
  1194. if _, err = sess.Where("repo_id = ?", repo.ID).In("type", deleteUnitTypes).Delete(new(RepoUnit)); err != nil {
  1195. return err
  1196. }
  1197. if len(units) > 0 {
  1198. if _, err = sess.Insert(units); err != nil {
  1199. return err
  1200. }
  1201. }
  1202. return sess.Commit()
  1203. }
  1204. // DeleteRepository deletes a repository for a user or organization.
  1205. // make sure if you call this func to close open sessions (sqlite will otherwise get a deadlock)
  1206. func DeleteRepository(doer *User, uid, repoID int64) error {
  1207. sess := x.NewSession()
  1208. defer sess.Close()
  1209. if err := sess.Begin(); err != nil {
  1210. return err
  1211. }
  1212. // In case is a organization.
  1213. org, err := getUserByID(sess, uid)
  1214. if err != nil {
  1215. return err
  1216. }
  1217. if org.IsOrganization() {
  1218. if err = org.getTeams(sess); err != nil {
  1219. return err
  1220. }
  1221. }
  1222. repo := &Repository{OwnerID: uid}
  1223. has, err := sess.ID(repoID).Get(repo)
  1224. if err != nil {
  1225. return err
  1226. } else if !has {
  1227. return ErrRepoNotExist{repoID, uid, "", ""}
  1228. }
  1229. // Delete Deploy Keys
  1230. deployKeys, err := listDeployKeys(sess, repo.ID, ListOptions{})
  1231. if err != nil {
  1232. return fmt.Errorf("listDeployKeys: %v", err)
  1233. }
  1234. for _, dKey := range deployKeys {
  1235. if err := deleteDeployKey(sess, doer, dKey.ID); err != nil {
  1236. return fmt.Errorf("deleteDeployKeys: %v", err)
  1237. }
  1238. }
  1239. if cnt, err := sess.ID(repoID).Delete(&Repository{}); err != nil {
  1240. return err
  1241. } else if cnt != 1 {
  1242. return ErrRepoNotExist{repoID, uid, "", ""}
  1243. }
  1244. if org.IsOrganization() {
  1245. for _, t := range org.Teams {
  1246. if !t.hasRepository(sess, repoID) {
  1247. continue
  1248. } else if err = t.removeRepository(sess, repo, false); err != nil {
  1249. return err
  1250. }
  1251. }
  1252. }
  1253. attachments := make([]*Attachment, 0, 20)
  1254. if err = sess.Join("INNER", "`release`", "`release`.id = `attachment`.release_id").
  1255. Where("`release`.repo_id = ?", repoID).
  1256. Find(&attachments); err != nil {
  1257. return err
  1258. }
  1259. releaseAttachments := make([]string, 0, len(attachments))
  1260. for i := 0; i < len(attachments); i++ {
  1261. releaseAttachments = append(releaseAttachments, attachments[i].RelativePath())
  1262. }
  1263. if _, err := sess.Exec("UPDATE `user` SET num_stars=num_stars-1 WHERE id IN (SELECT `uid` FROM `star` WHERE repo_id = ?)", repo.ID); err != nil {
  1264. return err
  1265. }
  1266. if err := deleteBeans(sess,
  1267. &Access{RepoID: repo.ID},
  1268. &Action{RepoID: repo.ID},
  1269. &Collaboration{RepoID: repoID},
  1270. &Comment{RefRepoID: repoID},
  1271. &CommitStatus{RepoID: repoID},
  1272. &DeletedBranch{RepoID: repoID},
  1273. &HookTask{RepoID: repoID},
  1274. &LFSLock{RepoID: repoID},
  1275. &LanguageStat{RepoID: repoID},
  1276. &Milestone{RepoID: repoID},
  1277. &Mirror{RepoID: repoID},
  1278. &Notification{RepoID: repoID},
  1279. &ProtectedBranch{RepoID: repoID},
  1280. &PullRequest{BaseRepoID: repoID},
  1281. &Release{RepoID: repoID},
  1282. &RepoIndexerStatus{RepoID: repoID},
  1283. &RepoRedirect{RedirectRepoID: repoID},
  1284. &RepoUnit{RepoID: repoID},
  1285. &Star{RepoID: repoID},
  1286. &Task{RepoID: repoID},
  1287. &Watch{RepoID: repoID},
  1288. &Webhook{RepoID: repoID},
  1289. ); err != nil {
  1290. return fmt.Errorf("deleteBeans: %v", err)
  1291. }
  1292. // Delete Labels and related objects
  1293. if err := deleteLabelsByRepoID(sess, repoID); err != nil {
  1294. return err
  1295. }
  1296. // Delete Issues and related objects
  1297. var attachmentPaths []string
  1298. if attachmentPaths, err = deleteIssuesByRepoID(sess, repoID); err != nil {
  1299. return err
  1300. }
  1301. if repo.IsFork {
  1302. if _, err := sess.Exec("UPDATE `repository` SET num_forks=num_forks-1 WHERE id=?", repo.ForkID); err != nil {
  1303. return fmt.Errorf("decrease fork count: %v", err)
  1304. }
  1305. }
  1306. if _, err := sess.Exec("UPDATE `user` SET num_repos=num_repos-1 WHERE id=?", uid); err != nil {
  1307. return err
  1308. }
  1309. if len(repo.Topics) > 0 {
  1310. if err := removeTopicsFromRepo(sess, repo.ID); err != nil {
  1311. return err
  1312. }
  1313. }
  1314. projects, _, err := getProjects(sess, ProjectSearchOptions{
  1315. RepoID: repoID,
  1316. })
  1317. if err != nil {
  1318. return fmt.Errorf("get projects: %v", err)
  1319. }
  1320. for i := range projects {
  1321. if err := deleteProjectByID(sess, projects[i].ID); err != nil {
  1322. return fmt.Errorf("delete project [%d]: %v", projects[i].ID, err)
  1323. }
  1324. }
  1325. // FIXME: Remove repository files should be executed after transaction succeed.
  1326. repoPath := repo.RepoPath()
  1327. removeAllWithNotice(sess, "Delete repository files", repoPath)
  1328. err = repo.deleteWiki(sess)
  1329. if err != nil {
  1330. return err
  1331. }
  1332. // Remove LFS objects
  1333. var lfsObjects []*LFSMetaObject
  1334. if err = sess.Where("repository_id=?", repoID).Find(&lfsObjects); err != nil {
  1335. return err
  1336. }
  1337. for _, v := range lfsObjects {
  1338. count, err := sess.Count(&LFSMetaObject{Oid: v.Oid})
  1339. if err != nil {
  1340. return err
  1341. }
  1342. if count > 1 {
  1343. continue
  1344. }
  1345. removeStorageWithNotice(sess, storage.LFS, "Delete orphaned LFS file", v.RelativePath())
  1346. }
  1347. if _, err := sess.Delete(&LFSMetaObject{RepositoryID: repoID}); err != nil {
  1348. return err
  1349. }
  1350. if repo.NumForks > 0 {
  1351. if _, err = sess.Exec("UPDATE `repository` SET fork_id=0,is_fork=? WHERE fork_id=?", false, repo.ID); err != nil {
  1352. log.Error("reset 'fork_id' and 'is_fork': %v", err)
  1353. }
  1354. }
  1355. if err = sess.Commit(); err != nil {
  1356. return err
  1357. }
  1358. sess.Close()
  1359. // We should always delete the files after the database transaction succeed. If
  1360. // we delete the file but the database rollback, the repository will be borken.
  1361. // Remove issue attachment files.
  1362. for i := range attachmentPaths {
  1363. RemoveStorageWithNotice(storage.Attachments, "Delete issue attachment", attachmentPaths[i])
  1364. }
  1365. // Remove release attachment files.
  1366. for i := range releaseAttachments {
  1367. RemoveStorageWithNotice(storage.Attachments, "Delete release attachment", releaseAttachments[i])
  1368. }
  1369. if len(repo.Avatar) > 0 {
  1370. if err := storage.RepoAvatars.Delete(repo.CustomAvatarRelativePath()); err != nil {
  1371. return fmt.Errorf("Failed to remove %s: %v", repo.Avatar, err)
  1372. }
  1373. }
  1374. return nil
  1375. }
  1376. // GetRepositoryByOwnerAndName returns the repository by given ownername and reponame.
  1377. func GetRepositoryByOwnerAndName(ownerName, repoName string) (*Repository, error) {
  1378. return getRepositoryByOwnerAndName(x, ownerName, repoName)
  1379. }
  1380. func getRepositoryByOwnerAndName(e Engine, ownerName, repoName string) (*Repository, error) {
  1381. var repo Repository
  1382. has, err := e.Table("repository").Select("repository.*").
  1383. Join("INNER", "`user`", "`user`.id = repository.owner_id").
  1384. Where("repository.lower_name = ?", strings.ToLower(repoName)).
  1385. And("`user`.lower_name = ?", strings.ToLower(ownerName)).
  1386. Get(&repo)
  1387. if err != nil {
  1388. return nil, err
  1389. } else if !has {
  1390. return nil, ErrRepoNotExist{0, 0, ownerName, repoName}
  1391. }
  1392. return &repo, nil
  1393. }
  1394. // GetRepositoryByName returns the repository by given name under user if exists.
  1395. func GetRepositoryByName(ownerID int64, name string) (*Repository, error) {
  1396. repo := &Repository{
  1397. OwnerID: ownerID,
  1398. LowerName: strings.ToLower(name),
  1399. }
  1400. has, err := x.Get(repo)
  1401. if err != nil {
  1402. return nil, err
  1403. } else if !has {
  1404. return nil, ErrRepoNotExist{0, ownerID, "", name}
  1405. }
  1406. return repo, err
  1407. }
  1408. func getRepositoryByID(e Engine, id int64) (*Repository, error) {
  1409. repo := new(Repository)
  1410. has, err := e.ID(id).Get(repo)
  1411. if err != nil {
  1412. return nil, err
  1413. } else if !has {
  1414. return nil, ErrRepoNotExist{id, 0, "", ""}
  1415. }
  1416. return repo, nil
  1417. }
  1418. // GetRepositoryByID returns the repository by given id if exists.
  1419. func GetRepositoryByID(id int64) (*Repository, error) {
  1420. return getRepositoryByID(x, id)
  1421. }
  1422. // GetRepositoryByIDCtx returns the repository by given id if exists.
  1423. func GetRepositoryByIDCtx(ctx DBContext, id int64) (*Repository, error) {
  1424. return getRepositoryByID(ctx.e, id)
  1425. }
  1426. // GetRepositoriesMapByIDs returns the repositories by given id slice.
  1427. func GetRepositoriesMapByIDs(ids []int64) (map[int64]*Repository, error) {
  1428. repos := make(map[int64]*Repository, len(ids))
  1429. return repos, x.In("id", ids).Find(&repos)
  1430. }
  1431. // GetUserRepositories returns a list of repositories of given user.
  1432. func GetUserRepositories(opts *SearchRepoOptions) ([]*Repository, int64, error) {
  1433. if len(opts.OrderBy) == 0 {
  1434. opts.OrderBy = "updated_unix DESC"
  1435. }
  1436. cond := builder.NewCond()
  1437. cond = cond.And(builder.Eq{"owner_id": opts.Actor.ID})
  1438. if !opts.Private {
  1439. cond = cond.And(builder.Eq{"is_private": false})
  1440. }
  1441. if opts.LowerNames != nil && len(opts.LowerNames) > 0 {
  1442. cond = cond.And(builder.In("lower_name", opts.LowerNames))
  1443. }
  1444. sess := x.NewSession()
  1445. defer sess.Close()
  1446. count, err := sess.Where(cond).Count(new(Repository))
  1447. if err != nil {
  1448. return nil, 0, fmt.Errorf("Count: %v", err)
  1449. }
  1450. sess.Where(cond).OrderBy(opts.OrderBy.String())
  1451. repos := make([]*Repository, 0, opts.PageSize)
  1452. return repos, count, opts.setSessionPagination(sess).Find(&repos)
  1453. }
  1454. // GetUserMirrorRepositories returns a list of mirror repositories of given user.
  1455. func GetUserMirrorRepositories(userID int64) ([]*Repository, error) {
  1456. repos := make([]*Repository, 0, 10)
  1457. return repos, x.
  1458. Where("owner_id = ?", userID).
  1459. And("is_mirror = ?", true).
  1460. Find(&repos)
  1461. }
  1462. func getRepositoryCount(e Engine, u *User) (int64, error) {
  1463. return e.Count(&Repository{OwnerID: u.ID})
  1464. }
  1465. func getPublicRepositoryCount(e Engine, u *User) (int64, error) {
  1466. return e.Where("is_private = ?", false).Count(&Repository{OwnerID: u.ID})
  1467. }
  1468. func getPrivateRepositoryCount(e Engine, u *User) (int64, error) {
  1469. return e.Where("is_private = ?", true).Count(&Repository{OwnerID: u.ID})
  1470. }
  1471. // GetRepositoryCount returns the total number of repositories of user.
  1472. func GetRepositoryCount(u *User) (int64, error) {
  1473. return getRepositoryCount(x, u)
  1474. }
  1475. // GetPublicRepositoryCount returns the total number of public repositories of user.
  1476. func GetPublicRepositoryCount(u *User) (int64, error) {
  1477. return getPublicRepositoryCount(x, u)
  1478. }
  1479. // GetPrivateRepositoryCount returns the total number of private repositories of user.
  1480. func GetPrivateRepositoryCount(u *User) (int64, error) {
  1481. return getPrivateRepositoryCount(x, u)
  1482. }
  1483. // DeleteRepositoryArchives deletes all repositories' archives.
  1484. func DeleteRepositoryArchives(ctx context.Context) error {
  1485. return x.
  1486. Where("id > 0").
  1487. Iterate(new(Repository),
  1488. func(idx int, bean interface{}) error {
  1489. repo := bean.(*Repository)
  1490. select {
  1491. case <-ctx.Done():
  1492. return ErrCancelledf("before deleting repository archives for %s", repo.FullName())
  1493. default:
  1494. }
  1495. return util.RemoveAll(filepath.Join(repo.RepoPath(), "archives"))
  1496. })
  1497. }
  1498. // DeleteOldRepositoryArchives deletes old repository archives.
  1499. func DeleteOldRepositoryArchives(ctx context.Context, olderThan time.Duration) error {
  1500. log.Trace("Doing: ArchiveCleanup")
  1501. if err := x.Where("id > 0").Iterate(new(Repository), func(idx int, bean interface{}) error {
  1502. return deleteOldRepositoryArchives(ctx, olderThan, idx, bean)
  1503. }); err != nil {
  1504. log.Trace("Error: ArchiveClean: %v", err)
  1505. return err
  1506. }
  1507. log.Trace("Finished: ArchiveCleanup")
  1508. return nil
  1509. }
  1510. func deleteOldRepositoryArchives(ctx context.Context, olderThan time.Duration, idx int, bean interface{}) error {
  1511. repo := bean.(*Repository)
  1512. basePath := filepath.Join(repo.RepoPath(), "archives")
  1513. for _, ty := range []string{"zip", "targz"} {
  1514. select {
  1515. case <-ctx.Done():
  1516. return ErrCancelledf("before deleting old repository archives with filetype %s for %s", ty, repo.FullName())
  1517. default:
  1518. }
  1519. path := filepath.Join(basePath, ty)
  1520. file, err := os.Open(path)
  1521. if err != nil {
  1522. if !os.IsNotExist(err) {
  1523. log.Warn("Unable to open directory %s: %v", path, err)
  1524. return err
  1525. }
  1526. // If the directory doesn't exist, that's okay.
  1527. continue
  1528. }
  1529. files, err := file.Readdir(0)
  1530. file.Close()
  1531. if err != nil {
  1532. log.Warn("Unable to read directory %s: %v", path, err)
  1533. return err
  1534. }
  1535. minimumOldestTime := time.Now().Add(-olderThan)
  1536. for _, info := range files {
  1537. if info.ModTime().Before(minimumOldestTime) && !info.IsDir() {
  1538. select {
  1539. case <-ctx.Done():
  1540. return ErrCancelledf("before deleting old repository archive file %s with filetype %s for %s", info.Name(), ty, repo.FullName())
  1541. default:
  1542. }
  1543. toDelete := filepath.Join(path, info.Name())
  1544. // This is a best-effort purge, so we do not check error codes to confirm removal.
  1545. if err = util.Remove(toDelete); err != nil {
  1546. log.Trace("Unable to delete %s, but proceeding: %v", toDelete, err)
  1547. }
  1548. }
  1549. }
  1550. }
  1551. return nil
  1552. }
  1553. type repoChecker struct {
  1554. querySQL, correctSQL string
  1555. desc string
  1556. }
  1557. func repoStatsCheck(ctx context.Context, checker *repoChecker) {
  1558. results, err := x.Query(checker.querySQL)
  1559. if err != nil {
  1560. log.Error("Select %s: %v", checker.desc, err)
  1561. return
  1562. }
  1563. for _, result := range results {
  1564. id, _ := strconv.ParseInt(string(result["id"]), 10, 64)
  1565. select {
  1566. case <-ctx.Done():
  1567. log.Warn("CheckRepoStats: Cancelled before checking %s for Repo[%d]", checker.desc, id)
  1568. return
  1569. default:
  1570. }
  1571. log.Trace("Updating %s: %d", checker.desc, id)
  1572. _, err = x.Exec(checker.correctSQL, id, id)
  1573. if err != nil {
  1574. log.Error("Update %s[%d]: %v", checker.desc, id, err)
  1575. }
  1576. }
  1577. }
  1578. // CheckRepoStats checks the repository stats
  1579. func CheckRepoStats(ctx context.Context) error {
  1580. log.Trace("Doing: CheckRepoStats")
  1581. checkers := []*repoChecker{
  1582. // Repository.NumWatches
  1583. {
  1584. "SELECT repo.id FROM `repository` repo WHERE repo.num_watches!=(SELECT COUNT(*) FROM `watch` WHERE repo_id=repo.id AND mode<>2)",
  1585. "UPDATE `repository` SET num_watches=(SELECT COUNT(*) FROM `watch` WHERE repo_id=? AND mode<>2) WHERE id=?",
  1586. "repository count 'num_watches'",
  1587. },
  1588. // Repository.NumStars
  1589. {
  1590. "SELECT repo.id FROM `repository` repo WHERE repo.num_stars!=(SELECT COUNT(*) FROM `star` WHERE repo_id=repo.id)",
  1591. "UPDATE `repository` SET num_stars=(SELECT COUNT(*) FROM `star` WHERE repo_id=?) WHERE id=?",
  1592. "repository count 'num_stars'",
  1593. },
  1594. // Label.NumIssues
  1595. {
  1596. "SELECT label.id FROM `label` WHERE label.num_issues!=(SELECT COUNT(*) FROM `issue_label` WHERE label_id=label.id)",
  1597. "UPDATE `label` SET num_issues=(SELECT COUNT(*) FROM `issue_label` WHERE label_id=?) WHERE id=?",
  1598. "label count 'num_issues'",
  1599. },
  1600. // User.NumRepos
  1601. {
  1602. "SELECT `user`.id FROM `user` WHERE `user`.num_repos!=(SELECT COUNT(*) FROM `repository` WHERE owner_id=`user`.id)",
  1603. "UPDATE `user` SET num_repos=(SELECT COUNT(*) FROM `repository` WHERE owner_id=?) WHERE id=?",
  1604. "user count 'num_repos'",
  1605. },
  1606. // Issue.NumComments
  1607. {
  1608. "SELECT `issue`.id FROM `issue` WHERE `issue`.num_comments!=(SELECT COUNT(*) FROM `comment` WHERE issue_id=`issue`.id AND type=0)",
  1609. "UPDATE `issue` SET num_comments=(SELECT COUNT(*) FROM `comment` WHERE issue_id=? AND type=0) WHERE id=?",
  1610. "issue count 'num_comments'",
  1611. },
  1612. }
  1613. for _, checker := range checkers {
  1614. select {
  1615. case <-ctx.Done():
  1616. log.Warn("CheckRepoStats: Cancelled before %s", checker.desc)
  1617. return ErrCancelledf("before checking %s", checker.desc)
  1618. default:
  1619. repoStatsCheck(ctx, checker)
  1620. }
  1621. }
  1622. // ***** START: Repository.NumClosedIssues *****
  1623. desc := "repository count 'num_closed_issues'"
  1624. results, err := x.Query("SELECT repo.id FROM `repository` repo WHERE repo.num_closed_issues!=(SELECT COUNT(*) FROM `issue` WHERE repo_id=repo.id AND is_closed=? AND is_pull=?)", true, false)
  1625. if err != nil {
  1626. log.Error("Select %s: %v", desc, err)
  1627. } else {
  1628. for _, result := range results {
  1629. id, _ := strconv.ParseInt(string(result["id"]), 10, 64)
  1630. select {
  1631. case <-ctx.Done():
  1632. log.Warn("CheckRepoStats: Cancelled during %s for repo ID %d", desc, id)
  1633. return ErrCancelledf("during %s for repo ID %d", desc, id)
  1634. default:
  1635. }
  1636. log.Trace("Updating %s: %d", desc, id)
  1637. _, err = x.Exec("UPDATE `repository` SET num_closed_issues=(SELECT COUNT(*) FROM `issue` WHERE repo_id=? AND is_closed=? AND is_pull=?) WHERE id=?", id, true, false, id)
  1638. if err != nil {
  1639. log.Error("Update %s[%d]: %v", desc, id, err)
  1640. }
  1641. }
  1642. }
  1643. // ***** END: Repository.NumClosedIssues *****
  1644. // ***** START: Repository.NumClosedPulls *****
  1645. desc = "repository count 'num_closed_pulls'"
  1646. results, err = x.Query("SELECT repo.id FROM `repository` repo WHERE repo.num_closed_pulls!=(SELECT COUNT(*) FROM `issue` WHERE repo_id=repo.id AND is_closed=? AND is_pull=?)", true, true)
  1647. if err != nil {
  1648. log.Error("Select %s: %v", desc, err)
  1649. } else {
  1650. for _, result := range results {
  1651. id, _ := strconv.ParseInt(string(result["id"]), 10, 64)
  1652. select {
  1653. case <-ctx.Done():
  1654. log.Warn("CheckRepoStats: Cancelled")
  1655. return ErrCancelledf("during %s for repo ID %d", desc, id)
  1656. default:
  1657. }
  1658. log.Trace("Updating %s: %d", desc, id)
  1659. _, err = x.Exec("UPDATE `repository` SET num_closed_pulls=(SELECT COUNT(*) FROM `issue` WHERE repo_id=? AND is_closed=? AND is_pull=?) WHERE id=?", id, true, true, id)
  1660. if err != nil {
  1661. log.Error("Update %s[%d]: %v", desc, id, err)
  1662. }
  1663. }
  1664. }
  1665. // ***** END: Repository.NumClosedPulls *****
  1666. // FIXME: use checker when stop supporting old fork repo format.
  1667. // ***** START: Repository.NumForks *****
  1668. results, err = x.Query("SELECT repo.id FROM `repository` repo WHERE repo.num_forks!=(SELECT COUNT(*) FROM `repository` WHERE fork_id=repo.id)")
  1669. if err != nil {
  1670. log.Error("Select repository count 'num_forks': %v", err)
  1671. } else {
  1672. for _, result := range results {
  1673. id, _ := strconv.ParseInt(string(result["id"]), 10, 64)
  1674. select {
  1675. case <-ctx.Done():
  1676. log.Warn("CheckRepoStats: Cancelled")
  1677. return ErrCancelledf("during %s for repo ID %d", desc, id)
  1678. default:
  1679. }
  1680. log.Trace("Updating repository count 'num_forks': %d", id)
  1681. repo, err := GetRepositoryByID(id)
  1682. if err != nil {
  1683. log.Error("GetRepositoryByID[%d]: %v", id, err)
  1684. continue
  1685. }
  1686. rawResult, err := x.Query("SELECT COUNT(*) FROM `repository` WHERE fork_id=?", repo.ID)
  1687. if err != nil {
  1688. log.Error("Select count of forks[%d]: %v", repo.ID, err)
  1689. continue
  1690. }
  1691. repo.NumForks = int(parseCountResult(rawResult))
  1692. if err = UpdateRepository(repo, false); err != nil {
  1693. log.Error("UpdateRepository[%d]: %v", id, err)
  1694. continue
  1695. }
  1696. }
  1697. }
  1698. // ***** END: Repository.NumForks *****
  1699. return nil
  1700. }
  1701. // SetArchiveRepoState sets if a repo is archived
  1702. func (repo *Repository) SetArchiveRepoState(isArchived bool) (err error) {
  1703. repo.IsArchived = isArchived
  1704. _, err = x.Where("id = ?", repo.ID).Cols("is_archived").NoAutoTime().Update(repo)
  1705. return
  1706. }
  1707. // ___________ __
  1708. // \_ _____/__________| | __
  1709. // | __)/ _ \_ __ \ |/ /
  1710. // | \( <_> ) | \/ <
  1711. // \___ / \____/|__| |__|_ \
  1712. // \/ \/
  1713. // HasForkedRepo checks if given user has already forked a repository with given ID.
  1714. func HasForkedRepo(ownerID, repoID int64) (*Repository, bool) {
  1715. repo := new(Repository)
  1716. has, _ := x.
  1717. Where("owner_id=? AND fork_id=?", ownerID, repoID).
  1718. Get(repo)
  1719. return repo, has
  1720. }
  1721. // CopyLFS copies LFS data from one repo to another
  1722. func CopyLFS(ctx DBContext, newRepo, oldRepo *Repository) error {
  1723. var lfsObjects []*LFSMetaObject
  1724. if err := ctx.e.Where("repository_id=?", oldRepo.ID).Find(&lfsObjects); err != nil {
  1725. return err
  1726. }
  1727. for _, v := range lfsObjects {
  1728. v.ID = 0
  1729. v.RepositoryID = newRepo.ID
  1730. if _, err := ctx.e.Insert(v); err != nil {
  1731. return err
  1732. }
  1733. }
  1734. return nil
  1735. }
  1736. // GetForks returns all the forks of the repository
  1737. func (repo *Repository) GetForks(listOptions ListOptions) ([]*Repository, error) {
  1738. if listOptions.Page == 0 {
  1739. forks := make([]*Repository, 0, repo.NumForks)
  1740. return forks, x.Find(&forks, &Repository{ForkID: repo.ID})
  1741. }
  1742. sess := listOptions.getPaginatedSession()
  1743. forks := make([]*Repository, 0, listOptions.PageSize)
  1744. return forks, sess.Find(&forks, &Repository{ForkID: repo.ID})
  1745. }
  1746. // GetUserFork return user forked repository from this repository, if not forked return nil
  1747. func (repo *Repository) GetUserFork(userID int64) (*Repository, error) {
  1748. var forkedRepo Repository
  1749. has, err := x.Where("fork_id = ?", repo.ID).And("owner_id = ?", userID).Get(&forkedRepo)
  1750. if err != nil {
  1751. return nil, err
  1752. }
  1753. if !has {
  1754. return nil, nil
  1755. }
  1756. return &forkedRepo, nil
  1757. }
  1758. // GetOriginalURLHostname returns the hostname of a URL or the URL
  1759. func (repo *Repository) GetOriginalURLHostname() string {
  1760. u, err := url.Parse(repo.OriginalURL)
  1761. if err != nil {
  1762. return repo.OriginalURL
  1763. }
  1764. return u.Host
  1765. }
  1766. // GetTreePathLock returns LSF lock for the treePath
  1767. func (repo *Repository) GetTreePathLock(treePath string) (*LFSLock, error) {
  1768. if setting.LFS.StartServer {
  1769. locks, err := GetLFSLockByRepoID(repo.ID, 0, 0)
  1770. if err != nil {
  1771. return nil, err
  1772. }
  1773. for _, lock := range locks {
  1774. if lock.Path == treePath {
  1775. return lock, nil
  1776. }
  1777. }
  1778. }
  1779. return nil, nil
  1780. }
  1781. func updateRepositoryCols(e Engine, repo *Repository, cols ...string) error {
  1782. _, err := e.ID(repo.ID).Cols(cols...).Update(repo)
  1783. return err
  1784. }
  1785. // UpdateRepositoryCols updates repository's columns
  1786. func UpdateRepositoryCols(repo *Repository, cols ...string) error {
  1787. return updateRepositoryCols(x, repo, cols...)
  1788. }
  1789. // GetTrustModel will get the TrustModel for the repo or the default trust model
  1790. func (repo *Repository) GetTrustModel() TrustModelType {
  1791. trustModel := repo.TrustModel
  1792. if trustModel == DefaultTrustModel {
  1793. trustModel = ToTrustModel(setting.Repository.Signing.DefaultTrustModel)
  1794. if trustModel == DefaultTrustModel {
  1795. return CollaboratorTrustModel
  1796. }
  1797. }
  1798. return trustModel
  1799. }
  1800. // DoctorUserStarNum recalculate Stars number for all user
  1801. func DoctorUserStarNum() (err error) {
  1802. const batchSize = 100
  1803. sess := x.NewSession()
  1804. defer sess.Close()
  1805. for start := 0; ; start += batchSize {
  1806. users := make([]User, 0, batchSize)
  1807. if err = sess.Limit(batchSize, start).Where("type = ?", 0).Cols("id").Find(&users); err != nil {
  1808. return
  1809. }
  1810. if len(users) == 0 {
  1811. break
  1812. }
  1813. if err = sess.Begin(); err != nil {
  1814. return
  1815. }
  1816. for _, user := range users {
  1817. if _, err = sess.Exec("UPDATE `user` SET num_stars=(SELECT COUNT(*) FROM `star` WHERE uid=?) WHERE id=?", user.ID, user.ID); err != nil {
  1818. return
  1819. }
  1820. }
  1821. if err = sess.Commit(); err != nil {
  1822. return
  1823. }
  1824. }
  1825. log.Debug("recalculate Stars number for all user finished")
  1826. return
  1827. }
  1828. // IterateRepository iterate repositories
  1829. func IterateRepository(f func(repo *Repository) error) error {
  1830. var start int
  1831. batchSize := setting.Database.IterateBufferSize
  1832. for {
  1833. repos := make([]*Repository, 0, batchSize)
  1834. if err := x.Limit(batchSize, start).Find(&repos); err != nil {
  1835. return err
  1836. }
  1837. if len(repos) == 0 {
  1838. return nil
  1839. }
  1840. start += len(repos)
  1841. for _, repo := range repos {
  1842. if err := f(repo); err != nil {
  1843. return err
  1844. }
  1845. }
  1846. }
  1847. }