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 73KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440
  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. "crypto/md5"
  9. "errors"
  10. "fmt"
  11. "html/template"
  12. // Needed for jpeg support
  13. _ "image/jpeg"
  14. "image/png"
  15. "io/ioutil"
  16. "net/url"
  17. "os"
  18. "path"
  19. "path/filepath"
  20. "sort"
  21. "strconv"
  22. "strings"
  23. "time"
  24. "code.gitea.io/gitea/modules/avatar"
  25. "code.gitea.io/gitea/modules/git"
  26. "code.gitea.io/gitea/modules/log"
  27. "code.gitea.io/gitea/modules/markup"
  28. "code.gitea.io/gitea/modules/options"
  29. "code.gitea.io/gitea/modules/setting"
  30. "code.gitea.io/gitea/modules/structs"
  31. api "code.gitea.io/gitea/modules/structs"
  32. "code.gitea.io/gitea/modules/timeutil"
  33. "code.gitea.io/gitea/modules/util"
  34. "github.com/unknwon/com"
  35. "xorm.io/builder"
  36. )
  37. var (
  38. // ErrMirrorNotExist mirror does not exist error
  39. ErrMirrorNotExist = errors.New("Mirror does not exist")
  40. // ErrNameEmpty name is empty error
  41. ErrNameEmpty = errors.New("Name is empty")
  42. )
  43. var (
  44. // Gitignores contains the gitiginore files
  45. Gitignores []string
  46. // Licenses contains the license files
  47. Licenses []string
  48. // Readmes contains the readme files
  49. Readmes []string
  50. // LabelTemplates contains the label template files and the list of labels for each file
  51. LabelTemplates map[string]string
  52. // ItemsPerPage maximum items per page in forks, watchers and stars of a repo
  53. ItemsPerPage = 40
  54. )
  55. // loadRepoConfig loads the repository config
  56. func loadRepoConfig() {
  57. // Load .gitignore and license files and readme templates.
  58. types := []string{"gitignore", "license", "readme", "label"}
  59. typeFiles := make([][]string, 4)
  60. for i, t := range types {
  61. files, err := options.Dir(t)
  62. if err != nil {
  63. log.Fatal("Failed to get %s files: %v", t, err)
  64. }
  65. customPath := path.Join(setting.CustomPath, "options", t)
  66. if com.IsDir(customPath) {
  67. customFiles, err := com.StatDir(customPath)
  68. if err != nil {
  69. log.Fatal("Failed to get custom %s files: %v", t, err)
  70. }
  71. for _, f := range customFiles {
  72. if !com.IsSliceContainsStr(files, f) {
  73. files = append(files, f)
  74. }
  75. }
  76. }
  77. typeFiles[i] = files
  78. }
  79. Gitignores = typeFiles[0]
  80. Licenses = typeFiles[1]
  81. Readmes = typeFiles[2]
  82. LabelTemplatesFiles := typeFiles[3]
  83. sort.Strings(Gitignores)
  84. sort.Strings(Licenses)
  85. sort.Strings(Readmes)
  86. sort.Strings(LabelTemplatesFiles)
  87. // Load label templates
  88. LabelTemplates = make(map[string]string)
  89. for _, templateFile := range LabelTemplatesFiles {
  90. labels, err := LoadLabelsFormatted(templateFile)
  91. if err != nil {
  92. log.Error("Failed to load labels: %v", err)
  93. }
  94. LabelTemplates[templateFile] = labels
  95. }
  96. // Filter out invalid names and promote preferred licenses.
  97. sortedLicenses := make([]string, 0, len(Licenses))
  98. for _, name := range setting.Repository.PreferredLicenses {
  99. if com.IsSliceContainsStr(Licenses, name) {
  100. sortedLicenses = append(sortedLicenses, name)
  101. }
  102. }
  103. for _, name := range Licenses {
  104. if !com.IsSliceContainsStr(setting.Repository.PreferredLicenses, name) {
  105. sortedLicenses = append(sortedLicenses, name)
  106. }
  107. }
  108. Licenses = sortedLicenses
  109. }
  110. // NewRepoContext creates a new repository context
  111. func NewRepoContext() {
  112. loadRepoConfig()
  113. loadUnitConfig()
  114. RemoveAllWithNotice("Clean up repository temporary data", filepath.Join(setting.AppDataPath, "tmp"))
  115. }
  116. // RepositoryStatus defines the status of repository
  117. type RepositoryStatus int
  118. // all kinds of RepositoryStatus
  119. const (
  120. RepositoryReady RepositoryStatus = iota // a normal repository
  121. RepositoryBeingMigrated // repository is migrating
  122. )
  123. // Repository represents a git repository.
  124. type Repository struct {
  125. ID int64 `xorm:"pk autoincr"`
  126. OwnerID int64 `xorm:"UNIQUE(s) index"`
  127. OwnerName string
  128. Owner *User `xorm:"-"`
  129. LowerName string `xorm:"UNIQUE(s) INDEX NOT NULL"`
  130. Name string `xorm:"INDEX NOT NULL"`
  131. Description string `xorm:"TEXT"`
  132. Website string `xorm:"VARCHAR(2048)"`
  133. OriginalServiceType structs.GitServiceType `xorm:"index"`
  134. OriginalURL string `xorm:"VARCHAR(2048)"`
  135. DefaultBranch string
  136. NumWatches int
  137. NumStars int
  138. NumForks int
  139. NumIssues int
  140. NumClosedIssues int
  141. NumOpenIssues int `xorm:"-"`
  142. NumPulls int
  143. NumClosedPulls int
  144. NumOpenPulls int `xorm:"-"`
  145. NumMilestones int `xorm:"NOT NULL DEFAULT 0"`
  146. NumClosedMilestones int `xorm:"NOT NULL DEFAULT 0"`
  147. NumOpenMilestones int `xorm:"-"`
  148. NumReleases int `xorm:"-"`
  149. IsPrivate bool `xorm:"INDEX"`
  150. IsEmpty bool `xorm:"INDEX"`
  151. IsArchived bool `xorm:"INDEX"`
  152. IsMirror bool `xorm:"INDEX"`
  153. *Mirror `xorm:"-"`
  154. Status RepositoryStatus `xorm:"NOT NULL DEFAULT 0"`
  155. RenderingMetas map[string]string `xorm:"-"`
  156. Units []*RepoUnit `xorm:"-"`
  157. IsFork bool `xorm:"INDEX NOT NULL DEFAULT false"`
  158. ForkID int64 `xorm:"INDEX"`
  159. BaseRepo *Repository `xorm:"-"`
  160. IsTemplate bool `xorm:"INDEX NOT NULL DEFAULT false"`
  161. TemplateID int64 `xorm:"INDEX"`
  162. TemplateRepo *Repository `xorm:"-"`
  163. Size int64 `xorm:"NOT NULL DEFAULT 0"`
  164. IndexerStatus *RepoIndexerStatus `xorm:"-"`
  165. IsFsckEnabled bool `xorm:"NOT NULL DEFAULT true"`
  166. CloseIssuesViaCommitInAnyBranch bool `xorm:"NOT NULL DEFAULT false"`
  167. Topics []string `xorm:"TEXT JSON"`
  168. // Avatar: ID(10-20)-md5(32) - must fit into 64 symbols
  169. Avatar string `xorm:"VARCHAR(64)"`
  170. CreatedUnix timeutil.TimeStamp `xorm:"INDEX created"`
  171. UpdatedUnix timeutil.TimeStamp `xorm:"INDEX updated"`
  172. }
  173. // ColorFormat returns a colored string to represent this repo
  174. func (repo *Repository) ColorFormat(s fmt.State) {
  175. var ownerName interface{}
  176. if repo.OwnerName != "" {
  177. ownerName = repo.OwnerName
  178. } else if repo.Owner != nil {
  179. ownerName = repo.Owner.Name
  180. } else {
  181. ownerName = log.NewColoredIDValue(strconv.FormatInt(repo.OwnerID, 10))
  182. }
  183. log.ColorFprintf(s, "%d:%s/%s",
  184. log.NewColoredIDValue(repo.ID),
  185. ownerName,
  186. repo.Name)
  187. }
  188. // IsBeingMigrated indicates that repository is being migtated
  189. func (repo *Repository) IsBeingMigrated() bool {
  190. return repo.Status == RepositoryBeingMigrated
  191. }
  192. // IsBeingCreated indicates that repository is being migrated or forked
  193. func (repo *Repository) IsBeingCreated() bool {
  194. return repo.IsBeingMigrated()
  195. }
  196. // AfterLoad is invoked from XORM after setting the values of all fields of this object.
  197. func (repo *Repository) AfterLoad() {
  198. // FIXME: use models migration to solve all at once.
  199. if len(repo.DefaultBranch) == 0 {
  200. repo.DefaultBranch = "master"
  201. }
  202. repo.NumOpenIssues = repo.NumIssues - repo.NumClosedIssues
  203. repo.NumOpenPulls = repo.NumPulls - repo.NumClosedPulls
  204. repo.NumOpenMilestones = repo.NumMilestones - repo.NumClosedMilestones
  205. }
  206. // MustOwner always returns a valid *User object to avoid
  207. // conceptually impossible error handling.
  208. // It creates a fake object that contains error details
  209. // when error occurs.
  210. func (repo *Repository) MustOwner() *User {
  211. return repo.mustOwner(x)
  212. }
  213. // FullName returns the repository full name
  214. func (repo *Repository) FullName() string {
  215. return repo.OwnerName + "/" + repo.Name
  216. }
  217. // HTMLURL returns the repository HTML URL
  218. func (repo *Repository) HTMLURL() string {
  219. return setting.AppURL + repo.FullName()
  220. }
  221. // APIURL returns the repository API URL
  222. func (repo *Repository) APIURL() string {
  223. return setting.AppURL + path.Join("api/v1/repos", repo.FullName())
  224. }
  225. // APIFormat converts a Repository to api.Repository
  226. func (repo *Repository) APIFormat(mode AccessMode) *api.Repository {
  227. return repo.innerAPIFormat(x, mode, false)
  228. }
  229. // GetCommitsCountCacheKey returns cache key used for commits count caching.
  230. func (repo *Repository) GetCommitsCountCacheKey(contextName string, isRef bool) string {
  231. var prefix string
  232. if isRef {
  233. prefix = "ref"
  234. } else {
  235. prefix = "commit"
  236. }
  237. return fmt.Sprintf("commits-count-%d-%s-%s", repo.ID, prefix, contextName)
  238. }
  239. func (repo *Repository) innerAPIFormat(e Engine, mode AccessMode, isParent bool) *api.Repository {
  240. var parent *api.Repository
  241. cloneLink := repo.cloneLink(false)
  242. permission := &api.Permission{
  243. Admin: mode >= AccessModeAdmin,
  244. Push: mode >= AccessModeWrite,
  245. Pull: mode >= AccessModeRead,
  246. }
  247. if !isParent {
  248. err := repo.getBaseRepo(e)
  249. if err != nil {
  250. log.Error("APIFormat: %v", err)
  251. }
  252. if repo.BaseRepo != nil {
  253. parent = repo.BaseRepo.innerAPIFormat(e, mode, true)
  254. }
  255. }
  256. hasIssues := false
  257. var externalTracker *api.ExternalTracker
  258. var internalTracker *api.InternalTracker
  259. if unit, err := repo.getUnit(e, UnitTypeIssues); err == nil {
  260. config := unit.IssuesConfig()
  261. hasIssues = true
  262. internalTracker = &api.InternalTracker{
  263. EnableTimeTracker: config.EnableTimetracker,
  264. AllowOnlyContributorsToTrackTime: config.AllowOnlyContributorsToTrackTime,
  265. EnableIssueDependencies: config.EnableDependencies,
  266. }
  267. } else if unit, err := repo.getUnit(e, UnitTypeExternalTracker); err == nil {
  268. config := unit.ExternalTrackerConfig()
  269. hasIssues = true
  270. externalTracker = &api.ExternalTracker{
  271. ExternalTrackerURL: config.ExternalTrackerURL,
  272. ExternalTrackerFormat: config.ExternalTrackerFormat,
  273. ExternalTrackerStyle: config.ExternalTrackerStyle,
  274. }
  275. }
  276. hasWiki := false
  277. var externalWiki *api.ExternalWiki
  278. if _, err := repo.getUnit(e, UnitTypeWiki); err == nil {
  279. hasWiki = true
  280. } else if unit, err := repo.getUnit(e, UnitTypeExternalWiki); err == nil {
  281. hasWiki = true
  282. config := unit.ExternalWikiConfig()
  283. externalWiki = &api.ExternalWiki{
  284. ExternalWikiURL: config.ExternalWikiURL,
  285. }
  286. }
  287. hasPullRequests := false
  288. ignoreWhitespaceConflicts := false
  289. allowMerge := false
  290. allowRebase := false
  291. allowRebaseMerge := false
  292. allowSquash := false
  293. if unit, err := repo.getUnit(e, UnitTypePullRequests); err == nil {
  294. config := unit.PullRequestsConfig()
  295. hasPullRequests = true
  296. ignoreWhitespaceConflicts = config.IgnoreWhitespaceConflicts
  297. allowMerge = config.AllowMerge
  298. allowRebase = config.AllowRebase
  299. allowRebaseMerge = config.AllowRebaseMerge
  300. allowSquash = config.AllowSquash
  301. }
  302. repo.mustOwner(e)
  303. return &api.Repository{
  304. ID: repo.ID,
  305. Owner: repo.Owner.APIFormat(),
  306. Name: repo.Name,
  307. FullName: repo.FullName(),
  308. Description: repo.Description,
  309. Private: repo.IsPrivate,
  310. Template: repo.IsTemplate,
  311. Empty: repo.IsEmpty,
  312. Archived: repo.IsArchived,
  313. Size: int(repo.Size / 1024),
  314. Fork: repo.IsFork,
  315. Parent: parent,
  316. Mirror: repo.IsMirror,
  317. HTMLURL: repo.HTMLURL(),
  318. SSHURL: cloneLink.SSH,
  319. CloneURL: cloneLink.HTTPS,
  320. Website: repo.Website,
  321. Stars: repo.NumStars,
  322. Forks: repo.NumForks,
  323. Watchers: repo.NumWatches,
  324. OpenIssues: repo.NumOpenIssues,
  325. OpenPulls: repo.NumOpenPulls,
  326. Releases: repo.NumReleases,
  327. DefaultBranch: repo.DefaultBranch,
  328. Created: repo.CreatedUnix.AsTime(),
  329. Updated: repo.UpdatedUnix.AsTime(),
  330. Permissions: permission,
  331. HasIssues: hasIssues,
  332. ExternalTracker: externalTracker,
  333. InternalTracker: internalTracker,
  334. HasWiki: hasWiki,
  335. ExternalWiki: externalWiki,
  336. HasPullRequests: hasPullRequests,
  337. IgnoreWhitespaceConflicts: ignoreWhitespaceConflicts,
  338. AllowMerge: allowMerge,
  339. AllowRebase: allowRebase,
  340. AllowRebaseMerge: allowRebaseMerge,
  341. AllowSquash: allowSquash,
  342. AvatarURL: repo.avatarLink(e),
  343. }
  344. }
  345. func (repo *Repository) getUnits(e Engine) (err error) {
  346. if repo.Units != nil {
  347. return nil
  348. }
  349. repo.Units, err = getUnitsByRepoID(e, repo.ID)
  350. log.Trace("repo.Units: %-+v", repo.Units)
  351. return err
  352. }
  353. // CheckUnitUser check whether user could visit the unit of this repository
  354. func (repo *Repository) CheckUnitUser(userID int64, isAdmin bool, unitType UnitType) bool {
  355. return repo.checkUnitUser(x, userID, isAdmin, unitType)
  356. }
  357. func (repo *Repository) checkUnitUser(e Engine, userID int64, isAdmin bool, unitType UnitType) bool {
  358. if isAdmin {
  359. return true
  360. }
  361. user, err := getUserByID(e, userID)
  362. if err != nil {
  363. return false
  364. }
  365. perm, err := getUserRepoPermission(e, repo, user)
  366. if err != nil {
  367. return false
  368. }
  369. return perm.CanRead(unitType)
  370. }
  371. // UnitEnabled if this repository has the given unit enabled
  372. func (repo *Repository) UnitEnabled(tp UnitType) bool {
  373. if err := repo.getUnits(x); err != nil {
  374. log.Warn("Error loading repository (ID: %d) units: %s", repo.ID, err.Error())
  375. }
  376. for _, unit := range repo.Units {
  377. if unit.Type == tp {
  378. return true
  379. }
  380. }
  381. return false
  382. }
  383. // ErrUnitTypeNotExist represents a "UnitTypeNotExist" kind of error.
  384. type ErrUnitTypeNotExist struct {
  385. UT UnitType
  386. }
  387. // IsErrUnitTypeNotExist checks if an error is a ErrUnitNotExist.
  388. func IsErrUnitTypeNotExist(err error) bool {
  389. _, ok := err.(ErrUnitTypeNotExist)
  390. return ok
  391. }
  392. func (err ErrUnitTypeNotExist) Error() string {
  393. return fmt.Sprintf("Unit type does not exist: %s", err.UT.String())
  394. }
  395. // MustGetUnit always returns a RepoUnit object
  396. func (repo *Repository) MustGetUnit(tp UnitType) *RepoUnit {
  397. ru, err := repo.GetUnit(tp)
  398. if err == nil {
  399. return ru
  400. }
  401. if tp == UnitTypeExternalWiki {
  402. return &RepoUnit{
  403. Type: tp,
  404. Config: new(ExternalWikiConfig),
  405. }
  406. } else if tp == UnitTypeExternalTracker {
  407. return &RepoUnit{
  408. Type: tp,
  409. Config: new(ExternalTrackerConfig),
  410. }
  411. } else if tp == UnitTypePullRequests {
  412. return &RepoUnit{
  413. Type: tp,
  414. Config: new(PullRequestsConfig),
  415. }
  416. } else if tp == UnitTypeIssues {
  417. return &RepoUnit{
  418. Type: tp,
  419. Config: new(IssuesConfig),
  420. }
  421. }
  422. return &RepoUnit{
  423. Type: tp,
  424. Config: new(UnitConfig),
  425. }
  426. }
  427. // GetUnit returns a RepoUnit object
  428. func (repo *Repository) GetUnit(tp UnitType) (*RepoUnit, error) {
  429. return repo.getUnit(x, tp)
  430. }
  431. func (repo *Repository) getUnit(e Engine, tp UnitType) (*RepoUnit, error) {
  432. if err := repo.getUnits(e); err != nil {
  433. return nil, err
  434. }
  435. for _, unit := range repo.Units {
  436. if unit.Type == tp {
  437. return unit, nil
  438. }
  439. }
  440. return nil, ErrUnitTypeNotExist{tp}
  441. }
  442. func (repo *Repository) getOwner(e Engine) (err error) {
  443. if repo.Owner != nil {
  444. return nil
  445. }
  446. repo.Owner, err = getUserByID(e, repo.OwnerID)
  447. return err
  448. }
  449. // GetOwner returns the repository owner
  450. func (repo *Repository) GetOwner() error {
  451. return repo.getOwner(x)
  452. }
  453. func (repo *Repository) mustOwner(e Engine) *User {
  454. if err := repo.getOwner(e); err != nil {
  455. return &User{
  456. Name: "error",
  457. FullName: err.Error(),
  458. }
  459. }
  460. return repo.Owner
  461. }
  462. // ComposeMetas composes a map of metas for properly rendering issue links and external issue trackers.
  463. func (repo *Repository) ComposeMetas() map[string]string {
  464. if repo.RenderingMetas == nil {
  465. metas := map[string]string{
  466. "user": repo.OwnerName,
  467. "repo": repo.Name,
  468. "repoPath": repo.RepoPath(),
  469. }
  470. unit, err := repo.GetUnit(UnitTypeExternalTracker)
  471. if err == nil {
  472. metas["format"] = unit.ExternalTrackerConfig().ExternalTrackerFormat
  473. switch unit.ExternalTrackerConfig().ExternalTrackerStyle {
  474. case markup.IssueNameStyleAlphanumeric:
  475. metas["style"] = markup.IssueNameStyleAlphanumeric
  476. default:
  477. metas["style"] = markup.IssueNameStyleNumeric
  478. }
  479. }
  480. repo.MustOwner()
  481. if repo.Owner.IsOrganization() {
  482. teams := make([]string, 0, 5)
  483. _ = x.Table("team_repo").
  484. Join("INNER", "team", "team.id = team_repo.team_id").
  485. Where("team_repo.repo_id = ?", repo.ID).
  486. Select("team.lower_name").
  487. OrderBy("team.lower_name").
  488. Find(&teams)
  489. metas["teams"] = "," + strings.Join(teams, ",") + ","
  490. metas["org"] = strings.ToLower(repo.OwnerName)
  491. }
  492. repo.RenderingMetas = metas
  493. }
  494. return repo.RenderingMetas
  495. }
  496. // DeleteWiki removes the actual and local copy of repository wiki.
  497. func (repo *Repository) DeleteWiki() error {
  498. return repo.deleteWiki(x)
  499. }
  500. func (repo *Repository) deleteWiki(e Engine) error {
  501. wikiPaths := []string{repo.WikiPath()}
  502. for _, wikiPath := range wikiPaths {
  503. removeAllWithNotice(e, "Delete repository wiki", wikiPath)
  504. }
  505. _, err := e.Where("repo_id = ?", repo.ID).And("type = ?", UnitTypeWiki).Delete(new(RepoUnit))
  506. return err
  507. }
  508. func (repo *Repository) getAssignees(e Engine) (_ []*User, err error) {
  509. if err = repo.getOwner(e); err != nil {
  510. return nil, err
  511. }
  512. accesses := make([]*Access, 0, 10)
  513. if err = e.
  514. Where("repo_id = ? AND mode >= ?", repo.ID, AccessModeWrite).
  515. Find(&accesses); err != nil {
  516. return nil, err
  517. }
  518. // Leave a seat for owner itself to append later, but if owner is an organization
  519. // and just waste 1 unit is cheaper than re-allocate memory once.
  520. users := make([]*User, 0, len(accesses)+1)
  521. if len(accesses) > 0 {
  522. userIDs := make([]int64, len(accesses))
  523. for i := 0; i < len(accesses); i++ {
  524. userIDs[i] = accesses[i].UserID
  525. }
  526. if err = e.In("id", userIDs).Find(&users); err != nil {
  527. return nil, err
  528. }
  529. }
  530. if !repo.Owner.IsOrganization() {
  531. users = append(users, repo.Owner)
  532. }
  533. return users, nil
  534. }
  535. // GetAssignees returns all users that have write access and can be assigned to issues
  536. // of the repository,
  537. func (repo *Repository) GetAssignees() (_ []*User, err error) {
  538. return repo.getAssignees(x)
  539. }
  540. // GetMilestoneByID returns the milestone belongs to repository by given ID.
  541. func (repo *Repository) GetMilestoneByID(milestoneID int64) (*Milestone, error) {
  542. return GetMilestoneByRepoID(repo.ID, milestoneID)
  543. }
  544. // IssueStats returns number of open and closed repository issues by given filter mode.
  545. func (repo *Repository) IssueStats(uid int64, filterMode int, isPull bool) (int64, int64) {
  546. return GetRepoIssueStats(repo.ID, uid, filterMode, isPull)
  547. }
  548. // GetMirror sets the repository mirror, returns an error upon failure
  549. func (repo *Repository) GetMirror() (err error) {
  550. repo.Mirror, err = GetMirrorByRepoID(repo.ID)
  551. return err
  552. }
  553. // GetBaseRepo populates repo.BaseRepo for a fork repository and
  554. // returns an error on failure (NOTE: no error is returned for
  555. // non-fork repositories, and BaseRepo will be left untouched)
  556. func (repo *Repository) GetBaseRepo() (err error) {
  557. return repo.getBaseRepo(x)
  558. }
  559. func (repo *Repository) getBaseRepo(e Engine) (err error) {
  560. if !repo.IsFork {
  561. return nil
  562. }
  563. repo.BaseRepo, err = getRepositoryByID(e, repo.ForkID)
  564. return err
  565. }
  566. // IsGenerated returns whether _this_ repository was generated from a template
  567. func (repo *Repository) IsGenerated() bool {
  568. return repo.TemplateID != 0
  569. }
  570. // GetTemplateRepo populates repo.TemplateRepo for a generated repository and
  571. // returns an error on failure (NOTE: no error is returned for
  572. // non-generated repositories, and TemplateRepo will be left untouched)
  573. func (repo *Repository) GetTemplateRepo() (err error) {
  574. return repo.getTemplateRepo(x)
  575. }
  576. func (repo *Repository) getTemplateRepo(e Engine) (err error) {
  577. if !repo.IsGenerated() {
  578. return nil
  579. }
  580. repo.TemplateRepo, err = getRepositoryByID(e, repo.TemplateID)
  581. return err
  582. }
  583. // RepoPath returns the repository path
  584. func (repo *Repository) RepoPath() string {
  585. return RepoPath(repo.OwnerName, repo.Name)
  586. }
  587. // GitConfigPath returns the path to a repository's git config/ directory
  588. func GitConfigPath(repoPath string) string {
  589. return filepath.Join(repoPath, "config")
  590. }
  591. // GitConfigPath returns the repository git config path
  592. func (repo *Repository) GitConfigPath() string {
  593. return GitConfigPath(repo.RepoPath())
  594. }
  595. // RelLink returns the repository relative link
  596. func (repo *Repository) RelLink() string {
  597. return "/" + repo.FullName()
  598. }
  599. // Link returns the repository link
  600. func (repo *Repository) Link() string {
  601. return setting.AppSubURL + "/" + repo.FullName()
  602. }
  603. // ComposeCompareURL returns the repository comparison URL
  604. func (repo *Repository) ComposeCompareURL(oldCommitID, newCommitID string) string {
  605. return fmt.Sprintf("%s/compare/%s...%s", repo.FullName(), oldCommitID, newCommitID)
  606. }
  607. // UpdateDefaultBranch updates the default branch
  608. func (repo *Repository) UpdateDefaultBranch() error {
  609. _, err := x.ID(repo.ID).Cols("default_branch").Update(repo)
  610. return err
  611. }
  612. // IsOwnedBy returns true when user owns this repository
  613. func (repo *Repository) IsOwnedBy(userID int64) bool {
  614. return repo.OwnerID == userID
  615. }
  616. func (repo *Repository) updateSize(e Engine) error {
  617. size, err := util.GetDirectorySize(repo.RepoPath())
  618. if err != nil {
  619. return fmt.Errorf("updateSize: %v", err)
  620. }
  621. repo.Size = size
  622. _, err = e.ID(repo.ID).Cols("size").Update(repo)
  623. return err
  624. }
  625. // UpdateSize updates the repository size, calculating it using util.GetDirectorySize
  626. func (repo *Repository) UpdateSize(ctx DBContext) error {
  627. return repo.updateSize(ctx.e)
  628. }
  629. // CanUserFork returns true if specified user can fork repository.
  630. func (repo *Repository) CanUserFork(user *User) (bool, error) {
  631. if user == nil {
  632. return false, nil
  633. }
  634. if repo.OwnerID != user.ID && !user.HasForkedRepo(repo.ID) {
  635. return true, nil
  636. }
  637. if err := user.GetOwnedOrganizations(); err != nil {
  638. return false, err
  639. }
  640. for _, org := range user.OwnedOrgs {
  641. if repo.OwnerID != org.ID && !org.HasForkedRepo(repo.ID) {
  642. return true, nil
  643. }
  644. }
  645. return false, nil
  646. }
  647. // CanUserDelete returns true if user could delete the repository
  648. func (repo *Repository) CanUserDelete(user *User) (bool, error) {
  649. if user.IsAdmin || user.ID == repo.OwnerID {
  650. return true, nil
  651. }
  652. if err := repo.GetOwner(); err != nil {
  653. return false, err
  654. }
  655. if repo.Owner.IsOrganization() {
  656. isOwner, err := repo.Owner.IsOwnedBy(user.ID)
  657. if err != nil {
  658. return false, err
  659. } else if isOwner {
  660. return true, nil
  661. }
  662. }
  663. return false, nil
  664. }
  665. // CanEnablePulls returns true if repository meets the requirements of accepting pulls.
  666. func (repo *Repository) CanEnablePulls() bool {
  667. return !repo.IsMirror && !repo.IsEmpty
  668. }
  669. // AllowsPulls returns true if repository meets the requirements of accepting pulls and has them enabled.
  670. func (repo *Repository) AllowsPulls() bool {
  671. return repo.CanEnablePulls() && repo.UnitEnabled(UnitTypePullRequests)
  672. }
  673. // CanEnableEditor returns true if repository meets the requirements of web editor.
  674. func (repo *Repository) CanEnableEditor() bool {
  675. return !repo.IsMirror
  676. }
  677. // GetReaders returns all users that have explicit read access or higher to the repository.
  678. func (repo *Repository) GetReaders() (_ []*User, err error) {
  679. return repo.getUsersWithAccessMode(x, AccessModeRead)
  680. }
  681. // GetWriters returns all users that have write access to the repository.
  682. func (repo *Repository) GetWriters() (_ []*User, err error) {
  683. return repo.getUsersWithAccessMode(x, AccessModeWrite)
  684. }
  685. // IsReader returns true if user has explicit read access or higher to the repository.
  686. func (repo *Repository) IsReader(userID int64) (bool, error) {
  687. if repo.OwnerID == userID {
  688. return true, nil
  689. }
  690. return x.Where("repo_id = ? AND user_id = ? AND mode >= ?", repo.ID, userID, AccessModeRead).Get(&Access{})
  691. }
  692. // getUsersWithAccessMode returns users that have at least given access mode to the repository.
  693. func (repo *Repository) getUsersWithAccessMode(e Engine, mode AccessMode) (_ []*User, err error) {
  694. if err = repo.getOwner(e); err != nil {
  695. return nil, err
  696. }
  697. accesses := make([]*Access, 0, 10)
  698. if err = e.Where("repo_id = ? AND mode >= ?", repo.ID, mode).Find(&accesses); err != nil {
  699. return nil, err
  700. }
  701. // Leave a seat for owner itself to append later, but if owner is an organization
  702. // and just waste 1 unit is cheaper than re-allocate memory once.
  703. users := make([]*User, 0, len(accesses)+1)
  704. if len(accesses) > 0 {
  705. userIDs := make([]int64, len(accesses))
  706. for i := 0; i < len(accesses); i++ {
  707. userIDs[i] = accesses[i].UserID
  708. }
  709. if err = e.In("id", userIDs).Find(&users); err != nil {
  710. return nil, err
  711. }
  712. }
  713. if !repo.Owner.IsOrganization() {
  714. users = append(users, repo.Owner)
  715. }
  716. return users, nil
  717. }
  718. // DescriptionHTML does special handles to description and return HTML string.
  719. func (repo *Repository) DescriptionHTML() template.HTML {
  720. desc, err := markup.RenderDescriptionHTML([]byte(repo.Description), repo.HTMLURL(), repo.ComposeMetas())
  721. if err != nil {
  722. log.Error("Failed to render description for %s (ID: %d): %v", repo.Name, repo.ID, err)
  723. return template.HTML(markup.Sanitize(repo.Description))
  724. }
  725. return template.HTML(markup.Sanitize(string(desc)))
  726. }
  727. func isRepositoryExist(e Engine, u *User, repoName string) (bool, error) {
  728. has, err := e.Get(&Repository{
  729. OwnerID: u.ID,
  730. LowerName: strings.ToLower(repoName),
  731. })
  732. return has && com.IsDir(RepoPath(u.Name, repoName)), err
  733. }
  734. // IsRepositoryExist returns true if the repository with given name under user has already existed.
  735. func IsRepositoryExist(u *User, repoName string) (bool, error) {
  736. return isRepositoryExist(x, u, repoName)
  737. }
  738. // CloneLink represents different types of clone URLs of repository.
  739. type CloneLink struct {
  740. SSH string
  741. HTTPS string
  742. Git string
  743. }
  744. // ComposeHTTPSCloneURL returns HTTPS clone URL based on given owner and repository name.
  745. func ComposeHTTPSCloneURL(owner, repo string) string {
  746. return fmt.Sprintf("%s%s/%s.git", setting.AppURL, url.PathEscape(owner), url.PathEscape(repo))
  747. }
  748. func (repo *Repository) cloneLink(isWiki bool) *CloneLink {
  749. repoName := repo.Name
  750. if isWiki {
  751. repoName += ".wiki"
  752. }
  753. sshUser := setting.RunUser
  754. if setting.SSH.StartBuiltinServer {
  755. sshUser = setting.SSH.BuiltinServerUser
  756. }
  757. cl := new(CloneLink)
  758. if setting.SSH.Port != 22 {
  759. cl.SSH = fmt.Sprintf("ssh://%s@%s:%d/%s/%s.git", sshUser, setting.SSH.Domain, setting.SSH.Port, repo.OwnerName, repoName)
  760. } else if setting.Repository.UseCompatSSHURI {
  761. cl.SSH = fmt.Sprintf("ssh://%s@%s/%s/%s.git", sshUser, setting.SSH.Domain, repo.OwnerName, repoName)
  762. } else {
  763. cl.SSH = fmt.Sprintf("%s@%s:%s/%s.git", sshUser, setting.SSH.Domain, repo.OwnerName, repoName)
  764. }
  765. cl.HTTPS = ComposeHTTPSCloneURL(repo.OwnerName, repoName)
  766. return cl
  767. }
  768. // CloneLink returns clone URLs of repository.
  769. func (repo *Repository) CloneLink() (cl *CloneLink) {
  770. return repo.cloneLink(false)
  771. }
  772. // CheckCreateRepository check if could created a repository
  773. func CheckCreateRepository(doer, u *User, name string) error {
  774. if !doer.CanCreateRepo() {
  775. return ErrReachLimitOfRepo{u.MaxRepoCreation}
  776. }
  777. if err := IsUsableRepoName(name); err != nil {
  778. return err
  779. }
  780. has, err := isRepositoryExist(x, u, name)
  781. if err != nil {
  782. return fmt.Errorf("IsRepositoryExist: %v", err)
  783. } else if has {
  784. return ErrRepoAlreadyExist{u.Name, name}
  785. }
  786. return nil
  787. }
  788. // CreateDelegateHooks creates all the hooks scripts for the repo
  789. func CreateDelegateHooks(repoPath string) error {
  790. return createDelegateHooks(repoPath)
  791. }
  792. // createDelegateHooks creates all the hooks scripts for the repo
  793. func createDelegateHooks(repoPath string) (err error) {
  794. var (
  795. hookNames = []string{"pre-receive", "update", "post-receive"}
  796. hookTpls = []string{
  797. fmt.Sprintf("#!/usr/bin/env %s\ndata=$(cat)\nexitcodes=\"\"\nhookname=$(basename $0)\nGIT_DIR=${GIT_DIR:-$(dirname $0)}\n\nfor hook in ${GIT_DIR}/hooks/${hookname}.d/*; do\ntest -x \"${hook}\" || continue\necho \"${data}\" | \"${hook}\"\nexitcodes=\"${exitcodes} $?\"\ndone\n\nfor i in ${exitcodes}; do\n[ ${i} -eq 0 ] || exit ${i}\ndone\n", setting.ScriptType),
  798. fmt.Sprintf("#!/usr/bin/env %s\nexitcodes=\"\"\nhookname=$(basename $0)\nGIT_DIR=${GIT_DIR:-$(dirname $0)}\n\nfor hook in ${GIT_DIR}/hooks/${hookname}.d/*; do\ntest -x \"${hook}\" || continue\n\"${hook}\" $1 $2 $3\nexitcodes=\"${exitcodes} $?\"\ndone\n\nfor i in ${exitcodes}; do\n[ ${i} -eq 0 ] || exit ${i}\ndone\n", setting.ScriptType),
  799. fmt.Sprintf("#!/usr/bin/env %s\ndata=$(cat)\nexitcodes=\"\"\nhookname=$(basename $0)\nGIT_DIR=${GIT_DIR:-$(dirname $0)}\n\nfor hook in ${GIT_DIR}/hooks/${hookname}.d/*; do\ntest -x \"${hook}\" || continue\necho \"${data}\" | \"${hook}\"\nexitcodes=\"${exitcodes} $?\"\ndone\n\nfor i in ${exitcodes}; do\n[ ${i} -eq 0 ] || exit ${i}\ndone\n", setting.ScriptType),
  800. }
  801. giteaHookTpls = []string{
  802. fmt.Sprintf("#!/usr/bin/env %s\n\"%s\" hook --config='%s' pre-receive\n", setting.ScriptType, setting.AppPath, setting.CustomConf),
  803. fmt.Sprintf("#!/usr/bin/env %s\n\"%s\" hook --config='%s' update $1 $2 $3\n", setting.ScriptType, setting.AppPath, setting.CustomConf),
  804. fmt.Sprintf("#!/usr/bin/env %s\n\"%s\" hook --config='%s' post-receive\n", setting.ScriptType, setting.AppPath, setting.CustomConf),
  805. }
  806. )
  807. hookDir := filepath.Join(repoPath, "hooks")
  808. for i, hookName := range hookNames {
  809. oldHookPath := filepath.Join(hookDir, hookName)
  810. newHookPath := filepath.Join(hookDir, hookName+".d", "gitea")
  811. if err := os.MkdirAll(filepath.Join(hookDir, hookName+".d"), os.ModePerm); err != nil {
  812. return fmt.Errorf("create hooks dir '%s': %v", filepath.Join(hookDir, hookName+".d"), err)
  813. }
  814. // WARNING: This will override all old server-side hooks
  815. if err = os.Remove(oldHookPath); err != nil && !os.IsNotExist(err) {
  816. return fmt.Errorf("unable to pre-remove old hook file '%s' prior to rewriting: %v ", oldHookPath, err)
  817. }
  818. if err = ioutil.WriteFile(oldHookPath, []byte(hookTpls[i]), 0777); err != nil {
  819. return fmt.Errorf("write old hook file '%s': %v", oldHookPath, err)
  820. }
  821. if err = os.Remove(newHookPath); err != nil && !os.IsNotExist(err) {
  822. return fmt.Errorf("unable to pre-remove new hook file '%s' prior to rewriting: %v", newHookPath, err)
  823. }
  824. if err = ioutil.WriteFile(newHookPath, []byte(giteaHookTpls[i]), 0777); err != nil {
  825. return fmt.Errorf("write new hook file '%s': %v", newHookPath, err)
  826. }
  827. }
  828. return nil
  829. }
  830. // CreateRepoOptions contains the create repository options
  831. type CreateRepoOptions struct {
  832. Name string
  833. Description string
  834. OriginalURL string
  835. GitServiceType structs.GitServiceType
  836. Gitignores string
  837. IssueLabels string
  838. License string
  839. Readme string
  840. IsPrivate bool
  841. IsMirror bool
  842. AutoInit bool
  843. Status RepositoryStatus
  844. }
  845. // GetRepoInitFile returns repository init files
  846. func GetRepoInitFile(tp, name string) ([]byte, error) {
  847. cleanedName := strings.TrimLeft(path.Clean("/"+name), "/")
  848. relPath := path.Join("options", tp, cleanedName)
  849. // Use custom file when available.
  850. customPath := path.Join(setting.CustomPath, relPath)
  851. if com.IsFile(customPath) {
  852. return ioutil.ReadFile(customPath)
  853. }
  854. switch tp {
  855. case "readme":
  856. return options.Readme(cleanedName)
  857. case "gitignore":
  858. return options.Gitignore(cleanedName)
  859. case "license":
  860. return options.License(cleanedName)
  861. case "label":
  862. return options.Labels(cleanedName)
  863. default:
  864. return []byte{}, fmt.Errorf("Invalid init file type")
  865. }
  866. }
  867. var (
  868. reservedRepoNames = []string{".", ".."}
  869. reservedRepoPatterns = []string{"*.git", "*.wiki"}
  870. )
  871. // IsUsableRepoName returns true when repository is usable
  872. func IsUsableRepoName(name string) error {
  873. return isUsableName(reservedRepoNames, reservedRepoPatterns, name)
  874. }
  875. // CreateRepository creates a repository for the user/organization.
  876. func CreateRepository(ctx DBContext, doer, u *User, repo *Repository) (err error) {
  877. if err = IsUsableRepoName(repo.Name); err != nil {
  878. return err
  879. }
  880. has, err := isRepositoryExist(ctx.e, u, repo.Name)
  881. if err != nil {
  882. return fmt.Errorf("IsRepositoryExist: %v", err)
  883. } else if has {
  884. return ErrRepoAlreadyExist{u.Name, repo.Name}
  885. }
  886. if _, err = ctx.e.Insert(repo); err != nil {
  887. return err
  888. }
  889. if err = deleteRepoRedirect(ctx.e, u.ID, repo.Name); err != nil {
  890. return err
  891. }
  892. // insert units for repo
  893. var units = make([]RepoUnit, 0, len(DefaultRepoUnits))
  894. for _, tp := range DefaultRepoUnits {
  895. if tp == UnitTypeIssues {
  896. units = append(units, RepoUnit{
  897. RepoID: repo.ID,
  898. Type: tp,
  899. Config: &IssuesConfig{
  900. EnableTimetracker: setting.Service.DefaultEnableTimetracking,
  901. AllowOnlyContributorsToTrackTime: setting.Service.DefaultAllowOnlyContributorsToTrackTime,
  902. EnableDependencies: setting.Service.DefaultEnableDependencies,
  903. },
  904. })
  905. } else if tp == UnitTypePullRequests {
  906. units = append(units, RepoUnit{
  907. RepoID: repo.ID,
  908. Type: tp,
  909. Config: &PullRequestsConfig{AllowMerge: true, AllowRebase: true, AllowRebaseMerge: true, AllowSquash: true},
  910. })
  911. } else {
  912. units = append(units, RepoUnit{
  913. RepoID: repo.ID,
  914. Type: tp,
  915. })
  916. }
  917. }
  918. if _, err = ctx.e.Insert(&units); err != nil {
  919. return err
  920. }
  921. // Remember visibility preference.
  922. u.LastRepoVisibility = repo.IsPrivate
  923. if err = updateUserCols(ctx.e, u, "last_repo_visibility"); err != nil {
  924. return fmt.Errorf("updateUser: %v", err)
  925. }
  926. if _, err = ctx.e.Incr("num_repos").ID(u.ID).Update(new(User)); err != nil {
  927. return fmt.Errorf("increment user total_repos: %v", err)
  928. }
  929. u.NumRepos++
  930. // Give access to all members in teams with access to all repositories.
  931. if u.IsOrganization() {
  932. if err := u.GetTeams(); err != nil {
  933. return fmt.Errorf("GetTeams: %v", err)
  934. }
  935. for _, t := range u.Teams {
  936. if t.IncludesAllRepositories {
  937. if err := t.addRepository(ctx.e, repo); err != nil {
  938. return fmt.Errorf("addRepository: %v", err)
  939. }
  940. }
  941. }
  942. if isAdmin, err := isUserRepoAdmin(ctx.e, repo, doer); err != nil {
  943. return fmt.Errorf("isUserRepoAdmin: %v", err)
  944. } else if !isAdmin {
  945. // Make creator repo admin if it wan't assigned automatically
  946. if err = repo.addCollaborator(ctx.e, doer); err != nil {
  947. return fmt.Errorf("AddCollaborator: %v", err)
  948. }
  949. if err = repo.changeCollaborationAccessMode(ctx.e, doer.ID, AccessModeAdmin); err != nil {
  950. return fmt.Errorf("ChangeCollaborationAccessMode: %v", err)
  951. }
  952. }
  953. } else if err = repo.recalculateAccesses(ctx.e); err != nil {
  954. // Organization automatically called this in addRepository method.
  955. return fmt.Errorf("recalculateAccesses: %v", err)
  956. }
  957. if setting.Service.AutoWatchNewRepos {
  958. if err = watchRepo(ctx.e, doer.ID, repo.ID, true); err != nil {
  959. return fmt.Errorf("watchRepo: %v", err)
  960. }
  961. }
  962. if err = copyDefaultWebhooksToRepo(ctx.e, repo.ID); err != nil {
  963. return fmt.Errorf("copyDefaultWebhooksToRepo: %v", err)
  964. }
  965. return nil
  966. }
  967. func countRepositories(userID int64, private bool) int64 {
  968. sess := x.Where("id > 0")
  969. if userID > 0 {
  970. sess.And("owner_id = ?", userID)
  971. }
  972. if !private {
  973. sess.And("is_private=?", false)
  974. }
  975. count, err := sess.Count(new(Repository))
  976. if err != nil {
  977. log.Error("countRepositories: %v", err)
  978. }
  979. return count
  980. }
  981. // CountRepositories returns number of repositories.
  982. // Argument private only takes effect when it is false,
  983. // set it true to count all repositories.
  984. func CountRepositories(private bool) int64 {
  985. return countRepositories(-1, private)
  986. }
  987. // CountUserRepositories returns number of repositories user owns.
  988. // Argument private only takes effect when it is false,
  989. // set it true to count all repositories.
  990. func CountUserRepositories(userID int64, private bool) int64 {
  991. return countRepositories(userID, private)
  992. }
  993. // RepoPath returns repository path by given user and repository name.
  994. func RepoPath(userName, repoName string) string {
  995. return filepath.Join(UserPath(userName), strings.ToLower(repoName)+".git")
  996. }
  997. // IncrementRepoForkNum increment repository fork number
  998. func IncrementRepoForkNum(ctx DBContext, repoID int64) error {
  999. _, err := ctx.e.Exec("UPDATE `repository` SET num_forks=num_forks+1 WHERE id=?", repoID)
  1000. return err
  1001. }
  1002. // TransferOwnership transfers all corresponding setting from old user to new one.
  1003. func TransferOwnership(doer *User, newOwnerName string, repo *Repository) error {
  1004. newOwner, err := GetUserByName(newOwnerName)
  1005. if err != nil {
  1006. return fmt.Errorf("get new owner '%s': %v", newOwnerName, err)
  1007. }
  1008. // Check if new owner has repository with same name.
  1009. has, err := IsRepositoryExist(newOwner, repo.Name)
  1010. if err != nil {
  1011. return fmt.Errorf("IsRepositoryExist: %v", err)
  1012. } else if has {
  1013. return ErrRepoAlreadyExist{newOwnerName, repo.Name}
  1014. }
  1015. sess := x.NewSession()
  1016. defer sess.Close()
  1017. if err = sess.Begin(); err != nil {
  1018. return fmt.Errorf("sess.Begin: %v", err)
  1019. }
  1020. oldOwner := repo.Owner
  1021. // Note: we have to set value here to make sure recalculate accesses is based on
  1022. // new owner.
  1023. repo.OwnerID = newOwner.ID
  1024. repo.Owner = newOwner
  1025. repo.OwnerName = newOwner.Name
  1026. // Update repository.
  1027. if _, err := sess.ID(repo.ID).Update(repo); err != nil {
  1028. return fmt.Errorf("update owner: %v", err)
  1029. }
  1030. // Remove redundant collaborators.
  1031. collaborators, err := repo.getCollaborators(sess)
  1032. if err != nil {
  1033. return fmt.Errorf("getCollaborators: %v", err)
  1034. }
  1035. // Dummy object.
  1036. collaboration := &Collaboration{RepoID: repo.ID}
  1037. for _, c := range collaborators {
  1038. if c.ID != newOwner.ID {
  1039. isMember, err := isOrganizationMember(sess, newOwner.ID, c.ID)
  1040. if err != nil {
  1041. return fmt.Errorf("IsOrgMember: %v", err)
  1042. } else if !isMember {
  1043. continue
  1044. }
  1045. }
  1046. collaboration.UserID = c.ID
  1047. if _, err = sess.Delete(collaboration); err != nil {
  1048. return fmt.Errorf("remove collaborator '%d': %v", c.ID, err)
  1049. }
  1050. }
  1051. // Remove old team-repository relations.
  1052. if oldOwner.IsOrganization() {
  1053. if err = oldOwner.removeOrgRepo(sess, repo.ID); err != nil {
  1054. return fmt.Errorf("removeOrgRepo: %v", err)
  1055. }
  1056. }
  1057. if newOwner.IsOrganization() {
  1058. if err := newOwner.GetTeams(); err != nil {
  1059. return fmt.Errorf("GetTeams: %v", err)
  1060. }
  1061. for _, t := range newOwner.Teams {
  1062. if t.IncludesAllRepositories {
  1063. if err := t.addRepository(sess, repo); err != nil {
  1064. return fmt.Errorf("addRepository: %v", err)
  1065. }
  1066. }
  1067. }
  1068. } else if err = repo.recalculateAccesses(sess); err != nil {
  1069. // Organization called this in addRepository method.
  1070. return fmt.Errorf("recalculateAccesses: %v", err)
  1071. }
  1072. // Update repository count.
  1073. if _, err = sess.Exec("UPDATE `user` SET num_repos=num_repos+1 WHERE id=?", newOwner.ID); err != nil {
  1074. return fmt.Errorf("increase new owner repository count: %v", err)
  1075. } else if _, err = sess.Exec("UPDATE `user` SET num_repos=num_repos-1 WHERE id=?", oldOwner.ID); err != nil {
  1076. return fmt.Errorf("decrease old owner repository count: %v", err)
  1077. }
  1078. if err = watchRepo(sess, doer.ID, repo.ID, true); err != nil {
  1079. return fmt.Errorf("watchRepo: %v", err)
  1080. }
  1081. // Remove watch for organization.
  1082. if oldOwner.IsOrganization() {
  1083. if err = watchRepo(sess, oldOwner.ID, repo.ID, false); err != nil {
  1084. return fmt.Errorf("watchRepo [false]: %v", err)
  1085. }
  1086. }
  1087. // Rename remote repository to new path and delete local copy.
  1088. dir := UserPath(newOwner.Name)
  1089. if err := os.MkdirAll(dir, os.ModePerm); err != nil {
  1090. return fmt.Errorf("Failed to create dir %s: %v", dir, err)
  1091. }
  1092. if err = os.Rename(RepoPath(oldOwner.Name, repo.Name), RepoPath(newOwner.Name, repo.Name)); err != nil {
  1093. return fmt.Errorf("rename repository directory: %v", err)
  1094. }
  1095. // Rename remote wiki repository to new path and delete local copy.
  1096. wikiPath := WikiPath(oldOwner.Name, repo.Name)
  1097. if com.IsExist(wikiPath) {
  1098. if err = os.Rename(wikiPath, WikiPath(newOwner.Name, repo.Name)); err != nil {
  1099. return fmt.Errorf("rename repository wiki: %v", err)
  1100. }
  1101. }
  1102. // If there was previously a redirect at this location, remove it.
  1103. if err = deleteRepoRedirect(sess, newOwner.ID, repo.Name); err != nil {
  1104. return fmt.Errorf("delete repo redirect: %v", err)
  1105. }
  1106. if err := NewRepoRedirect(DBContext{sess}, oldOwner.ID, repo.ID, repo.Name, repo.Name); err != nil {
  1107. return fmt.Errorf("NewRepoRedirect: %v", err)
  1108. }
  1109. return sess.Commit()
  1110. }
  1111. // ChangeRepositoryName changes all corresponding setting from old repository name to new one.
  1112. func ChangeRepositoryName(doer *User, repo *Repository, newRepoName string) (err error) {
  1113. oldRepoName := repo.Name
  1114. newRepoName = strings.ToLower(newRepoName)
  1115. if err = IsUsableRepoName(newRepoName); err != nil {
  1116. return err
  1117. }
  1118. if err := repo.GetOwner(); err != nil {
  1119. return err
  1120. }
  1121. has, err := IsRepositoryExist(repo.Owner, newRepoName)
  1122. if err != nil {
  1123. return fmt.Errorf("IsRepositoryExist: %v", err)
  1124. } else if has {
  1125. return ErrRepoAlreadyExist{repo.Owner.Name, newRepoName}
  1126. }
  1127. newRepoPath := RepoPath(repo.Owner.Name, newRepoName)
  1128. if err = os.Rename(repo.RepoPath(), newRepoPath); err != nil {
  1129. return fmt.Errorf("rename repository directory: %v", err)
  1130. }
  1131. wikiPath := repo.WikiPath()
  1132. if com.IsExist(wikiPath) {
  1133. if err = os.Rename(wikiPath, WikiPath(repo.Owner.Name, newRepoName)); err != nil {
  1134. return fmt.Errorf("rename repository wiki: %v", err)
  1135. }
  1136. }
  1137. sess := x.NewSession()
  1138. defer sess.Close()
  1139. if err = sess.Begin(); err != nil {
  1140. return fmt.Errorf("sess.Begin: %v", err)
  1141. }
  1142. // If there was previously a redirect at this location, remove it.
  1143. if err = deleteRepoRedirect(sess, repo.OwnerID, newRepoName); err != nil {
  1144. return fmt.Errorf("delete repo redirect: %v", err)
  1145. }
  1146. if err := NewRepoRedirect(DBContext{sess}, repo.Owner.ID, repo.ID, oldRepoName, newRepoName); err != nil {
  1147. return err
  1148. }
  1149. return sess.Commit()
  1150. }
  1151. func getRepositoriesByForkID(e Engine, forkID int64) ([]*Repository, error) {
  1152. repos := make([]*Repository, 0, 10)
  1153. return repos, e.
  1154. Where("fork_id=?", forkID).
  1155. Find(&repos)
  1156. }
  1157. // GetRepositoriesByForkID returns all repositories with given fork ID.
  1158. func GetRepositoriesByForkID(forkID int64) ([]*Repository, error) {
  1159. return getRepositoriesByForkID(x, forkID)
  1160. }
  1161. func updateRepository(e Engine, repo *Repository, visibilityChanged bool) (err error) {
  1162. repo.LowerName = strings.ToLower(repo.Name)
  1163. if len(repo.Description) > 255 {
  1164. repo.Description = repo.Description[:255]
  1165. }
  1166. if len(repo.Website) > 255 {
  1167. repo.Website = repo.Website[:255]
  1168. }
  1169. if _, err = e.ID(repo.ID).AllCols().Update(repo); err != nil {
  1170. return fmt.Errorf("update: %v", err)
  1171. }
  1172. if visibilityChanged {
  1173. if err = repo.getOwner(e); err != nil {
  1174. return fmt.Errorf("getOwner: %v", err)
  1175. }
  1176. if repo.Owner.IsOrganization() {
  1177. // Organization repository need to recalculate access table when visibility is changed.
  1178. if err = repo.recalculateTeamAccesses(e, 0); err != nil {
  1179. return fmt.Errorf("recalculateTeamAccesses: %v", err)
  1180. }
  1181. }
  1182. // If repo has become private, we need to set its actions to private.
  1183. if repo.IsPrivate {
  1184. _, err = e.Where("repo_id = ?", repo.ID).Cols("is_private").Update(&Action{
  1185. IsPrivate: true,
  1186. })
  1187. if err != nil {
  1188. return err
  1189. }
  1190. }
  1191. // Create/Remove git-daemon-export-ok for git-daemon...
  1192. daemonExportFile := path.Join(repo.RepoPath(), `git-daemon-export-ok`)
  1193. if repo.IsPrivate && com.IsExist(daemonExportFile) {
  1194. if err = os.Remove(daemonExportFile); err != nil {
  1195. log.Error("Failed to remove %s: %v", daemonExportFile, err)
  1196. }
  1197. } else if !repo.IsPrivate && !com.IsExist(daemonExportFile) {
  1198. if f, err := os.Create(daemonExportFile); err != nil {
  1199. log.Error("Failed to create %s: %v", daemonExportFile, err)
  1200. } else {
  1201. f.Close()
  1202. }
  1203. }
  1204. forkRepos, err := getRepositoriesByForkID(e, repo.ID)
  1205. if err != nil {
  1206. return fmt.Errorf("getRepositoriesByForkID: %v", err)
  1207. }
  1208. for i := range forkRepos {
  1209. forkRepos[i].IsPrivate = repo.IsPrivate
  1210. if err = updateRepository(e, forkRepos[i], true); err != nil {
  1211. return fmt.Errorf("updateRepository[%d]: %v", forkRepos[i].ID, err)
  1212. }
  1213. }
  1214. if err = repo.updateSize(e); err != nil {
  1215. log.Error("Failed to update size for repository: %v", err)
  1216. }
  1217. }
  1218. return nil
  1219. }
  1220. // UpdateRepositoryCtx updates a repository with db context
  1221. func UpdateRepositoryCtx(ctx DBContext, repo *Repository, visibilityChanged bool) error {
  1222. return updateRepository(ctx.e, repo, visibilityChanged)
  1223. }
  1224. // UpdateRepository updates a repository
  1225. func UpdateRepository(repo *Repository, visibilityChanged bool) (err error) {
  1226. sess := x.NewSession()
  1227. defer sess.Close()
  1228. if err = sess.Begin(); err != nil {
  1229. return err
  1230. }
  1231. if err = updateRepository(sess, repo, visibilityChanged); err != nil {
  1232. return fmt.Errorf("updateRepository: %v", err)
  1233. }
  1234. return sess.Commit()
  1235. }
  1236. // UpdateRepositoryUpdatedTime updates a repository's updated time
  1237. func UpdateRepositoryUpdatedTime(repoID int64, updateTime time.Time) error {
  1238. _, err := x.Exec("UPDATE repository SET updated_unix = ? WHERE id = ?", updateTime.Unix(), repoID)
  1239. return err
  1240. }
  1241. // UpdateRepositoryUnits updates a repository's units
  1242. func UpdateRepositoryUnits(repo *Repository, units []RepoUnit, deleteUnitTypes []UnitType) (err error) {
  1243. sess := x.NewSession()
  1244. defer sess.Close()
  1245. if err = sess.Begin(); err != nil {
  1246. return err
  1247. }
  1248. // Delete existing settings of units before adding again
  1249. for _, u := range units {
  1250. deleteUnitTypes = append(deleteUnitTypes, u.Type)
  1251. }
  1252. if _, err = sess.Where("repo_id = ?", repo.ID).In("type", deleteUnitTypes).Delete(new(RepoUnit)); err != nil {
  1253. return err
  1254. }
  1255. if _, err = sess.Insert(units); err != nil {
  1256. return err
  1257. }
  1258. return sess.Commit()
  1259. }
  1260. // DeleteRepository deletes a repository for a user or organization.
  1261. func DeleteRepository(doer *User, uid, repoID int64) error {
  1262. // In case is a organization.
  1263. org, err := GetUserByID(uid)
  1264. if err != nil {
  1265. return err
  1266. }
  1267. if org.IsOrganization() {
  1268. if err = org.GetTeams(); err != nil {
  1269. return err
  1270. }
  1271. }
  1272. sess := x.NewSession()
  1273. defer sess.Close()
  1274. if err = sess.Begin(); err != nil {
  1275. return err
  1276. }
  1277. repo := &Repository{ID: repoID, OwnerID: uid}
  1278. has, err := sess.Get(repo)
  1279. if err != nil {
  1280. return err
  1281. } else if !has {
  1282. return ErrRepoNotExist{repoID, uid, "", ""}
  1283. }
  1284. // Delete Deploy Keys
  1285. deployKeys, err := listDeployKeys(sess, repo.ID)
  1286. if err != nil {
  1287. return fmt.Errorf("listDeployKeys: %v", err)
  1288. }
  1289. for _, dKey := range deployKeys {
  1290. if err := deleteDeployKey(sess, doer, dKey.ID); err != nil {
  1291. return fmt.Errorf("deleteDeployKeys: %v", err)
  1292. }
  1293. }
  1294. if cnt, err := sess.ID(repoID).Delete(&Repository{}); err != nil {
  1295. return err
  1296. } else if cnt != 1 {
  1297. return ErrRepoNotExist{repoID, uid, "", ""}
  1298. }
  1299. if org.IsOrganization() {
  1300. for _, t := range org.Teams {
  1301. if !t.hasRepository(sess, repoID) {
  1302. continue
  1303. } else if err = t.removeRepository(sess, repo, false); err != nil {
  1304. return err
  1305. }
  1306. }
  1307. }
  1308. attachments := make([]*Attachment, 0, 20)
  1309. if err = sess.Join("INNER", "`release`", "`release`.id = `attachment`.release_id").
  1310. Where("`release`.repo_id = ?", repoID).
  1311. Find(&attachments); err != nil {
  1312. return err
  1313. }
  1314. releaseAttachments := make([]string, 0, len(attachments))
  1315. for i := 0; i < len(attachments); i++ {
  1316. releaseAttachments = append(releaseAttachments, attachments[i].LocalPath())
  1317. }
  1318. if err = deleteBeans(sess,
  1319. &Access{RepoID: repo.ID},
  1320. &Action{RepoID: repo.ID},
  1321. &Watch{RepoID: repoID},
  1322. &Star{RepoID: repoID},
  1323. &Mirror{RepoID: repoID},
  1324. &Milestone{RepoID: repoID},
  1325. &Release{RepoID: repoID},
  1326. &Collaboration{RepoID: repoID},
  1327. &PullRequest{BaseRepoID: repoID},
  1328. &RepoUnit{RepoID: repoID},
  1329. &RepoRedirect{RedirectRepoID: repoID},
  1330. &Webhook{RepoID: repoID},
  1331. &HookTask{RepoID: repoID},
  1332. &Notification{RepoID: repoID},
  1333. &CommitStatus{RepoID: repoID},
  1334. &RepoIndexerStatus{RepoID: repoID},
  1335. &Comment{RefRepoID: repoID},
  1336. &Task{RepoID: repoID},
  1337. ); err != nil {
  1338. return fmt.Errorf("deleteBeans: %v", err)
  1339. }
  1340. deleteCond := builder.Select("id").From("issue").Where(builder.Eq{"repo_id": repoID})
  1341. // Delete comments and attachments
  1342. if _, err = sess.In("issue_id", deleteCond).
  1343. Delete(&Comment{}); err != nil {
  1344. return err
  1345. }
  1346. if _, err = sess.In("issue_id", deleteCond).
  1347. Delete(&IssueUser{}); err != nil {
  1348. return err
  1349. }
  1350. if _, err = sess.In("issue_id", deleteCond).
  1351. Delete(&Reaction{}); err != nil {
  1352. return err
  1353. }
  1354. if _, err = sess.In("issue_id", deleteCond).
  1355. Delete(&IssueWatch{}); err != nil {
  1356. return err
  1357. }
  1358. if _, err = sess.In("issue_id", deleteCond).
  1359. Delete(&Stopwatch{}); err != nil {
  1360. return err
  1361. }
  1362. attachments = attachments[:0]
  1363. if err = sess.Join("INNER", "issue", "issue.id = attachment.issue_id").
  1364. Where("issue.repo_id = ?", repoID).
  1365. Find(&attachments); err != nil {
  1366. return err
  1367. }
  1368. attachmentPaths := make([]string, 0, len(attachments))
  1369. for j := range attachments {
  1370. attachmentPaths = append(attachmentPaths, attachments[j].LocalPath())
  1371. }
  1372. if _, err = sess.In("issue_id", deleteCond).
  1373. Delete(&Attachment{}); err != nil {
  1374. return err
  1375. }
  1376. if _, err = sess.Delete(&Issue{RepoID: repoID}); err != nil {
  1377. return err
  1378. }
  1379. if _, err = sess.Where("repo_id = ?", repoID).Delete(new(RepoUnit)); err != nil {
  1380. return err
  1381. }
  1382. if repo.IsFork {
  1383. if _, err = sess.Exec("UPDATE `repository` SET num_forks=num_forks-1 WHERE id=?", repo.ForkID); err != nil {
  1384. return fmt.Errorf("decrease fork count: %v", err)
  1385. }
  1386. }
  1387. if _, err = sess.Exec("UPDATE `user` SET num_repos=num_repos-1 WHERE id=?", uid); err != nil {
  1388. return err
  1389. }
  1390. // FIXME: Remove repository files should be executed after transaction succeed.
  1391. repoPath := repo.RepoPath()
  1392. removeAllWithNotice(sess, "Delete repository files", repoPath)
  1393. err = repo.deleteWiki(sess)
  1394. if err != nil {
  1395. return err
  1396. }
  1397. // Remove LFS objects
  1398. var lfsObjects []*LFSMetaObject
  1399. if err = sess.Where("repository_id=?", repoID).Find(&lfsObjects); err != nil {
  1400. return err
  1401. }
  1402. for _, v := range lfsObjects {
  1403. count, err := sess.Count(&LFSMetaObject{Oid: v.Oid})
  1404. if err != nil {
  1405. return err
  1406. }
  1407. if count > 1 {
  1408. continue
  1409. }
  1410. oidPath := filepath.Join(setting.LFS.ContentPath, v.Oid[0:2], v.Oid[2:4], v.Oid[4:len(v.Oid)])
  1411. removeAllWithNotice(sess, "Delete orphaned LFS file", oidPath)
  1412. }
  1413. if _, err := sess.Delete(&LFSMetaObject{RepositoryID: repoID}); err != nil {
  1414. return err
  1415. }
  1416. if repo.NumForks > 0 {
  1417. if _, err = sess.Exec("UPDATE `repository` SET fork_id=0,is_fork=? WHERE fork_id=?", false, repo.ID); err != nil {
  1418. log.Error("reset 'fork_id' and 'is_fork': %v", err)
  1419. }
  1420. }
  1421. if err = sess.Commit(); err != nil {
  1422. if len(deployKeys) > 0 {
  1423. // We need to rewrite the public keys because the commit failed
  1424. if err2 := RewriteAllPublicKeys(); err2 != nil {
  1425. return fmt.Errorf("Commit: %v SSH Keys: %v", err, err2)
  1426. }
  1427. }
  1428. return fmt.Errorf("Commit: %v", err)
  1429. }
  1430. sess.Close()
  1431. // We should always delete the files after the database transaction succeed. If
  1432. // we delete the file but the database rollback, the repository will be borken.
  1433. // Remove issue attachment files.
  1434. for i := range attachmentPaths {
  1435. removeAllWithNotice(x, "Delete issue attachment", attachmentPaths[i])
  1436. }
  1437. // Remove release attachment files.
  1438. for i := range releaseAttachments {
  1439. removeAllWithNotice(x, "Delete release attachment", releaseAttachments[i])
  1440. }
  1441. if len(repo.Avatar) > 0 {
  1442. avatarPath := repo.CustomAvatarPath()
  1443. if com.IsExist(avatarPath) {
  1444. if err := os.Remove(avatarPath); err != nil {
  1445. return fmt.Errorf("Failed to remove %s: %v", avatarPath, err)
  1446. }
  1447. }
  1448. }
  1449. return nil
  1450. }
  1451. // GetRepositoryByOwnerAndName returns the repository by given ownername and reponame.
  1452. func GetRepositoryByOwnerAndName(ownerName, repoName string) (*Repository, error) {
  1453. return getRepositoryByOwnerAndName(x, ownerName, repoName)
  1454. }
  1455. func getRepositoryByOwnerAndName(e Engine, ownerName, repoName string) (*Repository, error) {
  1456. var repo Repository
  1457. has, err := e.Table("repository").Select("repository.*").
  1458. Join("INNER", "`user`", "`user`.id = repository.owner_id").
  1459. Where("repository.lower_name = ?", strings.ToLower(repoName)).
  1460. And("`user`.lower_name = ?", strings.ToLower(ownerName)).
  1461. Get(&repo)
  1462. if err != nil {
  1463. return nil, err
  1464. } else if !has {
  1465. return nil, ErrRepoNotExist{0, 0, ownerName, repoName}
  1466. }
  1467. return &repo, nil
  1468. }
  1469. // GetRepositoryByName returns the repository by given name under user if exists.
  1470. func GetRepositoryByName(ownerID int64, name string) (*Repository, error) {
  1471. repo := &Repository{
  1472. OwnerID: ownerID,
  1473. LowerName: strings.ToLower(name),
  1474. }
  1475. has, err := x.Get(repo)
  1476. if err != nil {
  1477. return nil, err
  1478. } else if !has {
  1479. return nil, ErrRepoNotExist{0, ownerID, "", name}
  1480. }
  1481. return repo, err
  1482. }
  1483. func getRepositoryByID(e Engine, id int64) (*Repository, error) {
  1484. repo := new(Repository)
  1485. has, err := e.ID(id).Get(repo)
  1486. if err != nil {
  1487. return nil, err
  1488. } else if !has {
  1489. return nil, ErrRepoNotExist{id, 0, "", ""}
  1490. }
  1491. return repo, nil
  1492. }
  1493. // GetRepositoryByID returns the repository by given id if exists.
  1494. func GetRepositoryByID(id int64) (*Repository, error) {
  1495. return getRepositoryByID(x, id)
  1496. }
  1497. // GetRepositoryByIDCtx returns the repository by given id if exists.
  1498. func GetRepositoryByIDCtx(ctx DBContext, id int64) (*Repository, error) {
  1499. return getRepositoryByID(ctx.e, id)
  1500. }
  1501. // GetRepositoriesMapByIDs returns the repositories by given id slice.
  1502. func GetRepositoriesMapByIDs(ids []int64) (map[int64]*Repository, error) {
  1503. var repos = make(map[int64]*Repository, len(ids))
  1504. return repos, x.In("id", ids).Find(&repos)
  1505. }
  1506. // GetUserRepositories returns a list of repositories of given user.
  1507. func GetUserRepositories(userID int64, private bool, page, pageSize int, orderBy string) ([]*Repository, error) {
  1508. if len(orderBy) == 0 {
  1509. orderBy = "updated_unix DESC"
  1510. }
  1511. sess := x.
  1512. Where("owner_id = ?", userID).
  1513. OrderBy(orderBy)
  1514. if !private {
  1515. sess.And("is_private=?", false)
  1516. }
  1517. if page <= 0 {
  1518. page = 1
  1519. }
  1520. sess.Limit(pageSize, (page-1)*pageSize)
  1521. repos := make([]*Repository, 0, pageSize)
  1522. return repos, sess.Find(&repos)
  1523. }
  1524. // GetUserMirrorRepositories returns a list of mirror repositories of given user.
  1525. func GetUserMirrorRepositories(userID int64) ([]*Repository, error) {
  1526. repos := make([]*Repository, 0, 10)
  1527. return repos, x.
  1528. Where("owner_id = ?", userID).
  1529. And("is_mirror = ?", true).
  1530. Find(&repos)
  1531. }
  1532. func getRepositoryCount(e Engine, u *User) (int64, error) {
  1533. return e.Count(&Repository{OwnerID: u.ID})
  1534. }
  1535. func getPublicRepositoryCount(e Engine, u *User) (int64, error) {
  1536. return e.Where("is_private = ?", false).Count(&Repository{OwnerID: u.ID})
  1537. }
  1538. func getPrivateRepositoryCount(e Engine, u *User) (int64, error) {
  1539. return e.Where("is_private = ?", true).Count(&Repository{OwnerID: u.ID})
  1540. }
  1541. // GetRepositoryCount returns the total number of repositories of user.
  1542. func GetRepositoryCount(u *User) (int64, error) {
  1543. return getRepositoryCount(x, u)
  1544. }
  1545. // GetPublicRepositoryCount returns the total number of public repositories of user.
  1546. func GetPublicRepositoryCount(u *User) (int64, error) {
  1547. return getPublicRepositoryCount(x, u)
  1548. }
  1549. // GetPrivateRepositoryCount returns the total number of private repositories of user.
  1550. func GetPrivateRepositoryCount(u *User) (int64, error) {
  1551. return getPrivateRepositoryCount(x, u)
  1552. }
  1553. // DeleteRepositoryArchives deletes all repositories' archives.
  1554. func DeleteRepositoryArchives() error {
  1555. return x.
  1556. Where("id > 0").
  1557. Iterate(new(Repository),
  1558. func(idx int, bean interface{}) error {
  1559. repo := bean.(*Repository)
  1560. return os.RemoveAll(filepath.Join(repo.RepoPath(), "archives"))
  1561. })
  1562. }
  1563. // DeleteOldRepositoryArchives deletes old repository archives.
  1564. func DeleteOldRepositoryArchives(ctx context.Context) {
  1565. log.Trace("Doing: ArchiveCleanup")
  1566. if err := x.Where("id > 0").Iterate(new(Repository), func(idx int, bean interface{}) error {
  1567. return deleteOldRepositoryArchives(ctx, idx, bean)
  1568. }); err != nil {
  1569. log.Error("ArchiveClean: %v", err)
  1570. }
  1571. }
  1572. func deleteOldRepositoryArchives(ctx context.Context, idx int, bean interface{}) error {
  1573. repo := bean.(*Repository)
  1574. basePath := filepath.Join(repo.RepoPath(), "archives")
  1575. for _, ty := range []string{"zip", "targz"} {
  1576. select {
  1577. case <-ctx.Done():
  1578. return fmt.Errorf("Aborted due to shutdown:\nin delete of old repository archives %v\nat delete file %s", repo, ty)
  1579. default:
  1580. }
  1581. path := filepath.Join(basePath, ty)
  1582. file, err := os.Open(path)
  1583. if err != nil {
  1584. if !os.IsNotExist(err) {
  1585. log.Warn("Unable to open directory %s: %v", path, err)
  1586. return err
  1587. }
  1588. // If the directory doesn't exist, that's okay.
  1589. continue
  1590. }
  1591. files, err := file.Readdir(0)
  1592. file.Close()
  1593. if err != nil {
  1594. log.Warn("Unable to read directory %s: %v", path, err)
  1595. return err
  1596. }
  1597. minimumOldestTime := time.Now().Add(-setting.Cron.ArchiveCleanup.OlderThan)
  1598. for _, info := range files {
  1599. if info.ModTime().Before(minimumOldestTime) && !info.IsDir() {
  1600. select {
  1601. case <-ctx.Done():
  1602. return fmt.Errorf("Aborted due to shutdown:\nin delete of old repository archives %v\nat delete file %s - %s", repo, ty, info.Name())
  1603. default:
  1604. }
  1605. toDelete := filepath.Join(path, info.Name())
  1606. // This is a best-effort purge, so we do not check error codes to confirm removal.
  1607. if err = os.Remove(toDelete); err != nil {
  1608. log.Trace("Unable to delete %s, but proceeding: %v", toDelete, err)
  1609. }
  1610. }
  1611. }
  1612. }
  1613. return nil
  1614. }
  1615. func gatherMissingRepoRecords() ([]*Repository, error) {
  1616. repos := make([]*Repository, 0, 10)
  1617. if err := x.
  1618. Where("id > 0").
  1619. Iterate(new(Repository),
  1620. func(idx int, bean interface{}) error {
  1621. repo := bean.(*Repository)
  1622. if !com.IsDir(repo.RepoPath()) {
  1623. repos = append(repos, repo)
  1624. }
  1625. return nil
  1626. }); err != nil {
  1627. if err2 := CreateRepositoryNotice(fmt.Sprintf("gatherMissingRepoRecords: %v", err)); err2 != nil {
  1628. return nil, fmt.Errorf("CreateRepositoryNotice: %v", err)
  1629. }
  1630. }
  1631. return repos, nil
  1632. }
  1633. // DeleteMissingRepositories deletes all repository records that lost Git files.
  1634. func DeleteMissingRepositories(doer *User) error {
  1635. repos, err := gatherMissingRepoRecords()
  1636. if err != nil {
  1637. return fmt.Errorf("gatherMissingRepoRecords: %v", err)
  1638. }
  1639. if len(repos) == 0 {
  1640. return nil
  1641. }
  1642. for _, repo := range repos {
  1643. log.Trace("Deleting %d/%d...", repo.OwnerID, repo.ID)
  1644. if err := DeleteRepository(doer, repo.OwnerID, repo.ID); err != nil {
  1645. if err2 := CreateRepositoryNotice(fmt.Sprintf("DeleteRepository [%d]: %v", repo.ID, err)); err2 != nil {
  1646. return fmt.Errorf("CreateRepositoryNotice: %v", err)
  1647. }
  1648. }
  1649. }
  1650. return nil
  1651. }
  1652. // ReinitMissingRepositories reinitializes all repository records that lost Git files.
  1653. func ReinitMissingRepositories() error {
  1654. repos, err := gatherMissingRepoRecords()
  1655. if err != nil {
  1656. return fmt.Errorf("gatherMissingRepoRecords: %v", err)
  1657. }
  1658. if len(repos) == 0 {
  1659. return nil
  1660. }
  1661. for _, repo := range repos {
  1662. log.Trace("Initializing %d/%d...", repo.OwnerID, repo.ID)
  1663. if err := git.InitRepository(repo.RepoPath(), true); err != nil {
  1664. if err2 := CreateRepositoryNotice(fmt.Sprintf("InitRepository [%d]: %v", repo.ID, err)); err2 != nil {
  1665. return fmt.Errorf("CreateRepositoryNotice: %v", err)
  1666. }
  1667. }
  1668. }
  1669. return nil
  1670. }
  1671. // SyncRepositoryHooks rewrites all repositories' pre-receive, update and post-receive hooks
  1672. // to make sure the binary and custom conf path are up-to-date.
  1673. func SyncRepositoryHooks() error {
  1674. return x.Cols("owner_id", "name").Where("id > 0").Iterate(new(Repository),
  1675. func(idx int, bean interface{}) error {
  1676. if err := createDelegateHooks(bean.(*Repository).RepoPath()); err != nil {
  1677. return fmt.Errorf("SyncRepositoryHook: %v", err)
  1678. }
  1679. if bean.(*Repository).HasWiki() {
  1680. if err := createDelegateHooks(bean.(*Repository).WikiPath()); err != nil {
  1681. return fmt.Errorf("SyncRepositoryHook: %v", err)
  1682. }
  1683. }
  1684. return nil
  1685. })
  1686. }
  1687. // GitFsck calls 'git fsck' to check repository health.
  1688. func GitFsck(ctx context.Context) {
  1689. log.Trace("Doing: GitFsck")
  1690. if err := x.
  1691. Where("id>0 AND is_fsck_enabled=?", true).BufferSize(setting.Database.IterateBufferSize).
  1692. Iterate(new(Repository),
  1693. func(idx int, bean interface{}) error {
  1694. select {
  1695. case <-ctx.Done():
  1696. return fmt.Errorf("Aborted due to shutdown")
  1697. default:
  1698. }
  1699. repo := bean.(*Repository)
  1700. repoPath := repo.RepoPath()
  1701. log.Trace("Running health check on repository %s", repoPath)
  1702. if err := git.Fsck(repoPath, setting.Cron.RepoHealthCheck.Timeout, setting.Cron.RepoHealthCheck.Args...); err != nil {
  1703. desc := fmt.Sprintf("Failed to health check repository (%s): %v", repoPath, err)
  1704. log.Warn(desc)
  1705. if err = CreateRepositoryNotice(desc); err != nil {
  1706. log.Error("CreateRepositoryNotice: %v", err)
  1707. }
  1708. }
  1709. return nil
  1710. }); err != nil {
  1711. log.Error("GitFsck: %v", err)
  1712. }
  1713. log.Trace("Finished: GitFsck")
  1714. }
  1715. // GitGcRepos calls 'git gc' to remove unnecessary files and optimize the local repository
  1716. func GitGcRepos() error {
  1717. args := append([]string{"gc"}, setting.Git.GCArgs...)
  1718. return x.
  1719. Where("id > 0").BufferSize(setting.Database.IterateBufferSize).
  1720. Iterate(new(Repository),
  1721. func(idx int, bean interface{}) error {
  1722. repo := bean.(*Repository)
  1723. if err := repo.GetOwner(); err != nil {
  1724. return err
  1725. }
  1726. if stdout, err := git.NewCommand(args...).
  1727. SetDescription(fmt.Sprintf("Repository Garbage Collection: %s", repo.FullName())).
  1728. RunInDirTimeout(
  1729. time.Duration(setting.Git.Timeout.GC)*time.Second,
  1730. repo.RepoPath()); err != nil {
  1731. log.Error("Repository garbage collection failed for %v. Stdout: %s\nError: %v", repo, stdout, err)
  1732. return fmt.Errorf("Repository garbage collection failed: Error: %v", err)
  1733. }
  1734. return nil
  1735. })
  1736. }
  1737. type repoChecker struct {
  1738. querySQL, correctSQL string
  1739. desc string
  1740. }
  1741. func repoStatsCheck(ctx context.Context, checker *repoChecker) {
  1742. results, err := x.Query(checker.querySQL)
  1743. if err != nil {
  1744. log.Error("Select %s: %v", checker.desc, err)
  1745. return
  1746. }
  1747. for _, result := range results {
  1748. select {
  1749. case <-ctx.Done():
  1750. log.Warn("CheckRepoStats: Aborting due to shutdown")
  1751. return
  1752. default:
  1753. }
  1754. id := com.StrTo(result["id"]).MustInt64()
  1755. log.Trace("Updating %s: %d", checker.desc, id)
  1756. _, err = x.Exec(checker.correctSQL, id, id)
  1757. if err != nil {
  1758. log.Error("Update %s[%d]: %v", checker.desc, id, err)
  1759. }
  1760. }
  1761. }
  1762. // CheckRepoStats checks the repository stats
  1763. func CheckRepoStats(ctx context.Context) {
  1764. log.Trace("Doing: CheckRepoStats")
  1765. checkers := []*repoChecker{
  1766. // Repository.NumWatches
  1767. {
  1768. "SELECT repo.id FROM `repository` repo WHERE repo.num_watches!=(SELECT COUNT(*) FROM `watch` WHERE repo_id=repo.id AND mode<>2)",
  1769. "UPDATE `repository` SET num_watches=(SELECT COUNT(*) FROM `watch` WHERE repo_id=? AND mode<>2) WHERE id=?",
  1770. "repository count 'num_watches'",
  1771. },
  1772. // Repository.NumStars
  1773. {
  1774. "SELECT repo.id FROM `repository` repo WHERE repo.num_stars!=(SELECT COUNT(*) FROM `star` WHERE repo_id=repo.id)",
  1775. "UPDATE `repository` SET num_stars=(SELECT COUNT(*) FROM `star` WHERE repo_id=?) WHERE id=?",
  1776. "repository count 'num_stars'",
  1777. },
  1778. // Label.NumIssues
  1779. {
  1780. "SELECT label.id FROM `label` WHERE label.num_issues!=(SELECT COUNT(*) FROM `issue_label` WHERE label_id=label.id)",
  1781. "UPDATE `label` SET num_issues=(SELECT COUNT(*) FROM `issue_label` WHERE label_id=?) WHERE id=?",
  1782. "label count 'num_issues'",
  1783. },
  1784. // User.NumRepos
  1785. {
  1786. "SELECT `user`.id FROM `user` WHERE `user`.num_repos!=(SELECT COUNT(*) FROM `repository` WHERE owner_id=`user`.id)",
  1787. "UPDATE `user` SET num_repos=(SELECT COUNT(*) FROM `repository` WHERE owner_id=?) WHERE id=?",
  1788. "user count 'num_repos'",
  1789. },
  1790. // Issue.NumComments
  1791. {
  1792. "SELECT `issue`.id FROM `issue` WHERE `issue`.num_comments!=(SELECT COUNT(*) FROM `comment` WHERE issue_id=`issue`.id AND type=0)",
  1793. "UPDATE `issue` SET num_comments=(SELECT COUNT(*) FROM `comment` WHERE issue_id=? AND type=0) WHERE id=?",
  1794. "issue count 'num_comments'",
  1795. },
  1796. }
  1797. for i := range checkers {
  1798. select {
  1799. case <-ctx.Done():
  1800. log.Warn("CheckRepoStats: Aborting due to shutdown")
  1801. return
  1802. default:
  1803. repoStatsCheck(ctx, checkers[i])
  1804. }
  1805. }
  1806. // ***** START: Repository.NumClosedIssues *****
  1807. desc := "repository count 'num_closed_issues'"
  1808. 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)
  1809. if err != nil {
  1810. log.Error("Select %s: %v", desc, err)
  1811. } else {
  1812. for _, result := range results {
  1813. select {
  1814. case <-ctx.Done():
  1815. log.Warn("CheckRepoStats: Aborting due to shutdown")
  1816. return
  1817. default:
  1818. }
  1819. id := com.StrTo(result["id"]).MustInt64()
  1820. log.Trace("Updating %s: %d", desc, id)
  1821. _, 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)
  1822. if err != nil {
  1823. log.Error("Update %s[%d]: %v", desc, id, err)
  1824. }
  1825. }
  1826. }
  1827. // ***** END: Repository.NumClosedIssues *****
  1828. // ***** START: Repository.NumClosedPulls *****
  1829. desc = "repository count 'num_closed_pulls'"
  1830. 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)
  1831. if err != nil {
  1832. log.Error("Select %s: %v", desc, err)
  1833. } else {
  1834. for _, result := range results {
  1835. select {
  1836. case <-ctx.Done():
  1837. log.Warn("CheckRepoStats: Aborting due to shutdown")
  1838. return
  1839. default:
  1840. }
  1841. id := com.StrTo(result["id"]).MustInt64()
  1842. log.Trace("Updating %s: %d", desc, id)
  1843. _, 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)
  1844. if err != nil {
  1845. log.Error("Update %s[%d]: %v", desc, id, err)
  1846. }
  1847. }
  1848. }
  1849. // ***** END: Repository.NumClosedPulls *****
  1850. // FIXME: use checker when stop supporting old fork repo format.
  1851. // ***** START: Repository.NumForks *****
  1852. results, err = x.Query("SELECT repo.id FROM `repository` repo WHERE repo.num_forks!=(SELECT COUNT(*) FROM `repository` WHERE fork_id=repo.id)")
  1853. if err != nil {
  1854. log.Error("Select repository count 'num_forks': %v", err)
  1855. } else {
  1856. for _, result := range results {
  1857. select {
  1858. case <-ctx.Done():
  1859. log.Warn("CheckRepoStats: Aborting due to shutdown")
  1860. return
  1861. default:
  1862. }
  1863. id := com.StrTo(result["id"]).MustInt64()
  1864. log.Trace("Updating repository count 'num_forks': %d", id)
  1865. repo, err := GetRepositoryByID(id)
  1866. if err != nil {
  1867. log.Error("GetRepositoryByID[%d]: %v", id, err)
  1868. continue
  1869. }
  1870. rawResult, err := x.Query("SELECT COUNT(*) FROM `repository` WHERE fork_id=?", repo.ID)
  1871. if err != nil {
  1872. log.Error("Select count of forks[%d]: %v", repo.ID, err)
  1873. continue
  1874. }
  1875. repo.NumForks = int(parseCountResult(rawResult))
  1876. if err = UpdateRepository(repo, false); err != nil {
  1877. log.Error("UpdateRepository[%d]: %v", id, err)
  1878. continue
  1879. }
  1880. }
  1881. }
  1882. // ***** END: Repository.NumForks *****
  1883. }
  1884. // SetArchiveRepoState sets if a repo is archived
  1885. func (repo *Repository) SetArchiveRepoState(isArchived bool) (err error) {
  1886. repo.IsArchived = isArchived
  1887. _, err = x.Where("id = ?", repo.ID).Cols("is_archived").Update(repo)
  1888. return
  1889. }
  1890. // ___________ __
  1891. // \_ _____/__________| | __
  1892. // | __)/ _ \_ __ \ |/ /
  1893. // | \( <_> ) | \/ <
  1894. // \___ / \____/|__| |__|_ \
  1895. // \/ \/
  1896. // HasForkedRepo checks if given user has already forked a repository with given ID.
  1897. func HasForkedRepo(ownerID, repoID int64) (*Repository, bool) {
  1898. repo := new(Repository)
  1899. has, _ := x.
  1900. Where("owner_id=? AND fork_id=?", ownerID, repoID).
  1901. Get(repo)
  1902. return repo, has
  1903. }
  1904. // CopyLFS copies LFS data from one repo to another
  1905. func CopyLFS(ctx DBContext, newRepo, oldRepo *Repository) error {
  1906. var lfsObjects []*LFSMetaObject
  1907. if err := ctx.e.Where("repository_id=?", oldRepo.ID).Find(&lfsObjects); err != nil {
  1908. return err
  1909. }
  1910. for _, v := range lfsObjects {
  1911. v.ID = 0
  1912. v.RepositoryID = newRepo.ID
  1913. if _, err := ctx.e.Insert(v); err != nil {
  1914. return err
  1915. }
  1916. }
  1917. return nil
  1918. }
  1919. // GetForks returns all the forks of the repository
  1920. func (repo *Repository) GetForks() ([]*Repository, error) {
  1921. forks := make([]*Repository, 0, repo.NumForks)
  1922. return forks, x.Find(&forks, &Repository{ForkID: repo.ID})
  1923. }
  1924. // GetUserFork return user forked repository from this repository, if not forked return nil
  1925. func (repo *Repository) GetUserFork(userID int64) (*Repository, error) {
  1926. var forkedRepo Repository
  1927. has, err := x.Where("fork_id = ?", repo.ID).And("owner_id = ?", userID).Get(&forkedRepo)
  1928. if err != nil {
  1929. return nil, err
  1930. }
  1931. if !has {
  1932. return nil, nil
  1933. }
  1934. return &forkedRepo, nil
  1935. }
  1936. // CustomAvatarPath returns repository custom avatar file path.
  1937. func (repo *Repository) CustomAvatarPath() string {
  1938. // Avatar empty by default
  1939. if len(repo.Avatar) == 0 {
  1940. return ""
  1941. }
  1942. return filepath.Join(setting.RepositoryAvatarUploadPath, repo.Avatar)
  1943. }
  1944. // generateRandomAvatar generates a random avatar for repository.
  1945. func (repo *Repository) generateRandomAvatar(e Engine) error {
  1946. idToString := fmt.Sprintf("%d", repo.ID)
  1947. seed := idToString
  1948. img, err := avatar.RandomImage([]byte(seed))
  1949. if err != nil {
  1950. return fmt.Errorf("RandomImage: %v", err)
  1951. }
  1952. repo.Avatar = idToString
  1953. if err = os.MkdirAll(filepath.Dir(repo.CustomAvatarPath()), os.ModePerm); err != nil {
  1954. return fmt.Errorf("MkdirAll: %v", err)
  1955. }
  1956. fw, err := os.Create(repo.CustomAvatarPath())
  1957. if err != nil {
  1958. return fmt.Errorf("Create: %v", err)
  1959. }
  1960. defer fw.Close()
  1961. if err = png.Encode(fw, img); err != nil {
  1962. return fmt.Errorf("Encode: %v", err)
  1963. }
  1964. log.Info("New random avatar created for repository: %d", repo.ID)
  1965. if _, err := e.ID(repo.ID).Cols("avatar").NoAutoTime().Update(repo); err != nil {
  1966. return err
  1967. }
  1968. return nil
  1969. }
  1970. // RemoveRandomAvatars removes the randomly generated avatars that were created for repositories
  1971. func RemoveRandomAvatars() error {
  1972. return x.
  1973. Where("id > 0").BufferSize(setting.Database.IterateBufferSize).
  1974. Iterate(new(Repository),
  1975. func(idx int, bean interface{}) error {
  1976. repository := bean.(*Repository)
  1977. stringifiedID := strconv.FormatInt(repository.ID, 10)
  1978. if repository.Avatar == stringifiedID {
  1979. return repository.DeleteAvatar()
  1980. }
  1981. return nil
  1982. })
  1983. }
  1984. // RelAvatarLink returns a relative link to the repository's avatar.
  1985. func (repo *Repository) RelAvatarLink() string {
  1986. return repo.relAvatarLink(x)
  1987. }
  1988. func (repo *Repository) relAvatarLink(e Engine) string {
  1989. // If no avatar - path is empty
  1990. avatarPath := repo.CustomAvatarPath()
  1991. if len(avatarPath) == 0 || !com.IsFile(avatarPath) {
  1992. switch mode := setting.RepositoryAvatarFallback; mode {
  1993. case "image":
  1994. return setting.RepositoryAvatarFallbackImage
  1995. case "random":
  1996. if err := repo.generateRandomAvatar(e); err != nil {
  1997. log.Error("generateRandomAvatar: %v", err)
  1998. }
  1999. default:
  2000. // default behaviour: do not display avatar
  2001. return ""
  2002. }
  2003. }
  2004. return setting.AppSubURL + "/repo-avatars/" + repo.Avatar
  2005. }
  2006. // avatarLink returns user avatar absolute link.
  2007. func (repo *Repository) avatarLink(e Engine) string {
  2008. link := repo.relAvatarLink(e)
  2009. // link may be empty!
  2010. if len(link) > 0 {
  2011. if link[0] == '/' && link[1] != '/' {
  2012. return setting.AppURL + strings.TrimPrefix(link, setting.AppSubURL)[1:]
  2013. }
  2014. }
  2015. return link
  2016. }
  2017. // UploadAvatar saves custom avatar for repository.
  2018. // FIXME: split uploads to different subdirs in case we have massive number of repos.
  2019. func (repo *Repository) UploadAvatar(data []byte) error {
  2020. m, err := avatar.Prepare(data)
  2021. if err != nil {
  2022. return err
  2023. }
  2024. sess := x.NewSession()
  2025. defer sess.Close()
  2026. if err = sess.Begin(); err != nil {
  2027. return err
  2028. }
  2029. oldAvatarPath := repo.CustomAvatarPath()
  2030. // Users can upload the same image to other repo - prefix it with ID
  2031. // Then repo will be removed - only it avatar file will be removed
  2032. repo.Avatar = fmt.Sprintf("%d-%x", repo.ID, md5.Sum(data))
  2033. if _, err := sess.ID(repo.ID).Cols("avatar").Update(repo); err != nil {
  2034. return fmt.Errorf("UploadAvatar: Update repository avatar: %v", err)
  2035. }
  2036. if err := os.MkdirAll(setting.RepositoryAvatarUploadPath, os.ModePerm); err != nil {
  2037. return fmt.Errorf("UploadAvatar: Failed to create dir %s: %v", setting.RepositoryAvatarUploadPath, err)
  2038. }
  2039. fw, err := os.Create(repo.CustomAvatarPath())
  2040. if err != nil {
  2041. return fmt.Errorf("UploadAvatar: Create file: %v", err)
  2042. }
  2043. defer fw.Close()
  2044. if err = png.Encode(fw, *m); err != nil {
  2045. return fmt.Errorf("UploadAvatar: Encode png: %v", err)
  2046. }
  2047. if len(oldAvatarPath) > 0 && oldAvatarPath != repo.CustomAvatarPath() {
  2048. if err := os.Remove(oldAvatarPath); err != nil {
  2049. return fmt.Errorf("UploadAvatar: Failed to remove old repo avatar %s: %v", oldAvatarPath, err)
  2050. }
  2051. }
  2052. return sess.Commit()
  2053. }
  2054. // DeleteAvatar deletes the repos's custom avatar.
  2055. func (repo *Repository) DeleteAvatar() error {
  2056. // Avatar not exists
  2057. if len(repo.Avatar) == 0 {
  2058. return nil
  2059. }
  2060. avatarPath := repo.CustomAvatarPath()
  2061. log.Trace("DeleteAvatar[%d]: %s", repo.ID, avatarPath)
  2062. sess := x.NewSession()
  2063. defer sess.Close()
  2064. if err := sess.Begin(); err != nil {
  2065. return err
  2066. }
  2067. repo.Avatar = ""
  2068. if _, err := sess.ID(repo.ID).Cols("avatar").Update(repo); err != nil {
  2069. return fmt.Errorf("DeleteAvatar: Update repository avatar: %v", err)
  2070. }
  2071. if _, err := os.Stat(avatarPath); err == nil {
  2072. if err := os.Remove(avatarPath); err != nil {
  2073. return fmt.Errorf("DeleteAvatar: Failed to remove %s: %v", avatarPath, err)
  2074. }
  2075. } else {
  2076. // // Schrodinger: file may or may not exist. See err for details.
  2077. log.Trace("DeleteAvatar[%d]: %v", err)
  2078. }
  2079. return sess.Commit()
  2080. }
  2081. // GetOriginalURLHostname returns the hostname of a URL or the URL
  2082. func (repo *Repository) GetOriginalURLHostname() string {
  2083. u, err := url.Parse(repo.OriginalURL)
  2084. if err != nil {
  2085. return repo.OriginalURL
  2086. }
  2087. return u.Host
  2088. }
  2089. // GetTreePathLock returns LSF lock for the treePath
  2090. func (repo *Repository) GetTreePathLock(treePath string) (*LFSLock, error) {
  2091. if setting.LFS.StartServer {
  2092. locks, err := GetLFSLockByRepoID(repo.ID, 0, 0)
  2093. if err != nil {
  2094. return nil, err
  2095. }
  2096. for _, lock := range locks {
  2097. if lock.Path == treePath {
  2098. return lock, nil
  2099. }
  2100. }
  2101. }
  2102. return nil, nil
  2103. }
  2104. func updateRepositoryCols(e Engine, repo *Repository, cols ...string) error {
  2105. _, err := e.ID(repo.ID).Cols(cols...).Update(repo)
  2106. return err
  2107. }
  2108. // UpdateRepositoryCols updates repository's columns
  2109. func UpdateRepositoryCols(repo *Repository, cols ...string) error {
  2110. return updateRepositoryCols(x, repo, cols...)
  2111. }