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

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