action.go 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588
  1. // Copyright 2014 The Gogs Authors. All rights reserved.
  2. // Use of this source code is governed by a MIT-style
  3. // license that can be found in the LICENSE file.
  4. package models
  5. import (
  6. "encoding/json"
  7. "errors"
  8. "fmt"
  9. "path"
  10. "regexp"
  11. "strings"
  12. "time"
  13. "unicode"
  14. "github.com/Unknwon/com"
  15. "github.com/go-xorm/xorm"
  16. api "github.com/gogits/go-gogs-client"
  17. "github.com/gogits/gogs/modules/base"
  18. "github.com/gogits/gogs/modules/git"
  19. "github.com/gogits/gogs/modules/log"
  20. "github.com/gogits/gogs/modules/setting"
  21. )
  22. type ActionType int
  23. const (
  24. CREATE_REPO ActionType = iota + 1 // 1
  25. RENAME_REPO // 2
  26. STAR_REPO // 3
  27. FOLLOW_REPO // 4
  28. COMMIT_REPO // 5
  29. CREATE_ISSUE // 6
  30. CREATE_PULL_REQUEST // 7
  31. TRANSFER_REPO // 8
  32. PUSH_TAG // 9
  33. COMMENT_ISSUE // 10
  34. MERGE_PULL_REQUEST // 11
  35. )
  36. var (
  37. ErrNotImplemented = errors.New("Not implemented yet")
  38. )
  39. var (
  40. // Same as Github. See https://help.github.com/articles/closing-issues-via-commit-messages
  41. IssueCloseKeywords = []string{"close", "closes", "closed", "fix", "fixes", "fixed", "resolve", "resolves", "resolved"}
  42. IssueReopenKeywords = []string{"reopen", "reopens", "reopened"}
  43. IssueCloseKeywordsPat, IssueReopenKeywordsPat *regexp.Regexp
  44. IssueReferenceKeywordsPat *regexp.Regexp
  45. )
  46. func assembleKeywordsPattern(words []string) string {
  47. return fmt.Sprintf(`(?i)(?:%s) \S+`, strings.Join(words, "|"))
  48. }
  49. func init() {
  50. IssueCloseKeywordsPat = regexp.MustCompile(assembleKeywordsPattern(IssueCloseKeywords))
  51. IssueReopenKeywordsPat = regexp.MustCompile(assembleKeywordsPattern(IssueReopenKeywords))
  52. IssueReferenceKeywordsPat = regexp.MustCompile(`(?i)(?:)(^| )\S+`)
  53. }
  54. // Action represents user operation type and other information to repository.,
  55. // it implemented interface base.Actioner so that can be used in template render.
  56. type Action struct {
  57. ID int64 `xorm:"pk autoincr"`
  58. UserID int64 // Receiver user id.
  59. OpType ActionType
  60. ActUserID int64 // Action user id.
  61. ActUserName string // Action user name.
  62. ActEmail string
  63. ActAvatar string `xorm:"-"`
  64. RepoID int64
  65. RepoUserName string
  66. RepoName string
  67. RefName string
  68. IsPrivate bool `xorm:"NOT NULL DEFAULT false"`
  69. Content string `xorm:"TEXT"`
  70. Created time.Time `xorm:"created"`
  71. }
  72. func (a *Action) AfterSet(colName string, _ xorm.Cell) {
  73. switch colName {
  74. case "created":
  75. a.Created = regulateTimeZone(a.Created)
  76. }
  77. }
  78. func (a Action) GetOpType() int {
  79. return int(a.OpType)
  80. }
  81. func (a Action) GetActUserName() string {
  82. return a.ActUserName
  83. }
  84. func (a Action) GetActEmail() string {
  85. return a.ActEmail
  86. }
  87. func (a Action) GetRepoUserName() string {
  88. return a.RepoUserName
  89. }
  90. func (a Action) GetRepoName() string {
  91. return a.RepoName
  92. }
  93. func (a Action) GetRepoPath() string {
  94. return path.Join(a.RepoUserName, a.RepoName)
  95. }
  96. func (a Action) GetRepoLink() string {
  97. if len(setting.AppSubUrl) > 0 {
  98. return path.Join(setting.AppSubUrl, a.GetRepoPath())
  99. }
  100. return "/" + a.GetRepoPath()
  101. }
  102. func (a Action) GetBranch() string {
  103. return a.RefName
  104. }
  105. func (a Action) GetContent() string {
  106. return a.Content
  107. }
  108. func (a Action) GetCreate() time.Time {
  109. return a.Created
  110. }
  111. func (a Action) GetIssueInfos() []string {
  112. return strings.SplitN(a.Content, "|", 2)
  113. }
  114. func (a Action) GetIssueTitle() string {
  115. index := com.StrTo(a.GetIssueInfos()[0]).MustInt64()
  116. issue, err := GetIssueByIndex(a.RepoID, index)
  117. if err != nil {
  118. log.Error(4, "GetIssueByIndex: %v", err)
  119. return "500 when get issue"
  120. }
  121. return issue.Name
  122. }
  123. func (a Action) GetIssueContent() string {
  124. index := com.StrTo(a.GetIssueInfos()[0]).MustInt64()
  125. issue, err := GetIssueByIndex(a.RepoID, index)
  126. if err != nil {
  127. log.Error(4, "GetIssueByIndex: %v", err)
  128. return "500 when get issue"
  129. }
  130. return issue.Content
  131. }
  132. func newRepoAction(e Engine, u *User, repo *Repository) (err error) {
  133. if err = notifyWatchers(e, &Action{
  134. ActUserID: u.Id,
  135. ActUserName: u.Name,
  136. ActEmail: u.Email,
  137. OpType: CREATE_REPO,
  138. RepoID: repo.ID,
  139. RepoUserName: repo.Owner.Name,
  140. RepoName: repo.Name,
  141. IsPrivate: repo.IsPrivate,
  142. }); err != nil {
  143. return fmt.Errorf("notify watchers '%d/%d': %v", u.Id, repo.ID, err)
  144. }
  145. log.Trace("action.newRepoAction: %s/%s", u.Name, repo.Name)
  146. return err
  147. }
  148. // NewRepoAction adds new action for creating repository.
  149. func NewRepoAction(u *User, repo *Repository) (err error) {
  150. return newRepoAction(x, u, repo)
  151. }
  152. func renameRepoAction(e Engine, actUser *User, oldRepoName string, repo *Repository) (err error) {
  153. if err = notifyWatchers(e, &Action{
  154. ActUserID: actUser.Id,
  155. ActUserName: actUser.Name,
  156. ActEmail: actUser.Email,
  157. OpType: RENAME_REPO,
  158. RepoID: repo.ID,
  159. RepoUserName: repo.Owner.Name,
  160. RepoName: repo.Name,
  161. IsPrivate: repo.IsPrivate,
  162. Content: oldRepoName,
  163. }); err != nil {
  164. return fmt.Errorf("notify watchers: %v", err)
  165. }
  166. log.Trace("action.renameRepoAction: %s/%s", actUser.Name, repo.Name)
  167. return nil
  168. }
  169. // RenameRepoAction adds new action for renaming a repository.
  170. func RenameRepoAction(actUser *User, oldRepoName string, repo *Repository) error {
  171. return renameRepoAction(x, actUser, oldRepoName, repo)
  172. }
  173. func issueIndexTrimRight(c rune) bool {
  174. return !unicode.IsDigit(c)
  175. }
  176. type PushCommit struct {
  177. Sha1 string
  178. Message string
  179. AuthorEmail string
  180. AuthorName string
  181. }
  182. type PushCommits struct {
  183. Len int
  184. Commits []*PushCommit
  185. CompareUrl string
  186. avatars map[string]string
  187. }
  188. func NewPushCommits() *PushCommits {
  189. return &PushCommits{
  190. avatars: make(map[string]string),
  191. }
  192. }
  193. // AvatarLink tries to match user in database with e-mail
  194. // in order to show custom avatar, and falls back to general avatar link.
  195. func (push *PushCommits) AvatarLink(email string) string {
  196. _, ok := push.avatars[email]
  197. if !ok {
  198. u, err := GetUserByEmail(email)
  199. if err != nil {
  200. push.avatars[email] = base.AvatarLink(email)
  201. if !IsErrUserNotExist(err) {
  202. log.Error(4, "GetUserByEmail: %v", err)
  203. }
  204. } else {
  205. push.avatars[email] = u.AvatarLink()
  206. }
  207. }
  208. return push.avatars[email]
  209. }
  210. // updateIssuesCommit checks if issues are manipulated by commit message.
  211. func updateIssuesCommit(u *User, repo *Repository, repoUserName, repoName string, commits []*PushCommit) error {
  212. // Commits are appended in the reverse order.
  213. for i := len(commits) - 1; i >= 0; i-- {
  214. c := commits[i]
  215. refMarked := make(map[int64]bool)
  216. for _, ref := range IssueReferenceKeywordsPat.FindAllString(c.Message, -1) {
  217. ref = ref[strings.IndexByte(ref, byte(' '))+1:]
  218. ref = strings.TrimRightFunc(ref, issueIndexTrimRight)
  219. if len(ref) == 0 {
  220. continue
  221. }
  222. // Add repo name if missing
  223. if ref[0] == '#' {
  224. ref = fmt.Sprintf("%s/%s%s", repoUserName, repoName, ref)
  225. } else if !strings.Contains(ref, "/") {
  226. // FIXME: We don't support User#ID syntax yet
  227. // return ErrNotImplemented
  228. continue
  229. }
  230. issue, err := GetIssueByRef(ref)
  231. if err != nil {
  232. if IsErrIssueNotExist(err) {
  233. continue
  234. }
  235. return err
  236. }
  237. if refMarked[issue.ID] {
  238. continue
  239. }
  240. refMarked[issue.ID] = true
  241. url := fmt.Sprintf("%s/%s/%s/commit/%s", setting.AppSubUrl, repoUserName, repoName, c.Sha1)
  242. message := fmt.Sprintf(`<a href="%s">%s</a>`, url, c.Message)
  243. if err = CreateRefComment(u, repo, issue, message, c.Sha1); err != nil {
  244. return err
  245. }
  246. }
  247. refMarked = make(map[int64]bool)
  248. // FIXME: can merge this one and next one to a common function.
  249. for _, ref := range IssueCloseKeywordsPat.FindAllString(c.Message, -1) {
  250. ref = ref[strings.IndexByte(ref, byte(' '))+1:]
  251. ref = strings.TrimRightFunc(ref, issueIndexTrimRight)
  252. if len(ref) == 0 {
  253. continue
  254. }
  255. // Add repo name if missing
  256. if ref[0] == '#' {
  257. ref = fmt.Sprintf("%s/%s%s", repoUserName, repoName, ref)
  258. } else if !strings.Contains(ref, "/") {
  259. // We don't support User#ID syntax yet
  260. // return ErrNotImplemented
  261. continue
  262. }
  263. issue, err := GetIssueByRef(ref)
  264. if err != nil {
  265. if IsErrIssueNotExist(err) {
  266. continue
  267. }
  268. return err
  269. }
  270. if refMarked[issue.ID] {
  271. continue
  272. }
  273. refMarked[issue.ID] = true
  274. if issue.RepoID != repo.ID || issue.IsClosed {
  275. continue
  276. }
  277. if err = issue.ChangeStatus(u, true); err != nil {
  278. return err
  279. }
  280. }
  281. // It is conflict to have close and reopen at same time, so refsMarkd doesn't need to reinit here.
  282. for _, ref := range IssueReopenKeywordsPat.FindAllString(c.Message, -1) {
  283. ref = ref[strings.IndexByte(ref, byte(' '))+1:]
  284. ref = strings.TrimRightFunc(ref, issueIndexTrimRight)
  285. if len(ref) == 0 {
  286. continue
  287. }
  288. // Add repo name if missing
  289. if ref[0] == '#' {
  290. ref = fmt.Sprintf("%s/%s%s", repoUserName, repoName, ref)
  291. } else if !strings.Contains(ref, "/") {
  292. // We don't support User#ID syntax yet
  293. // return ErrNotImplemented
  294. continue
  295. }
  296. issue, err := GetIssueByRef(ref)
  297. if err != nil {
  298. if IsErrIssueNotExist(err) {
  299. continue
  300. }
  301. return err
  302. }
  303. if refMarked[issue.ID] {
  304. continue
  305. }
  306. refMarked[issue.ID] = true
  307. if issue.RepoID != repo.ID || !issue.IsClosed {
  308. continue
  309. }
  310. if err = issue.ChangeStatus(u, false); err != nil {
  311. return err
  312. }
  313. }
  314. }
  315. return nil
  316. }
  317. // CommitRepoAction adds new action for committing repository.
  318. func CommitRepoAction(
  319. userID, repoUserID int64,
  320. userName, actEmail string,
  321. repoID int64,
  322. repoUserName, repoName string,
  323. refFullName string,
  324. commit *PushCommits,
  325. oldCommitID string, newCommitID string) error {
  326. u, err := GetUserByID(userID)
  327. if err != nil {
  328. return fmt.Errorf("GetUserByID: %v", err)
  329. }
  330. repo, err := GetRepositoryByName(repoUserID, repoName)
  331. if err != nil {
  332. return fmt.Errorf("GetRepositoryByName: %v", err)
  333. } else if err = repo.GetOwner(); err != nil {
  334. return fmt.Errorf("GetOwner: %v", err)
  335. }
  336. // Change repository bare status and update last updated time.
  337. repo.IsBare = false
  338. if err = UpdateRepository(repo, false); err != nil {
  339. return fmt.Errorf("UpdateRepository: %v", err)
  340. }
  341. isNewBranch := false
  342. opType := COMMIT_REPO
  343. // Check it's tag push or branch.
  344. if strings.HasPrefix(refFullName, "refs/tags/") {
  345. opType = PUSH_TAG
  346. commit = &PushCommits{}
  347. } else {
  348. // if not the first commit, set the compareUrl
  349. if !strings.HasPrefix(oldCommitID, "0000000") {
  350. commit.CompareUrl = fmt.Sprintf("%s/%s/compare/%s...%s", repoUserName, repoName, oldCommitID, newCommitID)
  351. } else {
  352. isNewBranch = true
  353. }
  354. // NOTE: limit to detect latest 100 commits.
  355. if len(commit.Commits) > 100 {
  356. commit.Commits = commit.Commits[len(commit.Commits)-100:]
  357. }
  358. if err = updateIssuesCommit(u, repo, repoUserName, repoName, commit.Commits); err != nil {
  359. log.Error(4, "updateIssuesCommit: %v", err)
  360. }
  361. }
  362. if len(commit.Commits) > setting.FeedMaxCommitNum {
  363. commit.Commits = commit.Commits[:setting.FeedMaxCommitNum]
  364. }
  365. bs, err := json.Marshal(commit)
  366. if err != nil {
  367. return fmt.Errorf("Marshal: %v", err)
  368. }
  369. refName := git.RefEndName(refFullName)
  370. if err = NotifyWatchers(&Action{
  371. ActUserID: u.Id,
  372. ActUserName: userName,
  373. ActEmail: actEmail,
  374. OpType: opType,
  375. Content: string(bs),
  376. RepoID: repo.ID,
  377. RepoUserName: repoUserName,
  378. RepoName: repoName,
  379. RefName: refName,
  380. IsPrivate: repo.IsPrivate,
  381. }); err != nil {
  382. return fmt.Errorf("NotifyWatchers: %v", err)
  383. }
  384. payloadRepo := repo.ComposePayload()
  385. pusher_email, pusher_name := "", ""
  386. pusher, err := GetUserByName(userName)
  387. if err == nil {
  388. pusher_email = pusher.Email
  389. pusher_name = pusher.DisplayName()
  390. }
  391. payloadSender := &api.PayloadUser{
  392. UserName: pusher.Name,
  393. ID: pusher.Id,
  394. AvatarUrl: setting.AppUrl + pusher.RelAvatarLink(),
  395. }
  396. switch opType {
  397. case COMMIT_REPO: // Push
  398. commits := make([]*api.PayloadCommit, len(commit.Commits))
  399. for i, cmt := range commit.Commits {
  400. author_username := ""
  401. author, err := GetUserByEmail(cmt.AuthorEmail)
  402. if err == nil {
  403. author_username = author.Name
  404. }
  405. commits[i] = &api.PayloadCommit{
  406. ID: cmt.Sha1,
  407. Message: cmt.Message,
  408. URL: fmt.Sprintf("%s/commit/%s", repo.FullRepoLink(), cmt.Sha1),
  409. Author: &api.PayloadAuthor{
  410. Name: cmt.AuthorName,
  411. Email: cmt.AuthorEmail,
  412. UserName: author_username,
  413. },
  414. }
  415. }
  416. p := &api.PushPayload{
  417. Ref: refFullName,
  418. Before: oldCommitID,
  419. After: newCommitID,
  420. CompareUrl: setting.AppUrl + commit.CompareUrl,
  421. Commits: commits,
  422. Repo: payloadRepo,
  423. Pusher: &api.PayloadAuthor{
  424. Name: pusher_name,
  425. Email: pusher_email,
  426. UserName: userName,
  427. },
  428. Sender: payloadSender,
  429. }
  430. if err = PrepareWebhooks(repo, HOOK_EVENT_PUSH, p); err != nil {
  431. return fmt.Errorf("PrepareWebhooks: %v", err)
  432. }
  433. if isNewBranch {
  434. return PrepareWebhooks(repo, HOOK_EVENT_CREATE, &api.CreatePayload{
  435. Ref: refName,
  436. RefType: "branch",
  437. Repo: payloadRepo,
  438. Sender: payloadSender,
  439. })
  440. }
  441. case PUSH_TAG: // Create
  442. return PrepareWebhooks(repo, HOOK_EVENT_CREATE, &api.CreatePayload{
  443. Ref: refName,
  444. RefType: "tag",
  445. Repo: payloadRepo,
  446. Sender: payloadSender,
  447. })
  448. }
  449. return nil
  450. }
  451. func transferRepoAction(e Engine, actUser, oldOwner, newOwner *User, repo *Repository) (err error) {
  452. if err = notifyWatchers(e, &Action{
  453. ActUserID: actUser.Id,
  454. ActUserName: actUser.Name,
  455. ActEmail: actUser.Email,
  456. OpType: TRANSFER_REPO,
  457. RepoID: repo.ID,
  458. RepoUserName: newOwner.Name,
  459. RepoName: repo.Name,
  460. IsPrivate: repo.IsPrivate,
  461. Content: path.Join(oldOwner.LowerName, repo.LowerName),
  462. }); err != nil {
  463. return fmt.Errorf("notify watchers '%d/%d': %v", actUser.Id, repo.ID, err)
  464. }
  465. // Remove watch for organization.
  466. if repo.Owner.IsOrganization() {
  467. if err = watchRepo(e, repo.Owner.Id, repo.ID, false); err != nil {
  468. return fmt.Errorf("watch repository: %v", err)
  469. }
  470. }
  471. log.Trace("action.transferRepoAction: %s/%s", actUser.Name, repo.Name)
  472. return nil
  473. }
  474. // TransferRepoAction adds new action for transferring repository.
  475. func TransferRepoAction(actUser, oldOwner, newOwner *User, repo *Repository) error {
  476. return transferRepoAction(x, actUser, oldOwner, newOwner, repo)
  477. }
  478. func mergePullRequestAction(e Engine, actUser *User, repo *Repository, pull *Issue) error {
  479. return notifyWatchers(e, &Action{
  480. ActUserID: actUser.Id,
  481. ActUserName: actUser.Name,
  482. ActEmail: actUser.Email,
  483. OpType: MERGE_PULL_REQUEST,
  484. Content: fmt.Sprintf("%d|%s", pull.Index, pull.Name),
  485. RepoID: repo.ID,
  486. RepoUserName: repo.Owner.Name,
  487. RepoName: repo.Name,
  488. IsPrivate: repo.IsPrivate,
  489. })
  490. }
  491. // MergePullRequestAction adds new action for merging pull request.
  492. func MergePullRequestAction(actUser *User, repo *Repository, pull *Issue) error {
  493. return mergePullRequestAction(x, actUser, repo, pull)
  494. }
  495. // GetFeeds returns action list of given user in given context.
  496. func GetFeeds(uid, offset int64, isProfile bool) ([]*Action, error) {
  497. actions := make([]*Action, 0, 20)
  498. sess := x.Limit(20, int(offset)).Desc("id").Where("user_id=?", uid)
  499. if isProfile {
  500. sess.And("is_private=?", false).And("act_user_id=?", uid)
  501. }
  502. err := sess.Find(&actions)
  503. return actions, err
  504. }