1
0

repo.go 83 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772
  1. package database
  2. import (
  3. "bytes"
  4. "context"
  5. "fmt"
  6. "image"
  7. _ "image/jpeg"
  8. "image/png"
  9. "os"
  10. "os/exec"
  11. "path"
  12. "path/filepath"
  13. "sort"
  14. "strings"
  15. "time"
  16. "github.com/cockroachdb/errors"
  17. "github.com/nfnt/resize"
  18. "github.com/unknwon/cae/zip"
  19. "github.com/unknwon/com"
  20. "gopkg.in/ini.v1"
  21. "gorm.io/gorm"
  22. log "unknwon.dev/clog/v2"
  23. "github.com/gogs/git-module"
  24. api "github.com/gogs/go-gogs-client"
  25. embedConf "gogs.io/gogs/conf"
  26. "gogs.io/gogs/internal/avatar"
  27. "gogs.io/gogs/internal/conf"
  28. "gogs.io/gogs/internal/dbutil"
  29. "gogs.io/gogs/internal/errutil"
  30. "gogs.io/gogs/internal/markup"
  31. "gogs.io/gogs/internal/osutil"
  32. "gogs.io/gogs/internal/process"
  33. "gogs.io/gogs/internal/repoutil"
  34. "gogs.io/gogs/internal/semverutil"
  35. "gogs.io/gogs/internal/sync"
  36. )
  37. // RepoAvatarURLPrefix is used to identify a URL is to access repository avatar.
  38. const RepoAvatarURLPrefix = "repo-avatars"
  39. // InvalidRepoReference represents an error when repository reference is invalid.
  40. type InvalidRepoReference struct {
  41. Ref string
  42. }
  43. // IsInvalidRepoReference returns true if the error is InvalidRepoReference.
  44. func IsInvalidRepoReference(err error) bool {
  45. _, ok := err.(InvalidRepoReference)
  46. return ok
  47. }
  48. func (err InvalidRepoReference) Error() string {
  49. return fmt.Sprintf("invalid repository reference [ref: %s]", err.Ref)
  50. }
  51. var repoWorkingPool = sync.NewExclusivePool()
  52. var (
  53. Gitignores, Licenses, Readmes, LabelTemplates []string
  54. // Maximum items per page in forks, watchers and stars of a repo
  55. ItemsPerPage = 40
  56. )
  57. func LoadRepoConfig() {
  58. // Load .gitignore and license files and readme templates.
  59. types := []string{"gitignore", "license", "readme", "label"}
  60. typeFiles := make([][]string, 4)
  61. for i, t := range types {
  62. files, err := embedConf.FileNames(t)
  63. if err != nil {
  64. log.Fatal("Failed to get %q files: %v", t, err)
  65. }
  66. customPath := filepath.Join(conf.CustomDir(), "conf", t)
  67. if com.IsDir(customPath) {
  68. customFiles, err := com.StatDir(customPath)
  69. if err != nil {
  70. log.Fatal("Failed to get custom %s files: %v", t, err)
  71. }
  72. for _, f := range customFiles {
  73. if !com.IsSliceContainsStr(files, f) {
  74. files = append(files, f)
  75. }
  76. }
  77. }
  78. typeFiles[i] = files
  79. }
  80. Gitignores = typeFiles[0]
  81. Licenses = typeFiles[1]
  82. Readmes = typeFiles[2]
  83. LabelTemplates = typeFiles[3]
  84. sort.Strings(Gitignores)
  85. sort.Strings(Licenses)
  86. sort.Strings(Readmes)
  87. sort.Strings(LabelTemplates)
  88. // Filter out invalid names and promote preferred licenses.
  89. sortedLicenses := make([]string, 0, len(Licenses))
  90. for _, name := range conf.Repository.PreferredLicenses {
  91. if com.IsSliceContainsStr(Licenses, name) {
  92. sortedLicenses = append(sortedLicenses, name)
  93. }
  94. }
  95. for _, name := range Licenses {
  96. if !com.IsSliceContainsStr(conf.Repository.PreferredLicenses, name) {
  97. sortedLicenses = append(sortedLicenses, name)
  98. }
  99. }
  100. Licenses = sortedLicenses
  101. }
  102. func NewRepoContext() {
  103. zip.Verbose = false
  104. // Check Git installation.
  105. if _, err := exec.LookPath("git"); err != nil {
  106. log.Fatal("Failed to test 'git' command: %v (forgotten install?)", err)
  107. }
  108. // Check Git version.
  109. var err error
  110. conf.Git.Version, err = git.BinVersion()
  111. if err != nil {
  112. log.Fatal("Failed to get Git version: %v", err)
  113. }
  114. log.Trace("Git version: %s", conf.Git.Version)
  115. if semverutil.Compare(conf.Git.Version, "<", "1.8.3") {
  116. log.Fatal("Gogs requires Git version greater or equal to 1.8.3")
  117. }
  118. // Git requires setting user.name and user.email in order to commit changes.
  119. for configKey, defaultValue := range map[string]string{"user.name": "Gogs", "user.email": "gogs@fake.local"} {
  120. if stdout, stderr, err := process.Exec("NewRepoContext(get setting)", "git", "config", "--get", configKey); err != nil || strings.TrimSpace(stdout) == "" {
  121. // ExitError indicates this config is not set
  122. if _, ok := err.(*exec.ExitError); ok || strings.TrimSpace(stdout) == "" {
  123. if _, stderr, gerr := process.Exec("NewRepoContext(set "+configKey+")", "git", "config", "--global", configKey, defaultValue); gerr != nil {
  124. log.Fatal("Failed to set git %s(%s): %s", configKey, gerr, stderr)
  125. }
  126. log.Info("Git config %s set to %s", configKey, defaultValue)
  127. } else {
  128. log.Fatal("Failed to get git %s(%s): %s", configKey, err, stderr)
  129. }
  130. }
  131. }
  132. // Set git some configurations.
  133. if _, stderr, err := process.Exec("NewRepoContext(git config --global core.quotepath false)",
  134. "git", "config", "--global", "core.quotepath", "false"); err != nil {
  135. log.Fatal("Failed to execute 'git config --global core.quotepath false': %v - %s", err, stderr)
  136. }
  137. RemoveAllWithNotice("Clean up repository temporary data", filepath.Join(conf.Server.AppDataPath, "tmp"))
  138. }
  139. // Repository contains information of a repository.
  140. type Repository struct {
  141. ID int64 `gorm:"primaryKey"`
  142. OwnerID int64 `xorm:"UNIQUE(s)" gorm:"uniqueIndex:repo_owner_name_unique"`
  143. Owner *User `xorm:"-" gorm:"-" json:"-"`
  144. LowerName string `xorm:"UNIQUE(s) INDEX NOT NULL" gorm:"uniqueIndex:repo_owner_name_unique;index;not null"`
  145. Name string `xorm:"INDEX NOT NULL" gorm:"index;not null"`
  146. Description string `xorm:"VARCHAR(512)" gorm:"type:VARCHAR(512)"`
  147. Website string
  148. DefaultBranch string
  149. Size int64 `xorm:"NOT NULL DEFAULT 0" gorm:"not null;default:0"`
  150. UseCustomAvatar bool
  151. // Counters
  152. NumWatches int
  153. NumStars int
  154. NumForks int
  155. NumIssues int
  156. NumClosedIssues int
  157. NumOpenIssues int `xorm:"-" gorm:"-" json:"-"`
  158. NumPulls int
  159. NumClosedPulls int
  160. NumOpenPulls int `xorm:"-" gorm:"-" json:"-"`
  161. NumMilestones int `xorm:"NOT NULL DEFAULT 0" gorm:"not null;default:0"`
  162. NumClosedMilestones int `xorm:"NOT NULL DEFAULT 0" gorm:"not null;default:0"`
  163. NumOpenMilestones int `xorm:"-" gorm:"-" json:"-"`
  164. NumTags int `xorm:"-" gorm:"-" json:"-"`
  165. IsPrivate bool
  166. // TODO: When migrate to GORM, make sure to do a loose migration with `HasColumn` and `AddColumn`,
  167. // see docs in https://gorm.io/docs/migration.html.
  168. IsUnlisted bool `xorm:"NOT NULL DEFAULT false" gorm:"not null;default:FALSE"`
  169. IsBare bool
  170. IsMirror bool
  171. *Mirror `xorm:"-" gorm:"-" json:"-"`
  172. // Advanced settings
  173. EnableWiki bool `xorm:"NOT NULL DEFAULT true" gorm:"not null;default:TRUE"`
  174. AllowPublicWiki bool
  175. EnableExternalWiki bool
  176. ExternalWikiURL string
  177. EnableIssues bool `xorm:"NOT NULL DEFAULT true" gorm:"not null;default:TRUE"`
  178. AllowPublicIssues bool
  179. EnableExternalTracker bool
  180. ExternalTrackerURL string
  181. ExternalTrackerFormat string
  182. ExternalTrackerStyle string
  183. ExternalMetas map[string]string `xorm:"-" gorm:"-" json:"-"`
  184. EnablePulls bool `xorm:"NOT NULL DEFAULT true" gorm:"not null;default:TRUE"`
  185. PullsIgnoreWhitespace bool `xorm:"NOT NULL DEFAULT false" gorm:"not null;default:FALSE"`
  186. PullsAllowRebase bool `xorm:"NOT NULL DEFAULT false" gorm:"not null;default:FALSE"`
  187. IsFork bool `xorm:"NOT NULL DEFAULT false" gorm:"not null;default:FALSE"`
  188. ForkID int64
  189. BaseRepo *Repository `xorm:"-" gorm:"-" json:"-"`
  190. Created time.Time `xorm:"-" gorm:"-" json:"-"`
  191. CreatedUnix int64
  192. Updated time.Time `xorm:"-" gorm:"-" json:"-"`
  193. UpdatedUnix int64
  194. }
  195. func (r *Repository) BeforeInsert() {
  196. r.CreatedUnix = time.Now().Unix()
  197. r.UpdatedUnix = r.CreatedUnix
  198. }
  199. func (r *Repository) AfterFind(_ *gorm.DB) error {
  200. // FIXME: use db migration to solve all at once.
  201. if r.DefaultBranch == "" {
  202. r.DefaultBranch = conf.Repository.DefaultBranch
  203. }
  204. r.NumOpenIssues = r.NumIssues - r.NumClosedIssues
  205. r.NumOpenPulls = r.NumPulls - r.NumClosedPulls
  206. r.NumOpenMilestones = r.NumMilestones - r.NumClosedMilestones
  207. if r.ExternalTrackerStyle == "" {
  208. r.ExternalTrackerStyle = markup.IssueNameStyleNumeric
  209. }
  210. r.Created = time.Unix(r.CreatedUnix, 0).Local()
  211. r.Updated = time.Unix(r.UpdatedUnix, 0)
  212. return nil
  213. }
  214. func (r *Repository) loadAttributes(e *gorm.DB) (err error) {
  215. if r.Owner == nil {
  216. r.Owner, err = getUserByID(e, r.OwnerID)
  217. if err != nil {
  218. return errors.Newf("getUserByID [%d]: %v", r.OwnerID, err)
  219. }
  220. }
  221. if r.IsFork && r.BaseRepo == nil {
  222. r.BaseRepo, err = getRepositoryByID(e, r.ForkID)
  223. if err != nil {
  224. if IsErrRepoNotExist(err) {
  225. r.IsFork = false
  226. r.ForkID = 0
  227. } else {
  228. return errors.Newf("get fork repository by ID: %v", err)
  229. }
  230. }
  231. }
  232. return nil
  233. }
  234. func (r *Repository) LoadAttributes() error {
  235. return r.loadAttributes(db)
  236. }
  237. // IsPartialPublic returns true if repository is public or allow public access to wiki or issues.
  238. func (r *Repository) IsPartialPublic() bool {
  239. return !r.IsPrivate || r.AllowPublicWiki || r.AllowPublicIssues
  240. }
  241. func (r *Repository) CanGuestViewWiki() bool {
  242. return r.EnableWiki && !r.EnableExternalWiki && r.AllowPublicWiki
  243. }
  244. func (r *Repository) CanGuestViewIssues() bool {
  245. return r.EnableIssues && !r.EnableExternalTracker && r.AllowPublicIssues
  246. }
  247. // MustOwner always returns a valid *User object to avoid conceptually impossible error handling.
  248. // It creates a fake object that contains error details when error occurs.
  249. func (r *Repository) MustOwner() *User {
  250. return r.mustOwner(db)
  251. }
  252. func (r *Repository) FullName() string {
  253. return r.MustOwner().Name + "/" + r.Name
  254. }
  255. // Deprecated: Use repoutil.HTMLURL instead.
  256. func (r *Repository) HTMLURL() string {
  257. return conf.Server.ExternalURL + r.FullName()
  258. }
  259. // CustomAvatarPath returns repository custom avatar file path.
  260. func (r *Repository) CustomAvatarPath() string {
  261. return filepath.Join(conf.Picture.RepositoryAvatarUploadPath, com.ToStr(r.ID))
  262. }
  263. // RelAvatarLink returns relative avatar link to the site domain,
  264. // which includes app sub-url as prefix.
  265. // Since Gravatar support not needed here - just check for image path.
  266. func (r *Repository) RelAvatarLink() string {
  267. defaultImgURL := ""
  268. if !com.IsExist(r.CustomAvatarPath()) {
  269. return defaultImgURL
  270. }
  271. return fmt.Sprintf("%s/%s/%d", conf.Server.Subpath, RepoAvatarURLPrefix, r.ID)
  272. }
  273. // AvatarLink returns repository avatar absolute link.
  274. func (r *Repository) AvatarLink() string {
  275. link := r.RelAvatarLink()
  276. if link[0] == '/' && link[1] != '/' {
  277. return conf.Server.ExternalURL + strings.TrimPrefix(link, conf.Server.Subpath)[1:]
  278. }
  279. return link
  280. }
  281. // UploadAvatar saves custom avatar for repository.
  282. // FIXME: split uploads to different subdirs in case we have massive number of repositories.
  283. func (r *Repository) UploadAvatar(data []byte) error {
  284. img, _, err := image.Decode(bytes.NewReader(data))
  285. if err != nil {
  286. return errors.Newf("decode image: %v", err)
  287. }
  288. _ = os.MkdirAll(conf.Picture.RepositoryAvatarUploadPath, os.ModePerm)
  289. fw, err := os.Create(r.CustomAvatarPath())
  290. if err != nil {
  291. return errors.Newf("create custom avatar directory: %v", err)
  292. }
  293. defer fw.Close()
  294. m := resize.Resize(avatar.DefaultSize, avatar.DefaultSize, img, resize.NearestNeighbor)
  295. if err = png.Encode(fw, m); err != nil {
  296. return errors.Newf("encode image: %v", err)
  297. }
  298. return nil
  299. }
  300. // DeleteAvatar deletes the repository custom avatar.
  301. func (r *Repository) DeleteAvatar() error {
  302. log.Trace("DeleteAvatar [%d]: %s", r.ID, r.CustomAvatarPath())
  303. if err := os.Remove(r.CustomAvatarPath()); err != nil {
  304. return err
  305. }
  306. r.UseCustomAvatar = false
  307. return UpdateRepository(r, false)
  308. }
  309. // This method assumes following fields have been assigned with valid values:
  310. // Required - BaseRepo (if fork)
  311. // Arguments that are allowed to be nil: permission
  312. //
  313. // Deprecated: Use APIFormat instead.
  314. func (r *Repository) APIFormatLegacy(permission *api.Permission, user ...*User) *api.Repository {
  315. cloneLink := r.CloneLink()
  316. apiRepo := &api.Repository{
  317. ID: r.ID,
  318. Owner: r.Owner.APIFormat(),
  319. Name: r.Name,
  320. FullName: r.FullName(),
  321. Description: r.Description,
  322. Private: r.IsPrivate,
  323. Fork: r.IsFork,
  324. Empty: r.IsBare,
  325. Mirror: r.IsMirror,
  326. Size: r.Size,
  327. HTMLURL: r.HTMLURL(),
  328. SSHURL: cloneLink.SSH,
  329. CloneURL: cloneLink.HTTPS,
  330. Website: r.Website,
  331. Stars: r.NumStars,
  332. Forks: r.NumForks,
  333. Watchers: r.NumWatches,
  334. OpenIssues: r.NumOpenIssues,
  335. DefaultBranch: r.DefaultBranch,
  336. Created: r.Created,
  337. Updated: r.Updated,
  338. Permissions: permission,
  339. // Reserved for go-gogs-client change
  340. // AvatarUrl: r.AvatarLink(),
  341. }
  342. if r.IsFork {
  343. p := &api.Permission{Pull: true}
  344. if len(user) != 0 {
  345. accessMode := Handle.Permissions().AccessMode(
  346. context.TODO(),
  347. user[0].ID,
  348. r.ID,
  349. AccessModeOptions{
  350. OwnerID: r.OwnerID,
  351. Private: r.IsPrivate,
  352. },
  353. )
  354. p.Admin = accessMode >= AccessModeAdmin
  355. p.Push = accessMode >= AccessModeWrite
  356. }
  357. apiRepo.Parent = r.BaseRepo.APIFormatLegacy(p)
  358. }
  359. return apiRepo
  360. }
  361. func (r *Repository) getOwner(e *gorm.DB) (err error) {
  362. if r.Owner != nil {
  363. return nil
  364. }
  365. r.Owner, err = getUserByID(e, r.OwnerID)
  366. return err
  367. }
  368. func (r *Repository) GetOwner() error {
  369. return r.getOwner(db)
  370. }
  371. func (r *Repository) mustOwner(e *gorm.DB) *User {
  372. if err := r.getOwner(e); err != nil {
  373. return &User{
  374. Name: "error",
  375. FullName: err.Error(),
  376. }
  377. }
  378. return r.Owner
  379. }
  380. func (r *Repository) UpdateSize() error {
  381. countObject, err := git.CountObjects(r.RepoPath())
  382. if err != nil {
  383. return errors.Newf("count repository objects: %v", err)
  384. }
  385. r.Size = countObject.Size + countObject.SizePack
  386. if err = db.Model(&Repository{}).Where("id = ?", r.ID).Update("size", r.Size).Error; err != nil {
  387. return errors.Newf("update size: %v", err)
  388. }
  389. return nil
  390. }
  391. // ComposeMetas composes a map of metas for rendering SHA1 URL and external issue tracker URL.
  392. func (r *Repository) ComposeMetas() map[string]string {
  393. if r.ExternalMetas != nil {
  394. return r.ExternalMetas
  395. }
  396. r.ExternalMetas = map[string]string{
  397. "repoLink": r.Link(),
  398. }
  399. if r.EnableExternalTracker {
  400. r.ExternalMetas["user"] = r.MustOwner().Name
  401. r.ExternalMetas["repo"] = r.Name
  402. r.ExternalMetas["format"] = r.ExternalTrackerFormat
  403. switch r.ExternalTrackerStyle {
  404. case markup.IssueNameStyleAlphanumeric:
  405. r.ExternalMetas["style"] = markup.IssueNameStyleAlphanumeric
  406. default:
  407. r.ExternalMetas["style"] = markup.IssueNameStyleNumeric
  408. }
  409. }
  410. return r.ExternalMetas
  411. }
  412. // DeleteWiki removes the actual and local copy of repository wiki.
  413. func (r *Repository) DeleteWiki() {
  414. wikiPaths := []string{r.WikiPath(), r.LocalWikiPath()}
  415. for _, wikiPath := range wikiPaths {
  416. RemoveAllWithNotice("Delete repository wiki", wikiPath)
  417. }
  418. }
  419. // getUsersWithAccesMode returns users that have at least given access mode to the repository.
  420. func (r *Repository) getUsersWithAccesMode(e *gorm.DB, mode AccessMode) (_ []*User, err error) {
  421. if err = r.getOwner(e); err != nil {
  422. return nil, err
  423. }
  424. accesses := make([]*Access, 0, 10)
  425. if err = e.Where("repo_id = ? AND mode >= ?", r.ID, mode).Find(&accesses).Error; err != nil {
  426. return nil, err
  427. }
  428. // Leave a seat for owner itself to append later, but if owner is an organization
  429. // and just waste 1 unit is cheaper than re-allocate memory once.
  430. users := make([]*User, 0, len(accesses)+1)
  431. if len(accesses) > 0 {
  432. userIDs := make([]int64, len(accesses))
  433. for i := 0; i < len(accesses); i++ {
  434. userIDs[i] = accesses[i].UserID
  435. }
  436. if err = e.Where("id IN ?", userIDs).Find(&users).Error; err != nil {
  437. return nil, err
  438. }
  439. // TODO(unknwon): Rely on AfterFind hook to sanitize user full name.
  440. for _, u := range users {
  441. u.FullName = markup.Sanitize(u.FullName)
  442. }
  443. }
  444. if !r.Owner.IsOrganization() {
  445. users = append(users, r.Owner)
  446. }
  447. return users, nil
  448. }
  449. // getAssignees returns a list of users who can be assigned to issues in this repository.
  450. func (r *Repository) getAssignees(e *gorm.DB) (_ []*User, err error) {
  451. return r.getUsersWithAccesMode(e, AccessModeRead)
  452. }
  453. // GetAssignees returns all users that have read access and can be assigned to issues
  454. // of the repository,
  455. func (r *Repository) GetAssignees() (_ []*User, err error) {
  456. return r.getAssignees(db)
  457. }
  458. // GetAssigneeByID returns the user that has write access of repository by given ID.
  459. func (r *Repository) GetAssigneeByID(userID int64) (*User, error) {
  460. ctx := context.TODO()
  461. if !Handle.Permissions().Authorize(
  462. ctx,
  463. userID,
  464. r.ID,
  465. AccessModeRead,
  466. AccessModeOptions{
  467. OwnerID: r.OwnerID,
  468. Private: r.IsPrivate,
  469. },
  470. ) {
  471. return nil, ErrUserNotExist{args: errutil.Args{"userID": userID}}
  472. }
  473. return Handle.Users().GetByID(ctx, userID)
  474. }
  475. // GetWriters returns all users that have write access to the repository.
  476. func (r *Repository) GetWriters() (_ []*User, err error) {
  477. return r.getUsersWithAccesMode(db, AccessModeWrite)
  478. }
  479. // GetMilestoneByID returns the milestone belongs to repository by given ID.
  480. func (r *Repository) GetMilestoneByID(milestoneID int64) (*Milestone, error) {
  481. return GetMilestoneByRepoID(r.ID, milestoneID)
  482. }
  483. // IssueStats returns number of open and closed repository issues by given filter mode.
  484. func (r *Repository) IssueStats(userID int64, filterMode FilterMode, isPull bool) (int64, int64) {
  485. return GetRepoIssueStats(r.ID, userID, filterMode, isPull)
  486. }
  487. func (r *Repository) GetMirror() (err error) {
  488. r.Mirror, err = GetMirrorByRepoID(r.ID)
  489. return err
  490. }
  491. func (r *Repository) repoPath(e *gorm.DB) string {
  492. return RepoPath(r.mustOwner(e).Name, r.Name)
  493. }
  494. // Deprecated: Use repoutil.RepositoryPath instead.
  495. func (r *Repository) RepoPath() string {
  496. return r.repoPath(db)
  497. }
  498. func (r *Repository) GitConfigPath() string {
  499. return filepath.Join(r.RepoPath(), "config")
  500. }
  501. func (r *Repository) RelLink() string {
  502. return "/" + r.FullName()
  503. }
  504. func (r *Repository) Link() string {
  505. return conf.Server.Subpath + "/" + r.FullName()
  506. }
  507. // Deprecated: Use repoutil.ComparePath instead.
  508. func (r *Repository) ComposeCompareURL(oldCommitID, newCommitID string) string {
  509. return fmt.Sprintf("%s/%s/compare/%s...%s", r.MustOwner().Name, r.Name, oldCommitID, newCommitID)
  510. }
  511. func (r *Repository) HasAccess(userID int64) bool {
  512. return Handle.Permissions().Authorize(context.TODO(), userID, r.ID, AccessModeRead,
  513. AccessModeOptions{
  514. OwnerID: r.OwnerID,
  515. Private: r.IsPrivate,
  516. },
  517. )
  518. }
  519. func (r *Repository) IsOwnedBy(userID int64) bool {
  520. return r.OwnerID == userID
  521. }
  522. // CanBeForked returns true if repository meets the requirements of being forked.
  523. func (r *Repository) CanBeForked() bool {
  524. return !r.IsBare
  525. }
  526. // CanEnablePulls returns true if repository meets the requirements of accepting pulls.
  527. func (r *Repository) CanEnablePulls() bool {
  528. return !r.IsMirror && !r.IsBare
  529. }
  530. // AllowPulls returns true if repository meets the requirements of accepting pulls and has them enabled.
  531. func (r *Repository) AllowsPulls() bool {
  532. return r.CanEnablePulls() && r.EnablePulls
  533. }
  534. func (r *Repository) IsBranchRequirePullRequest(name string) bool {
  535. return IsBranchOfRepoRequirePullRequest(r.ID, name)
  536. }
  537. // CanEnableEditor returns true if repository meets the requirements of web editor.
  538. func (r *Repository) CanEnableEditor() bool {
  539. return !r.IsMirror
  540. }
  541. // FIXME: should have a mutex to prevent producing same index for two issues that are created
  542. // closely enough.
  543. func (r *Repository) NextIssueIndex() int64 {
  544. return int64(r.NumIssues+r.NumPulls) + 1
  545. }
  546. func (r *Repository) LocalCopyPath() string {
  547. return filepath.Join(conf.Server.AppDataPath, "tmp", "local-r", com.ToStr(r.ID))
  548. }
  549. // UpdateLocalCopy fetches latest changes of given branch from repoPath to localPath.
  550. // It creates a new clone if local copy does not exist, but does not checks out to a
  551. // specific branch if the local copy belongs to a wiki.
  552. // For existing local copy, it checks out to target branch by default, and safe to
  553. // assume subsequent operations are against target branch when caller has confidence
  554. // about no race condition.
  555. func UpdateLocalCopyBranch(repoPath, localPath, branch string, isWiki bool) (err error) {
  556. if !osutil.Exist(localPath) {
  557. // Checkout to a specific branch fails when wiki is an empty repository.
  558. if isWiki {
  559. branch = ""
  560. }
  561. if err = git.Clone(repoPath, localPath, git.CloneOptions{
  562. Branch: branch,
  563. Timeout: time.Duration(conf.Git.Timeout.Clone) * time.Second,
  564. }); err != nil {
  565. return errors.Newf("git clone [branch: %s]: %v", branch, err)
  566. }
  567. return nil
  568. }
  569. gitRepo, err := git.Open(localPath)
  570. if err != nil {
  571. return errors.Newf("open repository: %v", err)
  572. }
  573. if err = gitRepo.Fetch(git.FetchOptions{
  574. Prune: true,
  575. }); err != nil {
  576. return errors.Newf("fetch: %v", err)
  577. }
  578. if err = gitRepo.Checkout(branch); err != nil {
  579. return errors.Newf("checkout [branch: %s]: %v", branch, err)
  580. }
  581. // Reset to align with remote in case of force push.
  582. rev := "origin/" + branch
  583. if err = gitRepo.Reset(rev, git.ResetOptions{
  584. Hard: true,
  585. }); err != nil {
  586. return errors.Newf("reset [revision: %s]: %v", rev, err)
  587. }
  588. return nil
  589. }
  590. // UpdateLocalCopyBranch makes sure local copy of repository in given branch is up-to-date.
  591. func (r *Repository) UpdateLocalCopyBranch(branch string) error {
  592. return UpdateLocalCopyBranch(r.RepoPath(), r.LocalCopyPath(), branch, false)
  593. }
  594. // PatchPath returns corresponding patch file path of repository by given issue ID.
  595. func (r *Repository) PatchPath(index int64) (string, error) {
  596. if err := r.GetOwner(); err != nil {
  597. return "", err
  598. }
  599. return filepath.Join(RepoPath(r.Owner.Name, r.Name), "pulls", com.ToStr(index)+".patch"), nil
  600. }
  601. // SavePatch saves patch data to corresponding location by given issue ID.
  602. func (r *Repository) SavePatch(index int64, patch []byte) error {
  603. patchPath, err := r.PatchPath(index)
  604. if err != nil {
  605. return errors.Newf("PatchPath: %v", err)
  606. }
  607. if err = os.MkdirAll(filepath.Dir(patchPath), os.ModePerm); err != nil {
  608. return err
  609. }
  610. if err = os.WriteFile(patchPath, patch, 0o644); err != nil {
  611. return errors.Newf("WriteFile: %v", err)
  612. }
  613. return nil
  614. }
  615. func isRepositoryExist(e *gorm.DB, u *User, repoName string) (bool, error) {
  616. var repo Repository
  617. err := e.Where("owner_id = ? AND lower_name = ?", u.ID, strings.ToLower(repoName)).First(&repo).Error
  618. if err != nil {
  619. if err == gorm.ErrRecordNotFound {
  620. return false, nil
  621. }
  622. return false, err
  623. }
  624. return com.IsDir(RepoPath(u.Name, repoName)), nil
  625. }
  626. // IsRepositoryExist returns true if the repository with given name under user has already existed.
  627. func IsRepositoryExist(u *User, repoName string) (bool, error) {
  628. return isRepositoryExist(db, u, repoName)
  629. }
  630. // Deprecated: Use repoutil.NewCloneLink instead.
  631. func (r *Repository) cloneLink(isWiki bool) *repoutil.CloneLink {
  632. repoName := r.Name
  633. if isWiki {
  634. repoName += ".wiki"
  635. }
  636. r.Owner = r.MustOwner()
  637. cl := new(repoutil.CloneLink)
  638. if conf.SSH.Port != 22 {
  639. cl.SSH = fmt.Sprintf("ssh://%s@%s:%d/%s/%s.git", conf.App.RunUser, conf.SSH.Domain, conf.SSH.Port, r.Owner.Name, repoName)
  640. } else {
  641. cl.SSH = fmt.Sprintf("%s@%s:%s/%s.git", conf.App.RunUser, conf.SSH.Domain, r.Owner.Name, repoName)
  642. }
  643. cl.HTTPS = repoutil.HTTPSCloneURL(r.Owner.Name, repoName)
  644. return cl
  645. }
  646. // CloneLink returns clone URLs of repository.
  647. //
  648. // Deprecated: Use repoutil.NewCloneLink instead.
  649. func (r *Repository) CloneLink() (cl *repoutil.CloneLink) {
  650. return r.cloneLink(false)
  651. }
  652. type MigrateRepoOptions struct {
  653. Name string
  654. Description string
  655. IsPrivate bool
  656. IsUnlisted bool
  657. IsMirror bool
  658. RemoteAddr string
  659. }
  660. /*
  661. - GitHub, GitLab, Gogs: *.wiki.git
  662. - BitBucket: *.git/wiki
  663. */
  664. var commonWikiURLSuffixes = []string{".wiki.git", ".git/wiki"}
  665. // wikiRemoteURL returns accessible repository URL for wiki if exists.
  666. // Otherwise, it returns an empty string.
  667. func wikiRemoteURL(remote string) string {
  668. remote = strings.TrimSuffix(remote, ".git")
  669. for _, suffix := range commonWikiURLSuffixes {
  670. wikiURL := remote + suffix
  671. if git.IsURLAccessible(time.Minute, wikiURL) {
  672. return wikiURL
  673. }
  674. }
  675. return ""
  676. }
  677. // MigrateRepository migrates a existing repository from other project hosting.
  678. func MigrateRepository(doer, owner *User, opts MigrateRepoOptions) (*Repository, error) {
  679. repo, err := CreateRepository(doer, owner, CreateRepoOptionsLegacy{
  680. Name: opts.Name,
  681. Description: opts.Description,
  682. IsPrivate: opts.IsPrivate,
  683. IsUnlisted: opts.IsUnlisted,
  684. IsMirror: opts.IsMirror,
  685. })
  686. if err != nil {
  687. return nil, err
  688. }
  689. repoPath := RepoPath(owner.Name, opts.Name)
  690. wikiPath := WikiPath(owner.Name, opts.Name)
  691. if owner.IsOrganization() {
  692. t, err := owner.GetOwnerTeam()
  693. if err != nil {
  694. return nil, err
  695. }
  696. repo.NumWatches = t.NumMembers
  697. } else {
  698. repo.NumWatches = 1
  699. }
  700. migrateTimeout := time.Duration(conf.Git.Timeout.Migrate) * time.Second
  701. RemoveAllWithNotice("Repository path erase before creation", repoPath)
  702. if err = git.Clone(opts.RemoteAddr, repoPath, git.CloneOptions{
  703. Mirror: true,
  704. Quiet: true,
  705. Timeout: migrateTimeout,
  706. }); err != nil {
  707. return repo, errors.Newf("clone: %v", err)
  708. }
  709. wikiRemotePath := wikiRemoteURL(opts.RemoteAddr)
  710. if len(wikiRemotePath) > 0 {
  711. RemoveAllWithNotice("Repository wiki path erase before creation", wikiPath)
  712. if err = git.Clone(wikiRemotePath, wikiPath, git.CloneOptions{
  713. Mirror: true,
  714. Quiet: true,
  715. Timeout: migrateTimeout,
  716. }); err != nil {
  717. log.Error("Failed to clone wiki: %v", err)
  718. RemoveAllWithNotice("Delete repository wiki for initialization failure", wikiPath)
  719. }
  720. }
  721. // Check if repository is empty.
  722. _, stderr, err := com.ExecCmdDir(repoPath, "git", "log", "-1")
  723. if err != nil {
  724. if strings.Contains(stderr, "fatal: bad default revision 'HEAD'") {
  725. repo.IsBare = true
  726. } else {
  727. return repo, errors.Newf("check bare: %v - %s", err, stderr)
  728. }
  729. }
  730. if !repo.IsBare {
  731. // Try to get HEAD branch and set it as default branch.
  732. gitRepo, err := git.Open(repoPath)
  733. if err != nil {
  734. return repo, errors.Newf("open repository: %v", err)
  735. }
  736. refspec, err := gitRepo.SymbolicRef()
  737. if err != nil {
  738. return repo, errors.Newf("get HEAD branch: %v", err)
  739. }
  740. repo.DefaultBranch = git.RefShortName(refspec)
  741. if err = repo.UpdateSize(); err != nil {
  742. log.Error("UpdateSize [repo_id: %d]: %v", repo.ID, err)
  743. }
  744. }
  745. if opts.IsMirror {
  746. if err = db.Create(&Mirror{
  747. RepoID: repo.ID,
  748. Interval: conf.Mirror.DefaultInterval,
  749. EnablePrune: true,
  750. NextSync: time.Now().Add(time.Duration(conf.Mirror.DefaultInterval) * time.Hour),
  751. }).Error; err != nil {
  752. return repo, errors.Newf("Create: %v", err)
  753. }
  754. repo.IsMirror = true
  755. return repo, UpdateRepository(repo, false)
  756. }
  757. return CleanUpMigrateInfo(repo)
  758. }
  759. // cleanUpMigrateGitConfig removes mirror info which prevents "push --all".
  760. // This also removes possible user credentials.
  761. func cleanUpMigrateGitConfig(configPath string) error {
  762. cfg, err := ini.Load(configPath)
  763. if err != nil {
  764. return errors.Newf("open config file: %v", err)
  765. }
  766. cfg.DeleteSection("remote \"origin\"")
  767. if err = cfg.SaveToIndent(configPath, "\t"); err != nil {
  768. return errors.Newf("save config file: %v", err)
  769. }
  770. return nil
  771. }
  772. var hooksTpls = map[git.HookName]string{
  773. "pre-receive": "#!/usr/bin/env %s\n\"%s\" hook --config='%s' pre-receive\n",
  774. "update": "#!/usr/bin/env %s\n\"%s\" hook --config='%s' update $1 $2 $3\n",
  775. "post-receive": "#!/usr/bin/env %s\n\"%s\" hook --config='%s' post-receive\n",
  776. }
  777. func createDelegateHooks(repoPath string) (err error) {
  778. for _, name := range git.ServerSideHooks {
  779. hookPath := filepath.Join(repoPath, "hooks", string(name))
  780. if err = os.WriteFile(hookPath,
  781. []byte(fmt.Sprintf(hooksTpls[name], conf.Repository.ScriptType, conf.AppPath(), conf.CustomConf)),
  782. os.ModePerm); err != nil {
  783. return errors.Newf("create delegate hook '%s': %v", hookPath, err)
  784. }
  785. }
  786. return nil
  787. }
  788. // Finish migrating repository and/or wiki with things that don't need to be done for mirrors.
  789. func CleanUpMigrateInfo(repo *Repository) (*Repository, error) {
  790. repoPath := repo.RepoPath()
  791. if err := createDelegateHooks(repoPath); err != nil {
  792. return repo, errors.Newf("createDelegateHooks: %v", err)
  793. }
  794. if repo.HasWiki() {
  795. if err := createDelegateHooks(repo.WikiPath()); err != nil {
  796. return repo, errors.Newf("createDelegateHooks.(wiki): %v", err)
  797. }
  798. }
  799. if err := cleanUpMigrateGitConfig(repo.GitConfigPath()); err != nil {
  800. return repo, errors.Newf("cleanUpMigrateGitConfig: %v", err)
  801. }
  802. if repo.HasWiki() {
  803. if err := cleanUpMigrateGitConfig(path.Join(repo.WikiPath(), "config")); err != nil {
  804. return repo, errors.Newf("cleanUpMigrateGitConfig.(wiki): %v", err)
  805. }
  806. }
  807. return repo, UpdateRepository(repo, false)
  808. }
  809. // initRepoCommit temporarily changes with work directory.
  810. func initRepoCommit(tmpPath string, sig *git.Signature) (err error) {
  811. var stderr string
  812. if _, stderr, err = process.ExecDir(-1,
  813. tmpPath, fmt.Sprintf("initRepoCommit (git add): %s", tmpPath),
  814. "git", "add", "--all"); err != nil {
  815. return errors.Newf("git add: %s", stderr)
  816. }
  817. if _, stderr, err = process.ExecDir(-1,
  818. tmpPath, fmt.Sprintf("initRepoCommit (git commit): %s", tmpPath),
  819. "git", "commit", fmt.Sprintf("--author='%s <%s>'", sig.Name, sig.Email),
  820. "-m", "Initial commit"); err != nil {
  821. return errors.Newf("git commit: %s", stderr)
  822. }
  823. if _, stderr, err = process.ExecDir(-1,
  824. tmpPath, fmt.Sprintf("initRepoCommit (git push): %s", tmpPath),
  825. "git", "push"); err != nil {
  826. return errors.Newf("git push: %s", stderr)
  827. }
  828. return nil
  829. }
  830. type CreateRepoOptionsLegacy struct {
  831. Name string
  832. Description string
  833. Gitignores string
  834. License string
  835. Readme string
  836. IsPrivate bool
  837. IsUnlisted bool
  838. IsMirror bool
  839. AutoInit bool
  840. }
  841. func getRepoInitFile(tp, name string) ([]byte, error) {
  842. relPath := path.Join(tp, strings.TrimLeft(path.Clean("/"+name), "/"))
  843. // Use custom file when available.
  844. customPath := filepath.Join(conf.CustomDir(), "conf", relPath)
  845. if osutil.IsFile(customPath) {
  846. return os.ReadFile(customPath)
  847. }
  848. return embedConf.Files.ReadFile(relPath)
  849. }
  850. func prepareRepoCommit(repo *Repository, tmpDir, repoPath string, opts CreateRepoOptionsLegacy) error {
  851. // Clone to temporary path and do the init commit.
  852. err := git.Clone(repoPath, tmpDir, git.CloneOptions{})
  853. if err != nil {
  854. return errors.Wrap(err, "clone")
  855. }
  856. // README
  857. data, err := getRepoInitFile("readme", opts.Readme)
  858. if err != nil {
  859. return errors.Newf("getRepoInitFile[%s]: %v", opts.Readme, err)
  860. }
  861. cloneLink := repo.CloneLink()
  862. match := map[string]string{
  863. "Name": repo.Name,
  864. "Description": repo.Description,
  865. "CloneURL.SSH": cloneLink.SSH,
  866. "CloneURL.HTTPS": cloneLink.HTTPS,
  867. }
  868. if err = os.WriteFile(filepath.Join(tmpDir, "README.md"),
  869. []byte(com.Expand(string(data), match)), 0o644); err != nil {
  870. return errors.Newf("write README.md: %v", err)
  871. }
  872. // .gitignore
  873. if len(opts.Gitignores) > 0 {
  874. var buf bytes.Buffer
  875. names := strings.Split(opts.Gitignores, ",")
  876. for _, name := range names {
  877. data, err = getRepoInitFile("gitignore", name)
  878. if err != nil {
  879. return errors.Newf("getRepoInitFile[%s]: %v", name, err)
  880. }
  881. buf.WriteString("# ---> " + name + "\n")
  882. buf.Write(data)
  883. buf.WriteString("\n")
  884. }
  885. if buf.Len() > 0 {
  886. if err = os.WriteFile(filepath.Join(tmpDir, ".gitignore"), buf.Bytes(), 0o644); err != nil {
  887. return errors.Newf("write .gitignore: %v", err)
  888. }
  889. }
  890. }
  891. // LICENSE
  892. if len(opts.License) > 0 {
  893. data, err = getRepoInitFile("license", opts.License)
  894. if err != nil {
  895. return errors.Newf("getRepoInitFile[%s]: %v", opts.License, err)
  896. }
  897. if err = os.WriteFile(filepath.Join(tmpDir, "LICENSE"), data, 0o644); err != nil {
  898. return errors.Newf("write LICENSE: %v", err)
  899. }
  900. }
  901. return nil
  902. }
  903. // initRepository performs initial commit with chosen setup files on behave of doer.
  904. func initRepository(e *gorm.DB, repoPath string, doer *User, repo *Repository, opts CreateRepoOptionsLegacy) (err error) {
  905. // Init bare new repository.
  906. if err = git.Init(repoPath, git.InitOptions{Bare: true}); err != nil {
  907. return errors.Newf("init repository: %v", err)
  908. } else if err = createDelegateHooks(repoPath); err != nil {
  909. return errors.Newf("createDelegateHooks: %v", err)
  910. }
  911. // Set default branch
  912. _, err = git.SymbolicRef(
  913. repoPath,
  914. git.SymbolicRefOptions{
  915. Name: "HEAD",
  916. Ref: git.RefsHeads + conf.Repository.DefaultBranch,
  917. },
  918. )
  919. if err != nil {
  920. return errors.Wrap(err, "set default branch")
  921. }
  922. tmpDir := filepath.Join(os.TempDir(), "gogs-"+repo.Name+"-"+com.ToStr(time.Now().Nanosecond()))
  923. // Initialize repository according to user's choice.
  924. if opts.AutoInit {
  925. if err = os.MkdirAll(tmpDir, os.ModePerm); err != nil {
  926. return err
  927. }
  928. defer RemoveAllWithNotice("Delete repository for auto-initialization", tmpDir)
  929. if err = prepareRepoCommit(repo, tmpDir, repoPath, opts); err != nil {
  930. return errors.Newf("prepareRepoCommit: %v", err)
  931. }
  932. // Apply changes and commit.
  933. err = initRepoCommit(
  934. tmpDir,
  935. &git.Signature{
  936. Name: doer.DisplayName(),
  937. Email: doer.Email,
  938. When: time.Now(),
  939. },
  940. )
  941. if err != nil {
  942. return errors.Newf("initRepoCommit: %v", err)
  943. }
  944. }
  945. // Re-fetch the repository from database before updating it (else it would
  946. // override changes that were done earlier with sql)
  947. if repo, err = getRepositoryByID(e, repo.ID); err != nil {
  948. return errors.Newf("getRepositoryByID: %v", err)
  949. }
  950. if !opts.AutoInit {
  951. repo.IsBare = true
  952. }
  953. repo.DefaultBranch = conf.Repository.DefaultBranch
  954. if err = updateRepository(e, repo, false); err != nil {
  955. return errors.Newf("updateRepository: %v", err)
  956. }
  957. return nil
  958. }
  959. var (
  960. reservedRepoNames = map[string]struct{}{
  961. ".": {},
  962. "..": {},
  963. }
  964. reservedRepoPatterns = []string{
  965. "*.git",
  966. "*.wiki",
  967. }
  968. )
  969. // isRepoNameAllowed return an error if given name is a reserved name or pattern for repositories.
  970. func isRepoNameAllowed(name string) error {
  971. return isNameAllowed(reservedRepoNames, reservedRepoPatterns, name)
  972. }
  973. func createRepository(e *gorm.DB, doer, owner *User, repo *Repository) (err error) {
  974. if err = isRepoNameAllowed(repo.Name); err != nil {
  975. return err
  976. }
  977. has, err := isRepositoryExist(e, owner, repo.Name)
  978. if err != nil {
  979. return errors.Newf("IsRepositoryExist: %v", err)
  980. } else if has {
  981. return ErrRepoAlreadyExist{args: errutil.Args{"ownerID": owner.ID, "name": repo.Name}}
  982. }
  983. if err = e.Create(repo).Error; err != nil {
  984. return err
  985. }
  986. err = e.Exec(dbutil.Quote("UPDATE %s SET num_repos = num_repos + 1 WHERE id = ?", "user"), owner.ID).Error
  987. if err != nil {
  988. return errors.Wrap(err, "increase owned repository count")
  989. }
  990. // Give access to all members in owner team.
  991. if owner.IsOrganization() {
  992. t, err := owner.getOwnerTeam(e)
  993. if err != nil {
  994. return errors.Newf("getOwnerTeam: %v", err)
  995. } else if err = t.addRepository(e, repo); err != nil {
  996. return errors.Newf("addRepository: %v", err)
  997. }
  998. } else {
  999. // Organization automatically called this in addRepository method.
  1000. if err = repo.recalculateAccesses(e); err != nil {
  1001. return errors.Newf("recalculateAccesses: %v", err)
  1002. }
  1003. }
  1004. if err = watchRepo(e, owner.ID, repo.ID, true); err != nil {
  1005. return errors.Newf("watchRepo: %v", err)
  1006. }
  1007. // FIXME: This is identical to Actions.NewRepo but we are not yet able to wrap
  1008. // transaction with different ORM objects, should delete this once migrated to
  1009. // GORM for this part of logic.
  1010. newRepoAction := func(e *gorm.DB, doer *User, repo *Repository) (err error) {
  1011. opType := ActionCreateRepo
  1012. if repo.IsFork {
  1013. opType = ActionForkRepo
  1014. }
  1015. return notifyWatchers(e, &Action{
  1016. ActUserID: doer.ID,
  1017. ActUserName: doer.Name,
  1018. OpType: opType,
  1019. RepoID: repo.ID,
  1020. RepoUserName: repo.Owner.Name,
  1021. RepoName: repo.Name,
  1022. IsPrivate: repo.IsPrivate || repo.IsUnlisted,
  1023. CreatedUnix: time.Now().Unix(),
  1024. })
  1025. }
  1026. if err = newRepoAction(e, doer, repo); err != nil {
  1027. return errors.Newf("newRepoAction: %v", err)
  1028. }
  1029. return repo.loadAttributes(e)
  1030. }
  1031. type ErrReachLimitOfRepo struct {
  1032. Limit int
  1033. }
  1034. func IsErrReachLimitOfRepo(err error) bool {
  1035. _, ok := err.(ErrReachLimitOfRepo)
  1036. return ok
  1037. }
  1038. func (err ErrReachLimitOfRepo) Error() string {
  1039. return fmt.Sprintf("user has reached maximum limit of repositories [limit: %d]", err.Limit)
  1040. }
  1041. // CreateRepository creates a repository for given user or organization.
  1042. func CreateRepository(doer, owner *User, opts CreateRepoOptionsLegacy) (_ *Repository, err error) {
  1043. repoPath := RepoPath(owner.Name, opts.Name)
  1044. if osutil.Exist(repoPath) {
  1045. return nil, errors.Errorf("repository directory already exists: %s", repoPath)
  1046. }
  1047. if !owner.canCreateRepo() {
  1048. return nil, ErrReachLimitOfRepo{Limit: owner.maxNumRepos()}
  1049. }
  1050. repo := &Repository{
  1051. OwnerID: owner.ID,
  1052. Owner: owner,
  1053. Name: opts.Name,
  1054. LowerName: strings.ToLower(opts.Name),
  1055. Description: opts.Description,
  1056. IsPrivate: opts.IsPrivate,
  1057. IsUnlisted: opts.IsUnlisted,
  1058. EnableWiki: true,
  1059. EnableIssues: true,
  1060. EnablePulls: true,
  1061. }
  1062. err = db.Transaction(func(tx *gorm.DB) error {
  1063. if err := createRepository(tx, doer, owner, repo); err != nil {
  1064. return err
  1065. }
  1066. // No need for init mirror.
  1067. if !opts.IsMirror {
  1068. if err := initRepository(tx, repoPath, doer, repo, opts); err != nil {
  1069. RemoveAllWithNotice("Delete repository for initialization failure", repoPath)
  1070. return errors.Newf("initRepository: %v", err)
  1071. }
  1072. _, stderr, err := process.ExecDir(-1,
  1073. repoPath, fmt.Sprintf("CreateRepository 'git update-server-info': %s", repoPath),
  1074. "git", "update-server-info")
  1075. if err != nil {
  1076. return errors.Newf("CreateRepository 'git update-server-info': %s", stderr)
  1077. }
  1078. }
  1079. return nil
  1080. })
  1081. if err != nil {
  1082. return nil, err
  1083. }
  1084. // Remember visibility preference
  1085. err = Handle.Users().Update(context.TODO(), owner.ID, UpdateUserOptions{LastRepoVisibility: &repo.IsPrivate})
  1086. if err != nil {
  1087. return nil, errors.Wrap(err, "update user")
  1088. }
  1089. return repo, nil
  1090. }
  1091. func countRepositories(userID int64, private bool) int64 {
  1092. query := db.Model(&Repository{}).Where("id > 0")
  1093. if userID > 0 {
  1094. query = query.Where("owner_id = ?", userID)
  1095. }
  1096. if !private {
  1097. query = query.Where("is_private = ?", false)
  1098. }
  1099. var count int64
  1100. err := query.Count(&count).Error
  1101. if err != nil {
  1102. log.Error("countRepositories: %v", err)
  1103. }
  1104. return count
  1105. }
  1106. // CountRepositories returns number of repositories.
  1107. // Argument private only takes effect when it is false,
  1108. // set it true to count all repositories.
  1109. func CountRepositories(private bool) int64 {
  1110. return countRepositories(-1, private)
  1111. }
  1112. // CountUserRepositories returns number of repositories user owns.
  1113. // Argument private only takes effect when it is false,
  1114. // set it true to count all repositories.
  1115. func CountUserRepositories(userID int64, private bool) int64 {
  1116. return countRepositories(userID, private)
  1117. }
  1118. func Repositories(page, pageSize int) (_ []*Repository, err error) {
  1119. repos := make([]*Repository, 0, pageSize)
  1120. return repos, db.Limit(pageSize).Offset((page - 1) * pageSize).Order("id ASC").Find(&repos).Error
  1121. }
  1122. // RepositoriesWithUsers returns number of repos in given page.
  1123. func RepositoriesWithUsers(page, pageSize int) (_ []*Repository, err error) {
  1124. repos, err := Repositories(page, pageSize)
  1125. if err != nil {
  1126. return nil, errors.Newf("Repositories: %v", err)
  1127. }
  1128. for i := range repos {
  1129. if err = repos[i].GetOwner(); err != nil {
  1130. return nil, err
  1131. }
  1132. }
  1133. return repos, nil
  1134. }
  1135. // FilterRepositoryWithIssues selects repositories that are using internal issue tracker
  1136. // and has disabled external tracker from given set.
  1137. // It returns nil if result set is empty.
  1138. func FilterRepositoryWithIssues(repoIDs []int64) ([]int64, error) {
  1139. if len(repoIDs) == 0 {
  1140. return nil, nil
  1141. }
  1142. repos := make([]*Repository, 0, len(repoIDs))
  1143. if err := db.Select("id").
  1144. Where("enable_issues = ?", true).
  1145. Where("enable_external_tracker = ?", false).
  1146. Where("id IN ?", repoIDs).
  1147. Find(&repos).Error; err != nil {
  1148. return nil, errors.Newf("filter valid repositories %v: %v", repoIDs, err)
  1149. }
  1150. if len(repos) == 0 {
  1151. return nil, nil
  1152. }
  1153. repoIDs = make([]int64, len(repos))
  1154. for i := range repos {
  1155. repoIDs[i] = repos[i].ID
  1156. }
  1157. return repoIDs, nil
  1158. }
  1159. // RepoPath returns repository path by given user and repository name.
  1160. //
  1161. // Deprecated: Use repoutil.RepositoryPath instead.
  1162. func RepoPath(userName, repoName string) string {
  1163. return filepath.Join(repoutil.UserPath(userName), strings.ToLower(repoName)+".git")
  1164. }
  1165. // TransferOwnership transfers all corresponding setting from old user to new one.
  1166. func TransferOwnership(doer *User, newOwnerName string, repo *Repository) error {
  1167. newOwner, err := Handle.Users().GetByUsername(context.TODO(), newOwnerName)
  1168. if err != nil {
  1169. return errors.Newf("get new owner '%s': %v", newOwnerName, err)
  1170. }
  1171. // Check if new owner has repository with same name.
  1172. has, err := IsRepositoryExist(newOwner, repo.Name)
  1173. if err != nil {
  1174. return errors.Newf("IsRepositoryExist: %v", err)
  1175. } else if has {
  1176. return ErrRepoAlreadyExist{args: errutil.Args{"ownerName": newOwnerName, "name": repo.Name}}
  1177. }
  1178. owner := repo.Owner
  1179. // Note: we have to set value here to make sure recalculate accesses is based on
  1180. // new owner.
  1181. repo.OwnerID = newOwner.ID
  1182. repo.Owner = newOwner
  1183. return db.Transaction(func(tx *gorm.DB) error {
  1184. // Update repository.
  1185. if err := tx.Model(&Repository{}).Where("id = ?", repo.ID).Updates(repo).Error; err != nil {
  1186. return errors.Newf("update owner: %v", err)
  1187. }
  1188. // Remove redundant collaborators.
  1189. collaborators, err := repo.getCollaborators(tx)
  1190. if err != nil {
  1191. return errors.Newf("getCollaborators: %v", err)
  1192. }
  1193. // Dummy object.
  1194. collaboration := &Collaboration{RepoID: repo.ID}
  1195. for _, c := range collaborators {
  1196. collaboration.UserID = c.ID
  1197. if c.ID == newOwner.ID || newOwner.IsOrgMember(c.ID) {
  1198. if err := tx.Delete(collaboration).Error; err != nil {
  1199. return errors.Newf("remove collaborator '%d': %v", c.ID, err)
  1200. }
  1201. }
  1202. }
  1203. // Remove old team-repository relations.
  1204. if owner.IsOrganization() {
  1205. if err = owner.getTeams(tx); err != nil {
  1206. return errors.Newf("getTeams: %v", err)
  1207. }
  1208. for _, t := range owner.Teams {
  1209. if !t.hasRepository(tx, repo.ID) {
  1210. continue
  1211. }
  1212. t.NumRepos--
  1213. if err := tx.Model(&Team{}).Where("id = ?", t.ID).Updates(t).Error; err != nil {
  1214. return errors.Newf("decrease team repository count '%d': %v", t.ID, err)
  1215. }
  1216. }
  1217. if err = owner.removeOrgRepo(tx, repo.ID); err != nil {
  1218. return errors.Newf("removeOrgRepo: %v", err)
  1219. }
  1220. }
  1221. if newOwner.IsOrganization() {
  1222. t, err := newOwner.getOwnerTeam(tx)
  1223. if err != nil {
  1224. return errors.Newf("getOwnerTeam: %v", err)
  1225. } else if err = t.addRepository(tx, repo); err != nil {
  1226. return errors.Newf("add to owner team: %v", err)
  1227. }
  1228. } else {
  1229. // Organization called this in addRepository method.
  1230. if err = repo.recalculateAccesses(tx); err != nil {
  1231. return errors.Newf("recalculateAccesses: %v", err)
  1232. }
  1233. }
  1234. // Update repository count.
  1235. if err = tx.Exec("UPDATE `user` SET num_repos=num_repos+1 WHERE id=?", newOwner.ID).Error; err != nil {
  1236. return errors.Newf("increase new owner repository count: %v", err)
  1237. } else if err = tx.Exec("UPDATE `user` SET num_repos=num_repos-1 WHERE id=?", owner.ID).Error; err != nil {
  1238. return errors.Newf("decrease old owner repository count: %v", err)
  1239. }
  1240. // Remove watch for organization.
  1241. if owner.IsOrganization() {
  1242. if err = watchRepo(tx, owner.ID, repo.ID, false); err != nil {
  1243. return errors.Wrap(err, "unwatch repository for the organization owner")
  1244. }
  1245. }
  1246. if err = watchRepo(tx, newOwner.ID, repo.ID, true); err != nil {
  1247. return errors.Newf("watchRepo: %v", err)
  1248. }
  1249. // FIXME: This is identical to Actions.TransferRepo but we are not yet able to
  1250. // wrap transaction with different ORM objects, should delete this once migrated
  1251. // to GORM for this part of logic.
  1252. transferRepoAction := func(e *gorm.DB, doer, oldOwner *User, repo *Repository) error {
  1253. return notifyWatchers(e, &Action{
  1254. ActUserID: doer.ID,
  1255. ActUserName: doer.Name,
  1256. OpType: ActionTransferRepo,
  1257. RepoID: repo.ID,
  1258. RepoUserName: repo.Owner.Name,
  1259. RepoName: repo.Name,
  1260. IsPrivate: repo.IsPrivate || repo.IsUnlisted,
  1261. Content: path.Join(oldOwner.Name, repo.Name),
  1262. CreatedUnix: time.Now().Unix(),
  1263. })
  1264. }
  1265. if err = transferRepoAction(tx, doer, owner, repo); err != nil {
  1266. return errors.Newf("transferRepoAction: %v", err)
  1267. }
  1268. // Rename remote repository to new path and delete local copy.
  1269. if err = os.MkdirAll(repoutil.UserPath(newOwner.Name), os.ModePerm); err != nil {
  1270. return err
  1271. }
  1272. if err = os.Rename(RepoPath(owner.Name, repo.Name), RepoPath(newOwner.Name, repo.Name)); err != nil {
  1273. return errors.Newf("rename repository directory: %v", err)
  1274. }
  1275. deleteRepoLocalCopy(repo.ID)
  1276. // Rename remote wiki repository to new path and delete local copy.
  1277. wikiPath := WikiPath(owner.Name, repo.Name)
  1278. if com.IsExist(wikiPath) {
  1279. RemoveAllWithNotice("Delete repository wiki local copy", repo.LocalWikiPath())
  1280. if err = os.Rename(wikiPath, WikiPath(newOwner.Name, repo.Name)); err != nil {
  1281. return errors.Newf("rename repository wiki: %v", err)
  1282. }
  1283. }
  1284. return nil
  1285. })
  1286. }
  1287. func deleteRepoLocalCopy(repoID int64) {
  1288. repoWorkingPool.CheckIn(com.ToStr(repoID))
  1289. defer repoWorkingPool.CheckOut(com.ToStr(repoID))
  1290. RemoveAllWithNotice(fmt.Sprintf("Delete repository %d local copy", repoID), repoutil.RepositoryLocalPath(repoID))
  1291. }
  1292. // ChangeRepositoryName changes all corresponding setting from old repository name to new one.
  1293. func ChangeRepositoryName(u *User, oldRepoName, newRepoName string) (err error) {
  1294. oldRepoName = strings.ToLower(oldRepoName)
  1295. newRepoName = strings.ToLower(newRepoName)
  1296. if err = isRepoNameAllowed(newRepoName); err != nil {
  1297. return err
  1298. }
  1299. has, err := IsRepositoryExist(u, newRepoName)
  1300. if err != nil {
  1301. return errors.Newf("IsRepositoryExist: %v", err)
  1302. } else if has {
  1303. return ErrRepoAlreadyExist{args: errutil.Args{"ownerID": u.ID, "name": newRepoName}}
  1304. }
  1305. repo, err := GetRepositoryByName(u.ID, oldRepoName)
  1306. if err != nil {
  1307. return errors.Newf("GetRepositoryByName: %v", err)
  1308. }
  1309. // Change repository directory name
  1310. if err = os.Rename(repo.RepoPath(), RepoPath(u.Name, newRepoName)); err != nil {
  1311. return errors.Newf("rename repository directory: %v", err)
  1312. }
  1313. wikiPath := repo.WikiPath()
  1314. if com.IsExist(wikiPath) {
  1315. if err = os.Rename(wikiPath, WikiPath(u.Name, newRepoName)); err != nil {
  1316. return errors.Newf("rename repository wiki: %v", err)
  1317. }
  1318. RemoveAllWithNotice("Delete repository wiki local copy", repo.LocalWikiPath())
  1319. }
  1320. deleteRepoLocalCopy(repo.ID)
  1321. return nil
  1322. }
  1323. func getRepositoriesByForkID(e *gorm.DB, forkID int64) ([]*Repository, error) {
  1324. repos := make([]*Repository, 0, 10)
  1325. return repos, e.Where("fork_id = ?", forkID).Find(&repos).Error
  1326. }
  1327. // GetRepositoriesByForkID returns all repositories with given fork ID.
  1328. func GetRepositoriesByForkID(forkID int64) ([]*Repository, error) {
  1329. return getRepositoriesByForkID(db, forkID)
  1330. }
  1331. func getNonMirrorRepositories(e *gorm.DB) ([]*Repository, error) {
  1332. repos := make([]*Repository, 0, 10)
  1333. return repos, e.Where("is_mirror = ?", false).Find(&repos).Error
  1334. }
  1335. // GetRepositoriesMirror returns only mirror repositories with user.
  1336. func GetNonMirrorRepositories() ([]*Repository, error) {
  1337. return getNonMirrorRepositories(db)
  1338. }
  1339. func updateRepository(e *gorm.DB, repo *Repository, visibilityChanged bool) (err error) {
  1340. repo.LowerName = strings.ToLower(repo.Name)
  1341. if len(repo.Description) > 512 {
  1342. repo.Description = repo.Description[:512]
  1343. }
  1344. if len(repo.Website) > 255 {
  1345. repo.Website = repo.Website[:255]
  1346. }
  1347. if err = e.Model(&Repository{}).Where("id = ?", repo.ID).Updates(repo).Error; err != nil {
  1348. return errors.Newf("update: %v", err)
  1349. }
  1350. if visibilityChanged {
  1351. if err = repo.getOwner(e); err != nil {
  1352. return errors.Newf("getOwner: %v", err)
  1353. }
  1354. if repo.Owner.IsOrganization() {
  1355. // Organization repository need to recalculate access table when visibility is changed
  1356. if err = repo.recalculateTeamAccesses(e, 0); err != nil {
  1357. return errors.Newf("recalculateTeamAccesses: %v", err)
  1358. }
  1359. }
  1360. // Create/Remove git-daemon-export-ok for git-daemon
  1361. daemonExportFile := path.Join(repo.RepoPath(), "git-daemon-export-ok")
  1362. if repo.IsPrivate && com.IsExist(daemonExportFile) {
  1363. if err = os.Remove(daemonExportFile); err != nil {
  1364. log.Error("Failed to remove %s: %v", daemonExportFile, err)
  1365. }
  1366. } else if !repo.IsPrivate && !com.IsExist(daemonExportFile) {
  1367. if f, err := os.Create(daemonExportFile); err != nil {
  1368. log.Error("Failed to create %s: %v", daemonExportFile, err)
  1369. } else {
  1370. f.Close()
  1371. }
  1372. }
  1373. forkRepos, err := getRepositoriesByForkID(e, repo.ID)
  1374. if err != nil {
  1375. return errors.Newf("getRepositoriesByForkID: %v", err)
  1376. }
  1377. for i := range forkRepos {
  1378. forkRepos[i].IsPrivate = repo.IsPrivate
  1379. forkRepos[i].IsUnlisted = repo.IsUnlisted
  1380. if err = updateRepository(e, forkRepos[i], true); err != nil {
  1381. return errors.Newf("updateRepository[%d]: %v", forkRepos[i].ID, err)
  1382. }
  1383. }
  1384. // Change visibility of generated actions
  1385. if err = e.Model(&Action{}).Where("repo_id = ?", repo.ID).Update("is_private", repo.IsPrivate || repo.IsUnlisted).Error; err != nil {
  1386. return errors.Newf("change action visibility of repository: %v", err)
  1387. }
  1388. }
  1389. return nil
  1390. }
  1391. func UpdateRepository(repo *Repository, visibilityChanged bool) (err error) {
  1392. return db.Transaction(func(tx *gorm.DB) error {
  1393. if err := updateRepository(tx, repo, visibilityChanged); err != nil {
  1394. return errors.Newf("updateRepository: %v", err)
  1395. }
  1396. return nil
  1397. })
  1398. }
  1399. // DeleteRepository deletes a repository for a user or organization.
  1400. func DeleteRepository(ownerID, repoID int64) error {
  1401. repo := &Repository{ID: repoID, OwnerID: ownerID}
  1402. err := db.First(repo).Error
  1403. if err != nil {
  1404. if err == gorm.ErrRecordNotFound {
  1405. return ErrRepoNotExist{args: map[string]any{"ownerID": ownerID, "repoID": repoID}}
  1406. }
  1407. return err
  1408. }
  1409. // In case is a organization.
  1410. org, err := Handle.Users().GetByID(context.TODO(), ownerID)
  1411. if err != nil {
  1412. return err
  1413. }
  1414. if org.IsOrganization() {
  1415. if err = org.GetTeams(); err != nil {
  1416. return err
  1417. }
  1418. }
  1419. return db.Transaction(func(tx *gorm.DB) error {
  1420. if org.IsOrganization() {
  1421. for _, t := range org.Teams {
  1422. if !t.hasRepository(tx, repoID) {
  1423. continue
  1424. } else if err = t.removeRepository(tx, repo, false); err != nil {
  1425. return err
  1426. }
  1427. }
  1428. }
  1429. if err = deleteBeans(tx,
  1430. &Repository{ID: repoID},
  1431. &Access{RepoID: repo.ID},
  1432. &Action{RepoID: repo.ID},
  1433. &Watch{RepoID: repoID},
  1434. &Star{RepoID: repoID},
  1435. &Mirror{RepoID: repoID},
  1436. &IssueUser{RepoID: repoID},
  1437. &Milestone{RepoID: repoID},
  1438. &Release{RepoID: repoID},
  1439. &Collaboration{RepoID: repoID},
  1440. &PullRequest{BaseRepoID: repoID},
  1441. &ProtectBranch{RepoID: repoID},
  1442. &ProtectBranchWhitelist{RepoID: repoID},
  1443. &Webhook{RepoID: repoID},
  1444. &HookTask{RepoID: repoID},
  1445. &LFSObject{RepoID: repoID},
  1446. ); err != nil {
  1447. return errors.Newf("deleteBeans: %v", err)
  1448. }
  1449. // Delete comments and attachments.
  1450. issues := make([]*Issue, 0, 25)
  1451. attachmentPaths := make([]string, 0, len(issues))
  1452. if err = tx.Where("repo_id = ?", repoID).Find(&issues).Error; err != nil {
  1453. return err
  1454. }
  1455. for i := range issues {
  1456. if err = tx.Where("issue_id = ?", issues[i].ID).Delete(&Comment{}).Error; err != nil {
  1457. return err
  1458. }
  1459. attachments := make([]*Attachment, 0, 5)
  1460. if err = tx.Where("issue_id = ?", issues[i].ID).Find(&attachments).Error; err != nil {
  1461. return err
  1462. }
  1463. for j := range attachments {
  1464. attachmentPaths = append(attachmentPaths, attachments[j].LocalPath())
  1465. }
  1466. if err = tx.Where("issue_id = ?", issues[i].ID).Delete(&Attachment{}).Error; err != nil {
  1467. return err
  1468. }
  1469. }
  1470. if err = tx.Where("repo_id = ?", repoID).Delete(&Issue{}).Error; err != nil {
  1471. return err
  1472. }
  1473. if repo.IsFork {
  1474. if err = tx.Exec("UPDATE `repository` SET num_forks=num_forks-1 WHERE id=?", repo.ForkID).Error; err != nil {
  1475. return errors.Newf("decrease fork count: %v", err)
  1476. }
  1477. }
  1478. if err = tx.Exec("UPDATE `user` SET num_repos=num_repos-1 WHERE id=?", ownerID).Error; err != nil {
  1479. return err
  1480. }
  1481. // Remove repository files.
  1482. repoPath := repo.RepoPath()
  1483. RemoveAllWithNotice("Delete repository files", repoPath)
  1484. repo.DeleteWiki()
  1485. // Remove attachment files.
  1486. for i := range attachmentPaths {
  1487. RemoveAllWithNotice("Delete attachment", attachmentPaths[i])
  1488. }
  1489. if repo.NumForks > 0 {
  1490. if err = tx.Exec("UPDATE `repository` SET fork_id=0,is_fork=? WHERE fork_id=?", false, repo.ID).Error; err != nil {
  1491. log.Error("reset 'fork_id' and 'is_fork': %v", err)
  1492. }
  1493. }
  1494. return nil
  1495. })
  1496. }
  1497. // GetRepositoryByRef returns a Repository specified by a GFM reference.
  1498. // See https://help.github.com/articles/writing-on-github#references for more information on the syntax.
  1499. func GetRepositoryByRef(ref string) (*Repository, error) {
  1500. n := strings.IndexByte(ref, byte('/'))
  1501. if n < 2 {
  1502. return nil, InvalidRepoReference{Ref: ref}
  1503. }
  1504. userName, repoName := ref[:n], ref[n+1:]
  1505. user, err := Handle.Users().GetByUsername(context.TODO(), userName)
  1506. if err != nil {
  1507. return nil, err
  1508. }
  1509. return GetRepositoryByName(user.ID, repoName)
  1510. }
  1511. // GetRepositoryByName returns the repository by given name under user if exists.
  1512. // Deprecated: Use Repos.GetByName instead.
  1513. func GetRepositoryByName(ownerID int64, name string) (*Repository, error) {
  1514. repo := &Repository{
  1515. OwnerID: ownerID,
  1516. LowerName: strings.ToLower(name),
  1517. }
  1518. err := db.Where("owner_id = ? AND lower_name = ?", ownerID, strings.ToLower(name)).First(repo).Error
  1519. if err != nil {
  1520. if err == gorm.ErrRecordNotFound {
  1521. return nil, ErrRepoNotExist{args: map[string]any{"ownerID": ownerID, "name": name}}
  1522. }
  1523. return nil, err
  1524. }
  1525. return repo, repo.LoadAttributes()
  1526. }
  1527. func getRepositoryByID(e *gorm.DB, id int64) (*Repository, error) {
  1528. repo := new(Repository)
  1529. err := e.First(repo, id).Error
  1530. if err != nil {
  1531. if err == gorm.ErrRecordNotFound {
  1532. return nil, ErrRepoNotExist{args: map[string]any{"repoID": id}}
  1533. }
  1534. return nil, err
  1535. }
  1536. return repo, repo.loadAttributes(e)
  1537. }
  1538. // GetRepositoryByID returns the repository by given id if exists.
  1539. func GetRepositoryByID(id int64) (*Repository, error) {
  1540. return getRepositoryByID(db, id)
  1541. }
  1542. type UserRepoOptions struct {
  1543. UserID int64
  1544. Private bool
  1545. Page int
  1546. PageSize int
  1547. }
  1548. // GetUserRepositories returns a list of repositories of given user.
  1549. func GetUserRepositories(opts *UserRepoOptions) ([]*Repository, error) {
  1550. query := db.Where("owner_id = ?", opts.UserID).Order("updated_unix DESC")
  1551. if !opts.Private {
  1552. query = query.Where("is_private = ?", false).Where("is_unlisted = ?", false)
  1553. }
  1554. if opts.Page <= 0 {
  1555. opts.Page = 1
  1556. }
  1557. query = query.Limit(opts.PageSize).Offset((opts.Page - 1) * opts.PageSize)
  1558. repos := make([]*Repository, 0, opts.PageSize)
  1559. return repos, query.Find(&repos).Error
  1560. }
  1561. // GetUserRepositories returns a list of mirror repositories of given user.
  1562. func GetUserMirrorRepositories(userID int64) ([]*Repository, error) {
  1563. repos := make([]*Repository, 0, 10)
  1564. return repos, db.Where("owner_id = ?", userID).Where("is_mirror = ?", true).Find(&repos).Error
  1565. }
  1566. // GetRecentUpdatedRepositories returns the list of repositories that are recently updated.
  1567. func GetRecentUpdatedRepositories(page, pageSize int) (repos []*Repository, err error) {
  1568. return repos, db.Limit(pageSize).Offset((page-1)*pageSize).
  1569. Where("is_private = ?", false).Order("updated_unix DESC").Find(&repos).Error
  1570. }
  1571. // GetUserAndCollaborativeRepositories returns list of repositories the user owns and collaborates.
  1572. func GetUserAndCollaborativeRepositories(userID int64) ([]*Repository, error) {
  1573. repos := make([]*Repository, 0, 10)
  1574. if err := db.Table("repository AS repo").
  1575. Joins("INNER JOIN collaboration ON collaboration.repo_id = repo.id").
  1576. Where("collaboration.user_id = ?", userID).
  1577. Find(&repos).Error; err != nil {
  1578. return nil, errors.Newf("select collaborative repositories: %v", err)
  1579. }
  1580. ownRepos := make([]*Repository, 0, 10)
  1581. if err := db.Where("owner_id = ?", userID).Find(&ownRepos).Error; err != nil {
  1582. return nil, errors.Newf("select own repositories: %v", err)
  1583. }
  1584. return append(repos, ownRepos...), nil
  1585. }
  1586. type SearchRepoOptions struct {
  1587. Keyword string
  1588. OwnerID int64
  1589. UserID int64 // When set results will contain all public/private repositories user has access to
  1590. OrderBy string
  1591. Private bool // Include private repositories in results
  1592. Page int
  1593. PageSize int // Can be smaller than or equal to setting.ExplorePagingNum
  1594. }
  1595. // SearchRepositoryByName takes keyword and part of repository name to search,
  1596. // it returns results in given range and number of total results.
  1597. func SearchRepositoryByName(opts *SearchRepoOptions) (repos []*Repository, count int64, err error) {
  1598. if opts.Page <= 0 {
  1599. opts.Page = 1
  1600. }
  1601. repos = make([]*Repository, 0, opts.PageSize)
  1602. query := db.Table("repository AS repo")
  1603. // Attempt to find repositories that opts.UserID has access to,
  1604. // this does not include other people's private repositories even if opts.UserID is an admin.
  1605. if !opts.Private && opts.UserID > 0 {
  1606. query = query.Joins("LEFT JOIN access ON access.repo_id = repo.id").
  1607. Where("repo.owner_id = ? OR access.user_id = ? OR (repo.is_private = ? AND repo.is_unlisted = ?) OR (repo.is_private = ? AND (repo.allow_public_wiki = ? OR repo.allow_public_issues = ?))", opts.UserID, opts.UserID, false, false, true, true, true)
  1608. } else {
  1609. // Only return public repositories if opts.Private is not set
  1610. if !opts.Private {
  1611. query = query.Where("(repo.is_private = ? AND repo.is_unlisted = ?) OR (repo.is_private = ? AND (repo.allow_public_wiki = ? OR repo.allow_public_issues = ?))", false, false, true, true, true)
  1612. }
  1613. }
  1614. if len(opts.Keyword) > 0 {
  1615. query = query.Where("repo.lower_name LIKE ? OR repo.description LIKE ?", "%"+strings.ToLower(opts.Keyword)+"%", "%"+strings.ToLower(opts.Keyword)+"%")
  1616. }
  1617. if opts.OwnerID > 0 {
  1618. query = query.Where("repo.owner_id = ?", opts.OwnerID)
  1619. }
  1620. // We need all fields (repo.*) in final list but only ID (repo.id) is good enough for counting.
  1621. err = query.Session(&gorm.Session{}).Distinct("repo.id").Count(&count).Error
  1622. if err != nil {
  1623. return nil, 0, errors.Newf("Count: %v", err)
  1624. }
  1625. if len(opts.OrderBy) > 0 {
  1626. query = query.Order("repo." + opts.OrderBy)
  1627. }
  1628. return repos, count, query.Distinct("repo.*").Limit(opts.PageSize).Offset((opts.Page - 1) * opts.PageSize).Find(&repos).Error
  1629. }
  1630. func DeleteOldRepositoryArchives() {
  1631. if taskStatusTable.IsRunning(taskNameCleanOldArchives) {
  1632. return
  1633. }
  1634. taskStatusTable.Start(taskNameCleanOldArchives)
  1635. defer taskStatusTable.Stop(taskNameCleanOldArchives)
  1636. log.Trace("Doing: DeleteOldRepositoryArchives")
  1637. formats := []string{"zip", "targz"}
  1638. oldestTime := time.Now().Add(-conf.Cron.RepoArchiveCleanup.OlderThan)
  1639. var repos []Repository
  1640. err := db.Where("id > 0").FindInBatches(&repos, 100, func(tx *gorm.DB, batch int) error {
  1641. for _, repo := range repos {
  1642. basePath := filepath.Join(repo.RepoPath(), "archives")
  1643. for _, format := range formats {
  1644. dirPath := filepath.Join(basePath, format)
  1645. if !com.IsDir(dirPath) {
  1646. continue
  1647. }
  1648. dir, err := os.Open(dirPath)
  1649. if err != nil {
  1650. log.Error("Failed to open directory '%s': %v", dirPath, err)
  1651. continue
  1652. }
  1653. fis, err := dir.Readdir(0)
  1654. dir.Close()
  1655. if err != nil {
  1656. log.Error("Failed to read directory '%s': %v", dirPath, err)
  1657. continue
  1658. }
  1659. for _, fi := range fis {
  1660. if fi.IsDir() || fi.ModTime().After(oldestTime) {
  1661. continue
  1662. }
  1663. archivePath := filepath.Join(dirPath, fi.Name())
  1664. if err = os.Remove(archivePath); err != nil {
  1665. const fmtStr = "Failed to health delete archive %q: %v"
  1666. log.Warn(fmtStr, archivePath, err)
  1667. if err = Handle.Notices().Create(
  1668. context.TODO(),
  1669. NoticeTypeRepository,
  1670. fmt.Sprintf(fmtStr, archivePath, err),
  1671. ); err != nil {
  1672. log.Error("CreateRepositoryNotice: %v", err)
  1673. }
  1674. }
  1675. }
  1676. }
  1677. }
  1678. return nil
  1679. }).Error
  1680. if err != nil {
  1681. log.Error("DeleteOldRepositoryArchives: %v", err)
  1682. }
  1683. }
  1684. // DeleteRepositoryArchives deletes all repositories' archives.
  1685. func DeleteRepositoryArchives() error {
  1686. if taskStatusTable.IsRunning(taskNameCleanOldArchives) {
  1687. return nil
  1688. }
  1689. taskStatusTable.Start(taskNameCleanOldArchives)
  1690. defer taskStatusTable.Stop(taskNameCleanOldArchives)
  1691. return db.Where("id > 0").FindInBatches(&[]Repository{}, 100, func(tx *gorm.DB, batch int) error {
  1692. var repos []Repository
  1693. if err := tx.Find(&repos).Error; err != nil {
  1694. return err
  1695. }
  1696. for _, repo := range repos {
  1697. if err := os.RemoveAll(filepath.Join(repo.RepoPath(), "archives")); err != nil {
  1698. return err
  1699. }
  1700. }
  1701. return nil
  1702. }).Error
  1703. }
  1704. func gatherMissingRepoRecords() ([]*Repository, error) {
  1705. repos := make([]*Repository, 0, 10)
  1706. err := db.Where("id > 0").FindInBatches(&[]Repository{}, 100, func(tx *gorm.DB, batch int) error {
  1707. var batchRepos []Repository
  1708. if err := tx.Find(&batchRepos).Error; err != nil {
  1709. return err
  1710. }
  1711. for i := range batchRepos {
  1712. if !com.IsDir(batchRepos[i].RepoPath()) {
  1713. repos = append(repos, &batchRepos[i])
  1714. }
  1715. }
  1716. return nil
  1717. }).Error
  1718. if err != nil {
  1719. if err2 := Handle.Notices().Create(context.TODO(), NoticeTypeRepository, fmt.Sprintf("gatherMissingRepoRecords: %v", err)); err2 != nil {
  1720. return nil, errors.Newf("CreateRepositoryNotice: %v", err)
  1721. }
  1722. }
  1723. return repos, nil
  1724. }
  1725. // DeleteMissingRepositories deletes all repository records that lost Git files.
  1726. func DeleteMissingRepositories() error {
  1727. repos, err := gatherMissingRepoRecords()
  1728. if err != nil {
  1729. return errors.Newf("gatherMissingRepoRecords: %v", err)
  1730. }
  1731. if len(repos) == 0 {
  1732. return nil
  1733. }
  1734. for _, repo := range repos {
  1735. log.Trace("Deleting %d/%d...", repo.OwnerID, repo.ID)
  1736. if err := DeleteRepository(repo.OwnerID, repo.ID); err != nil {
  1737. if err2 := Handle.Notices().Create(context.TODO(), NoticeTypeRepository, fmt.Sprintf("DeleteRepository [%d]: %v", repo.ID, err)); err2 != nil {
  1738. return errors.Newf("CreateRepositoryNotice: %v", err)
  1739. }
  1740. }
  1741. }
  1742. return nil
  1743. }
  1744. // ReinitMissingRepositories reinitializes all repository records that lost Git files.
  1745. func ReinitMissingRepositories() error {
  1746. repos, err := gatherMissingRepoRecords()
  1747. if err != nil {
  1748. return errors.Newf("gatherMissingRepoRecords: %v", err)
  1749. }
  1750. if len(repos) == 0 {
  1751. return nil
  1752. }
  1753. for _, repo := range repos {
  1754. log.Trace("Initializing %d/%d...", repo.OwnerID, repo.ID)
  1755. if err := git.Init(repo.RepoPath(), git.InitOptions{Bare: true}); err != nil {
  1756. if err2 := Handle.Notices().Create(context.TODO(), NoticeTypeRepository, fmt.Sprintf("init repository [repo_id: %d]: %v", repo.ID, err)); err2 != nil {
  1757. return errors.Newf("create repository notice: %v", err)
  1758. }
  1759. }
  1760. }
  1761. return nil
  1762. }
  1763. // SyncRepositoryHooks rewrites all repositories' pre-receive, update and post-receive hooks
  1764. // to make sure the binary and custom conf path are up-to-date.
  1765. func SyncRepositoryHooks() error {
  1766. return db.Where("id > 0").FindInBatches(&[]Repository{}, 100, func(tx *gorm.DB, batch int) error {
  1767. var repos []Repository
  1768. if err := tx.Find(&repos).Error; err != nil {
  1769. return err
  1770. }
  1771. for _, repo := range repos {
  1772. if err := createDelegateHooks(repo.RepoPath()); err != nil {
  1773. return err
  1774. }
  1775. if repo.HasWiki() {
  1776. if err := createDelegateHooks(repo.WikiPath()); err != nil {
  1777. return err
  1778. }
  1779. }
  1780. }
  1781. return nil
  1782. }).Error
  1783. }
  1784. // Prevent duplicate running tasks.
  1785. var taskStatusTable = sync.NewStatusTable()
  1786. const (
  1787. taskNameMirrorUpdate = "mirror_update"
  1788. taskNameGitFSCK = "git_fsck"
  1789. taskNameCheckRepoStats = "check_repos_stats"
  1790. taskNameCleanOldArchives = "clean_old_archives"
  1791. )
  1792. // GitFsck calls 'git fsck' to check repository health.
  1793. func GitFsck() {
  1794. if taskStatusTable.IsRunning(taskNameGitFSCK) {
  1795. return
  1796. }
  1797. taskStatusTable.Start(taskNameGitFSCK)
  1798. defer taskStatusTable.Stop(taskNameGitFSCK)
  1799. log.Trace("Doing: GitFsck")
  1800. err := db.Where("id > 0").FindInBatches(&[]Repository{}, 100, func(tx *gorm.DB, batch int) error {
  1801. var repos []Repository
  1802. if err := tx.Find(&repos).Error; err != nil {
  1803. return err
  1804. }
  1805. for _, repo := range repos {
  1806. repoPath := repo.RepoPath()
  1807. err := git.Fsck(repoPath, git.FsckOptions{
  1808. CommandOptions: git.CommandOptions{
  1809. Args: conf.Cron.RepoHealthCheck.Args,
  1810. },
  1811. Timeout: conf.Cron.RepoHealthCheck.Timeout,
  1812. })
  1813. if err != nil {
  1814. const fmtStr = "Failed to perform health check on repository %q: %v"
  1815. log.Warn(fmtStr, repoPath, err)
  1816. if err = Handle.Notices().Create(
  1817. context.TODO(),
  1818. NoticeTypeRepository,
  1819. fmt.Sprintf(fmtStr, repoPath, err),
  1820. ); err != nil {
  1821. log.Error("CreateRepositoryNotice: %v", err)
  1822. }
  1823. }
  1824. }
  1825. return nil
  1826. }).Error
  1827. if err != nil {
  1828. log.Error("GitFsck: %v", err)
  1829. }
  1830. }
  1831. func GitGcRepos() error {
  1832. args := append([]string{"gc"}, conf.Git.GCArgs...)
  1833. return db.Where("id > 0").FindInBatches(&[]Repository{}, 100, func(tx *gorm.DB, batch int) error {
  1834. var repos []Repository
  1835. if err := tx.Find(&repos).Error; err != nil {
  1836. return err
  1837. }
  1838. for _, repo := range repos {
  1839. if err := repo.GetOwner(); err != nil {
  1840. return err
  1841. }
  1842. _, stderr, err := process.ExecDir(
  1843. time.Duration(conf.Git.Timeout.GC)*time.Second,
  1844. RepoPath(repo.Owner.Name, repo.Name), "Repository garbage collection",
  1845. "git", args...)
  1846. if err != nil {
  1847. return errors.Newf("%v: %v", err, stderr)
  1848. }
  1849. }
  1850. return nil
  1851. }).Error
  1852. }
  1853. type repoChecker struct {
  1854. querySQL, correctSQL string
  1855. desc string
  1856. }
  1857. func repoStatsCheck(checker *repoChecker) {
  1858. type Result struct {
  1859. ID int64 `gorm:"column:id"`
  1860. }
  1861. var results []Result
  1862. err := db.Raw(checker.querySQL).Scan(&results).Error
  1863. if err != nil {
  1864. log.Error("Select %s: %v", checker.desc, err)
  1865. return
  1866. }
  1867. for _, result := range results {
  1868. log.Trace("Updating %s: %d", checker.desc, result.ID)
  1869. err = db.Exec(checker.correctSQL, result.ID, result.ID).Error
  1870. if err != nil {
  1871. log.Error("Update %s[%d]: %v", checker.desc, result.ID, err)
  1872. }
  1873. }
  1874. }
  1875. func CheckRepoStats() {
  1876. if taskStatusTable.IsRunning(taskNameCheckRepoStats) {
  1877. return
  1878. }
  1879. taskStatusTable.Start(taskNameCheckRepoStats)
  1880. defer taskStatusTable.Stop(taskNameCheckRepoStats)
  1881. log.Trace("Doing: CheckRepoStats")
  1882. checkers := []*repoChecker{
  1883. // Repository.NumWatches
  1884. {
  1885. "SELECT repo.id FROM `repository` repo WHERE repo.num_watches!=(SELECT COUNT(*) FROM `watch` WHERE repo_id=repo.id)",
  1886. "UPDATE `repository` SET num_watches=(SELECT COUNT(*) FROM `watch` WHERE repo_id=?) WHERE id=?",
  1887. "repository count 'num_watches'",
  1888. },
  1889. // Repository.NumStars
  1890. {
  1891. "SELECT repo.id FROM `repository` repo WHERE repo.num_stars!=(SELECT COUNT(*) FROM `star` WHERE repo_id=repo.id)",
  1892. "UPDATE `repository` SET num_stars=(SELECT COUNT(*) FROM `star` WHERE repo_id=?) WHERE id=?",
  1893. "repository count 'num_stars'",
  1894. },
  1895. // Label.NumIssues
  1896. {
  1897. "SELECT label.id FROM `label` WHERE label.num_issues!=(SELECT COUNT(*) FROM `issue_label` WHERE label_id=label.id)",
  1898. "UPDATE `label` SET num_issues=(SELECT COUNT(*) FROM `issue_label` WHERE label_id=?) WHERE id=?",
  1899. "label count 'num_issues'",
  1900. },
  1901. // User.NumRepos
  1902. {
  1903. "SELECT `user`.id FROM `user` WHERE `user`.num_repos!=(SELECT COUNT(*) FROM `repository` WHERE owner_id=`user`.id)",
  1904. "UPDATE `user` SET num_repos=(SELECT COUNT(*) FROM `repository` WHERE owner_id=?) WHERE id=?",
  1905. "user count 'num_repos'",
  1906. },
  1907. // Issue.NumComments
  1908. {
  1909. "SELECT `issue`.id FROM `issue` WHERE `issue`.num_comments!=(SELECT COUNT(*) FROM `comment` WHERE issue_id=`issue`.id AND type=0)",
  1910. "UPDATE `issue` SET num_comments=(SELECT COUNT(*) FROM `comment` WHERE issue_id=? AND type=0) WHERE id=?",
  1911. "issue count 'num_comments'",
  1912. },
  1913. }
  1914. for i := range checkers {
  1915. repoStatsCheck(checkers[i])
  1916. }
  1917. // ***** START: Repository.NumClosedIssues *****
  1918. desc := "repository count 'num_closed_issues'"
  1919. type Result struct {
  1920. ID int64 `gorm:"column:id"`
  1921. }
  1922. var results []Result
  1923. err := db.Raw("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).Scan(&results).Error
  1924. if err != nil {
  1925. log.Error("Select %s: %v", desc, err)
  1926. } else {
  1927. for _, result := range results {
  1928. log.Trace("Updating %s: %d", desc, result.ID)
  1929. err = db.Exec("UPDATE `repository` SET num_closed_issues=(SELECT COUNT(*) FROM `issue` WHERE repo_id=? AND is_closed=? AND is_pull=?) WHERE id=?", result.ID, true, false, result.ID).Error
  1930. if err != nil {
  1931. log.Error("Update %s[%d]: %v", desc, result.ID, err)
  1932. }
  1933. }
  1934. }
  1935. // ***** END: Repository.NumClosedIssues *****
  1936. // FIXME: use checker when stop supporting old fork repo format.
  1937. // ***** START: Repository.NumForks *****
  1938. results = nil
  1939. err = db.Raw("SELECT repo.id FROM `repository` repo WHERE repo.num_forks!=(SELECT COUNT(*) FROM `repository` WHERE fork_id=repo.id)").Scan(&results).Error
  1940. if err != nil {
  1941. log.Error("Select repository count 'num_forks': %v", err)
  1942. } else {
  1943. for _, result := range results {
  1944. log.Trace("Updating repository count 'num_forks': %d", result.ID)
  1945. repo, err := GetRepositoryByID(result.ID)
  1946. if err != nil {
  1947. log.Error("GetRepositoryByID[%d]: %v", result.ID, err)
  1948. continue
  1949. }
  1950. var count int64
  1951. err = db.Model(&Repository{}).Where("fork_id = ?", repo.ID).Count(&count).Error
  1952. if err != nil {
  1953. log.Error("Select count of forks[%d]: %v", repo.ID, err)
  1954. continue
  1955. }
  1956. repo.NumForks = int(count)
  1957. if err = UpdateRepository(repo, false); err != nil {
  1958. log.Error("UpdateRepository[%d]: %v", result.ID, err)
  1959. continue
  1960. }
  1961. }
  1962. }
  1963. // ***** END: Repository.NumForks *****
  1964. }
  1965. type RepositoryList []*Repository
  1966. func (repos RepositoryList) loadAttributes(e *gorm.DB) error {
  1967. if len(repos) == 0 {
  1968. return nil
  1969. }
  1970. // Load owners
  1971. userSet := make(map[int64]*User)
  1972. for i := range repos {
  1973. userSet[repos[i].OwnerID] = nil
  1974. }
  1975. userIDs := make([]int64, 0, len(userSet))
  1976. for userID := range userSet {
  1977. userIDs = append(userIDs, userID)
  1978. }
  1979. users := make([]*User, 0, len(userIDs))
  1980. if err := e.Where("id > 0").Where("id IN ?", userIDs).Find(&users).Error; err != nil {
  1981. return errors.Newf("find users: %v", err)
  1982. }
  1983. for i := range users {
  1984. userSet[users[i].ID] = users[i]
  1985. }
  1986. for i := range repos {
  1987. repos[i].Owner = userSet[repos[i].OwnerID]
  1988. }
  1989. // Load base repositories
  1990. repoSet := make(map[int64]*Repository)
  1991. for i := range repos {
  1992. if repos[i].IsFork {
  1993. repoSet[repos[i].ForkID] = nil
  1994. }
  1995. }
  1996. baseIDs := make([]int64, 0, len(repoSet))
  1997. for baseID := range repoSet {
  1998. baseIDs = append(baseIDs, baseID)
  1999. }
  2000. baseRepos := make([]*Repository, 0, len(baseIDs))
  2001. if err := e.Where("id > 0").Where("id IN ?", baseIDs).Find(&baseRepos).Error; err != nil {
  2002. return errors.Newf("find base repositories: %v", err)
  2003. }
  2004. for i := range baseRepos {
  2005. repoSet[baseRepos[i].ID] = baseRepos[i]
  2006. }
  2007. for i := range repos {
  2008. if repos[i].IsFork {
  2009. repos[i].BaseRepo = repoSet[repos[i].ForkID]
  2010. }
  2011. }
  2012. return nil
  2013. }
  2014. func (repos RepositoryList) LoadAttributes() error {
  2015. return repos.loadAttributes(db)
  2016. }
  2017. type MirrorRepositoryList []*Repository
  2018. func (repos MirrorRepositoryList) loadAttributes(e *gorm.DB) error {
  2019. if len(repos) == 0 {
  2020. return nil
  2021. }
  2022. // Load mirrors.
  2023. repoIDs := make([]int64, 0, len(repos))
  2024. for i := range repos {
  2025. if !repos[i].IsMirror {
  2026. continue
  2027. }
  2028. repoIDs = append(repoIDs, repos[i].ID)
  2029. }
  2030. mirrors := make([]*Mirror, 0, len(repoIDs))
  2031. if err := e.Where("id > 0").Where("repo_id IN ?", repoIDs).Find(&mirrors).Error; err != nil {
  2032. return errors.Newf("find mirrors: %v", err)
  2033. }
  2034. set := make(map[int64]*Mirror)
  2035. for i := range mirrors {
  2036. set[mirrors[i].RepoID] = mirrors[i]
  2037. }
  2038. for i := range repos {
  2039. repos[i].Mirror = set[repos[i].ID]
  2040. }
  2041. return nil
  2042. }
  2043. func (repos MirrorRepositoryList) LoadAttributes() error {
  2044. return repos.loadAttributes(db)
  2045. }
  2046. // __ __ __ .__
  2047. // / \ / \_____ _/ |_ ____ | |__
  2048. // \ \/\/ /\__ \\ __\/ ___\| | \
  2049. // \ / / __ \| | \ \___| Y \
  2050. // \__/\ / (____ /__| \___ >___| /
  2051. // \/ \/ \/ \/
  2052. // Watch is connection request for receiving repository notification.
  2053. type Watch struct {
  2054. ID int64 `gorm:"primaryKey"`
  2055. UserID int64 `xorm:"UNIQUE(watch)" gorm:"uniqueIndex:watch_user_repo_unique;not null"`
  2056. RepoID int64 `xorm:"UNIQUE(watch)" gorm:"uniqueIndex:watch_user_repo_unique;not null"`
  2057. }
  2058. func isWatching(e *gorm.DB, userID, repoID int64) bool {
  2059. var watch Watch
  2060. err := e.Where("user_id = ? AND repo_id = ?", userID, repoID).First(&watch).Error
  2061. return err == nil
  2062. }
  2063. // IsWatching checks if user has watched given repository.
  2064. func IsWatching(userID, repoID int64) bool {
  2065. return isWatching(db, userID, repoID)
  2066. }
  2067. func watchRepo(e *gorm.DB, userID, repoID int64, watch bool) (err error) {
  2068. if watch {
  2069. if isWatching(e, userID, repoID) {
  2070. return nil
  2071. }
  2072. if err = e.Create(&Watch{RepoID: repoID, UserID: userID}).Error; err != nil {
  2073. return err
  2074. }
  2075. err = e.Exec("UPDATE `repository` SET num_watches = num_watches + 1 WHERE id = ?", repoID).Error
  2076. } else {
  2077. if !isWatching(e, userID, repoID) {
  2078. return nil
  2079. }
  2080. if err = e.Where("user_id = ? AND repo_id = ?", userID, repoID).Delete(&Watch{}).Error; err != nil {
  2081. return err
  2082. }
  2083. err = e.Exec("UPDATE `repository` SET num_watches = num_watches - 1 WHERE id = ?", repoID).Error
  2084. }
  2085. return err
  2086. }
  2087. // Watch or unwatch repository.
  2088. //
  2089. // Deprecated: Use Watches.Watch instead.
  2090. func WatchRepo(userID, repoID int64, watch bool) (err error) {
  2091. return watchRepo(db, userID, repoID, watch)
  2092. }
  2093. // Deprecated: Use Repos.ListByRepo instead.
  2094. func getWatchers(e *gorm.DB, repoID int64) ([]*Watch, error) {
  2095. watches := make([]*Watch, 0, 10)
  2096. return watches, e.Where("repo_id = ?", repoID).Find(&watches).Error
  2097. }
  2098. // GetWatchers returns all watchers of given repository.
  2099. //
  2100. // Deprecated: Use Repos.ListByRepo instead.
  2101. func GetWatchers(repoID int64) ([]*Watch, error) {
  2102. return getWatchers(db, repoID)
  2103. }
  2104. // Repository.GetWatchers returns range of users watching given repository.
  2105. func (r *Repository) GetWatchers(page int) ([]*User, error) {
  2106. users := make([]*User, 0, ItemsPerPage)
  2107. query := db.Limit(ItemsPerPage).Offset((page-1)*ItemsPerPage).Where("watch.repo_id = ?", r.ID)
  2108. if conf.UsePostgreSQL {
  2109. query = query.Table(`"user"`).Joins("LEFT JOIN watch ON \"user\".id = watch.user_id")
  2110. } else {
  2111. query = query.Table("user").Joins("LEFT JOIN watch ON user.id = watch.user_id")
  2112. }
  2113. return users, query.Find(&users).Error
  2114. }
  2115. // Deprecated: Use Actions.notifyWatchers instead.
  2116. func notifyWatchers(e *gorm.DB, act *Action) error {
  2117. if act.CreatedUnix <= 0 {
  2118. act.CreatedUnix = time.Now().Unix()
  2119. }
  2120. // Add feeds for user self and all watchers.
  2121. watchers, err := getWatchers(e, act.RepoID)
  2122. if err != nil {
  2123. return errors.Newf("getWatchers: %v", err)
  2124. }
  2125. // Reset ID to reuse Action object
  2126. act.ID = 0
  2127. // Add feed for actioner.
  2128. act.UserID = act.ActUserID
  2129. if err = e.Create(act).Error; err != nil {
  2130. return errors.Newf("insert new action: %v", err)
  2131. }
  2132. for i := range watchers {
  2133. if act.ActUserID == watchers[i].UserID {
  2134. continue
  2135. }
  2136. act.ID = 0
  2137. act.UserID = watchers[i].UserID
  2138. if err = e.Create(act).Error; err != nil {
  2139. return errors.Newf("insert new action: %v", err)
  2140. }
  2141. }
  2142. return nil
  2143. }
  2144. // NotifyWatchers creates batch of actions for every watcher.
  2145. //
  2146. // Deprecated: Use Actions.notifyWatchers instead.
  2147. func NotifyWatchers(act *Action) error {
  2148. return notifyWatchers(db, act)
  2149. }
  2150. // _________ __
  2151. // / _____// |______ _______
  2152. // \_____ \\ __\__ \\_ __ \
  2153. // / \| | / __ \| | \/
  2154. // /_______ /|__| (____ /__|
  2155. // \/ \/
  2156. type Star struct {
  2157. ID int64 `gorm:"primaryKey"`
  2158. UserID int64 `xorm:"uid UNIQUE(s)" gorm:"column:uid;uniqueIndex:star_user_repo_unique;not null"`
  2159. RepoID int64 `xorm:"UNIQUE(s)" gorm:"uniqueIndex:star_user_repo_unique;not null"`
  2160. }
  2161. // Star or unstar repository.
  2162. //
  2163. // Deprecated: Use Stars.Star instead.
  2164. func StarRepo(userID, repoID int64, star bool) (err error) {
  2165. if star {
  2166. if IsStaring(userID, repoID) {
  2167. return nil
  2168. }
  2169. if err = db.Create(&Star{UserID: userID, RepoID: repoID}).Error; err != nil {
  2170. return err
  2171. } else if err = db.Exec("UPDATE `repository` SET num_stars = num_stars + 1 WHERE id = ?", repoID).Error; err != nil {
  2172. return err
  2173. }
  2174. err = db.Exec("UPDATE `user` SET num_stars = num_stars + 1 WHERE id = ?", userID).Error
  2175. } else {
  2176. if !IsStaring(userID, repoID) {
  2177. return nil
  2178. }
  2179. if err = db.Where("uid = ? AND repo_id = ?", userID, repoID).Delete(&Star{}).Error; err != nil {
  2180. return err
  2181. } else if err = db.Exec("UPDATE `repository` SET num_stars = num_stars - 1 WHERE id = ?", repoID).Error; err != nil {
  2182. return err
  2183. }
  2184. err = db.Exec("UPDATE `user` SET num_stars = num_stars - 1 WHERE id = ?", userID).Error
  2185. }
  2186. return err
  2187. }
  2188. // IsStaring checks if user has starred given repository.
  2189. func IsStaring(userID, repoID int64) bool {
  2190. var star Star
  2191. err := db.Where("uid = ? AND repo_id = ?", userID, repoID).First(&star).Error
  2192. return err == nil
  2193. }
  2194. func (r *Repository) GetStargazers(page int) ([]*User, error) {
  2195. users := make([]*User, 0, ItemsPerPage)
  2196. query := db.Limit(ItemsPerPage).Offset((page-1)*ItemsPerPage).Where("star.repo_id = ?", r.ID)
  2197. if conf.UsePostgreSQL {
  2198. query = query.Table(`"user"`).Joins("LEFT JOIN star ON \"user\".id = star.uid")
  2199. } else {
  2200. query = query.Table("user").Joins("LEFT JOIN star ON user.id = star.uid")
  2201. }
  2202. return users, query.Find(&users).Error
  2203. }
  2204. // ___________ __
  2205. // \_ _____/__________| | __
  2206. // | __)/ _ \_ __ \ |/ /
  2207. // | \( <_> ) | \/ <
  2208. // \___ / \____/|__| |__|_ \
  2209. // \/ \/
  2210. // HasForkedRepo checks if given user has already forked a repository.
  2211. // When user has already forked, it returns true along with the repository.
  2212. func HasForkedRepo(ownerID, repoID int64) (*Repository, bool, error) {
  2213. repo := new(Repository)
  2214. err := db.Where("owner_id = ? AND fork_id = ?", ownerID, repoID).First(repo).Error
  2215. if err != nil {
  2216. if err == gorm.ErrRecordNotFound {
  2217. return nil, false, nil
  2218. }
  2219. return nil, false, err
  2220. }
  2221. return repo, true, repo.LoadAttributes()
  2222. }
  2223. // ForkRepository creates a fork of target repository under another user domain.
  2224. func ForkRepository(doer, owner *User, baseRepo *Repository, name, desc string) (_ *Repository, err error) {
  2225. if !owner.canCreateRepo() {
  2226. return nil, ErrReachLimitOfRepo{Limit: owner.maxNumRepos()}
  2227. }
  2228. repo := &Repository{
  2229. OwnerID: owner.ID,
  2230. Owner: owner,
  2231. Name: name,
  2232. LowerName: strings.ToLower(name),
  2233. Description: desc,
  2234. DefaultBranch: baseRepo.DefaultBranch,
  2235. IsPrivate: baseRepo.IsPrivate,
  2236. IsUnlisted: baseRepo.IsUnlisted,
  2237. IsFork: true,
  2238. ForkID: baseRepo.ID,
  2239. }
  2240. err = db.Transaction(func(tx *gorm.DB) error {
  2241. if err := createRepository(tx, doer, owner, repo); err != nil {
  2242. return err
  2243. } else if err := tx.Exec("UPDATE `repository` SET num_forks=num_forks+1 WHERE id=?", baseRepo.ID).Error; err != nil {
  2244. return err
  2245. }
  2246. repoPath := repo.repoPath(tx)
  2247. RemoveAllWithNotice("Repository path erase before creation", repoPath)
  2248. _, stderr, err := process.ExecTimeout(10*time.Minute,
  2249. fmt.Sprintf("ForkRepository 'git clone': %s/%s", owner.Name, repo.Name),
  2250. "git", "clone", "--bare", baseRepo.RepoPath(), repoPath)
  2251. if err != nil {
  2252. return errors.Newf("git clone: %v - %s", err, stderr)
  2253. }
  2254. _, stderr, err = process.ExecDir(-1,
  2255. repoPath, fmt.Sprintf("ForkRepository 'git update-server-info': %s", repoPath),
  2256. "git", "update-server-info")
  2257. if err != nil {
  2258. return errors.Newf("git update-server-info: %v - %s", err, stderr)
  2259. }
  2260. if err = createDelegateHooks(repoPath); err != nil {
  2261. return errors.Newf("createDelegateHooks: %v", err)
  2262. }
  2263. return nil
  2264. })
  2265. if err != nil {
  2266. return nil, errors.Newf("commit: %v", err)
  2267. }
  2268. // Remember visibility preference
  2269. err = Handle.Users().Update(context.TODO(), owner.ID, UpdateUserOptions{LastRepoVisibility: &repo.IsPrivate})
  2270. if err != nil {
  2271. return nil, errors.Wrap(err, "update user")
  2272. }
  2273. if err = repo.UpdateSize(); err != nil {
  2274. log.Error("UpdateSize [repo_id: %d]: %v", repo.ID, err)
  2275. }
  2276. if err = PrepareWebhooks(baseRepo, HookEventTypeFork, &api.ForkPayload{
  2277. Forkee: repo.APIFormatLegacy(nil),
  2278. Repo: baseRepo.APIFormatLegacy(nil),
  2279. Sender: doer.APIFormat(),
  2280. }); err != nil {
  2281. log.Error("PrepareWebhooks [repo_id: %d]: %v", baseRepo.ID, err)
  2282. }
  2283. return repo, nil
  2284. }
  2285. func (r *Repository) GetForks() ([]*Repository, error) {
  2286. forks := make([]*Repository, 0, r.NumForks)
  2287. if err := db.Where("fork_id = ?", r.ID).Find(&forks).Error; err != nil {
  2288. return nil, err
  2289. }
  2290. for _, fork := range forks {
  2291. fork.BaseRepo = r
  2292. }
  2293. return forks, nil
  2294. }
  2295. // __________ .__
  2296. // \______ \____________ ____ ____ | |__
  2297. // | | _/\_ __ \__ \ / \_/ ___\| | \
  2298. // | | \ | | \// __ \| | \ \___| Y \
  2299. // |______ / |__| (____ /___| /\___ >___| /
  2300. // \/ \/ \/ \/ \/
  2301. //
  2302. func (r *Repository) CreateNewBranch(oldBranch, newBranch string) (err error) {
  2303. repoWorkingPool.CheckIn(com.ToStr(r.ID))
  2304. defer repoWorkingPool.CheckOut(com.ToStr(r.ID))
  2305. localPath := r.LocalCopyPath()
  2306. if err = discardLocalRepoBranchChanges(localPath, oldBranch); err != nil {
  2307. return errors.Newf("discard changes in local copy [path: %s, branch: %s]: %v", localPath, oldBranch, err)
  2308. } else if err = r.UpdateLocalCopyBranch(oldBranch); err != nil {
  2309. return errors.Newf("update branch for local copy [path: %s, branch: %s]: %v", localPath, oldBranch, err)
  2310. }
  2311. if err = r.CheckoutNewBranch(oldBranch, newBranch); err != nil {
  2312. return errors.Newf("create new branch [base: %s, new: %s]: %v", oldBranch, newBranch, err)
  2313. }
  2314. if err = git.Push(localPath, "origin", newBranch); err != nil {
  2315. return errors.Newf("push [branch: %s]: %v", newBranch, err)
  2316. }
  2317. return nil
  2318. }
  2319. // Deprecated: Use Perms.SetRepoPerms instead.
  2320. func (r *Repository) refreshAccesses(e *gorm.DB, accessMap map[int64]AccessMode) (err error) {
  2321. newAccesses := make([]Access, 0, len(accessMap))
  2322. for userID, mode := range accessMap {
  2323. newAccesses = append(newAccesses, Access{
  2324. UserID: userID,
  2325. RepoID: r.ID,
  2326. Mode: mode,
  2327. })
  2328. }
  2329. // Delete old accesses and insert new ones for repository.
  2330. if err = e.Where("repo_id = ?", r.ID).Delete(&Access{}).Error; err != nil {
  2331. return errors.Newf("delete old accesses: %v", err)
  2332. } else if err = e.Create(newAccesses).Error; err != nil {
  2333. return errors.Newf("insert new accesses: %v", err)
  2334. }
  2335. return nil
  2336. }
  2337. // refreshCollaboratorAccesses retrieves repository collaborations with their access modes.
  2338. func (r *Repository) refreshCollaboratorAccesses(e *gorm.DB, accessMap map[int64]AccessMode) error {
  2339. collaborations, err := r.getCollaborations(e)
  2340. if err != nil {
  2341. return errors.Newf("getCollaborations: %v", err)
  2342. }
  2343. for _, c := range collaborations {
  2344. accessMap[c.UserID] = c.Mode
  2345. }
  2346. return nil
  2347. }
  2348. // recalculateTeamAccesses recalculates new accesses for teams of an organization
  2349. // except the team whose ID is given. It is used to assign a team ID when
  2350. // remove repository from that team.
  2351. func (r *Repository) recalculateTeamAccesses(e *gorm.DB, ignTeamID int64) (err error) {
  2352. accessMap := make(map[int64]AccessMode, 20)
  2353. if err = r.getOwner(e); err != nil {
  2354. return err
  2355. } else if !r.Owner.IsOrganization() {
  2356. return errors.Newf("owner is not an organization: %d", r.OwnerID)
  2357. }
  2358. if err = r.refreshCollaboratorAccesses(e, accessMap); err != nil {
  2359. return errors.Newf("refreshCollaboratorAccesses: %v", err)
  2360. }
  2361. if err = r.Owner.getTeams(e); err != nil {
  2362. return err
  2363. }
  2364. maxAccessMode := func(modes ...AccessMode) AccessMode {
  2365. max := AccessModeNone
  2366. for _, mode := range modes {
  2367. if mode > max {
  2368. max = mode
  2369. }
  2370. }
  2371. return max
  2372. }
  2373. for _, t := range r.Owner.Teams {
  2374. if t.ID == ignTeamID {
  2375. continue
  2376. }
  2377. // Owner team gets owner access, and skip for teams that do not
  2378. // have relations with repository.
  2379. if t.IsOwnerTeam() {
  2380. t.Authorize = AccessModeOwner
  2381. } else if !t.hasRepository(e, r.ID) {
  2382. continue
  2383. }
  2384. if err = t.getMembers(e); err != nil {
  2385. return errors.Newf("getMembers '%d': %v", t.ID, err)
  2386. }
  2387. for _, m := range t.Members {
  2388. accessMap[m.ID] = maxAccessMode(accessMap[m.ID], t.Authorize)
  2389. }
  2390. }
  2391. return r.refreshAccesses(e, accessMap)
  2392. }
  2393. func (r *Repository) recalculateAccesses(e *gorm.DB) error {
  2394. if r.Owner.IsOrganization() {
  2395. return r.recalculateTeamAccesses(e, 0)
  2396. }
  2397. accessMap := make(map[int64]AccessMode, 10)
  2398. if err := r.refreshCollaboratorAccesses(e, accessMap); err != nil {
  2399. return errors.Newf("refreshCollaboratorAccesses: %v", err)
  2400. }
  2401. return r.refreshAccesses(e, accessMap)
  2402. }
  2403. // RecalculateAccesses recalculates all accesses for repository.
  2404. func (r *Repository) RecalculateAccesses() error {
  2405. return r.recalculateAccesses(db)
  2406. }