Skip to content

Modify the code in server.go, and be called in rule-handler #1

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Modify the code in server.go, and be called in rule-handler #1

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
125 changes: 66 additions & 59 deletions src/server.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -57,12 +57,18 @@ func getToken() []byte {
}

func ServeHTTP(w http.ResponseWriter, r *http.Request) {
fmt.Fprint(w, "Version: 0.55 \n")
eventType, _, payload, ok, _ := gitee_utils.ValidateWebhook(w, r)
if !ok {
defer r.Body.Close()
payload, err := ioutil.ReadAll(r.Body)
if err != nil {
return
}

// fmt.Println(string(payload))

eventType := r.Header.Get("event_type")

var resp []string

switch eventType {
case "Issue Hook":
var ie gitee.IssueEvent
Expand All @@ -72,13 +78,15 @@ func ServeHTTP(w http.ResponseWriter, r *http.Request) {
if err := checkRepository(payload, ie.Repository); err != nil {
return
}
go handleIssueEvent(&ie)
resp = handleIssueEvent(&ie)
responseJson(w, resp)
case "Note Hook":
var ic gitee.NoteEvent
if err := json.Unmarshal(payload, &ic); err != nil {
return
}
go handleCommentEvent(&ic)
resp = handleCommentEvent(&ic)
responseJson(w, resp)
//case "Merge Request Hook":
//var ip gitee.PullRequestEvent
//if err := json.Unmarshal(payload, &ip); err != nil {
Expand All @@ -87,12 +95,14 @@ func ServeHTTP(w http.ResponseWriter, r *http.Request) {
//go handlePullRequestEvent(&ip)
default:
return

}
}

func handleIssueEvent(i *gitee.IssueEvent) {
func handleIssueEvent(i *gitee.IssueEvent) []string {
var resp []string
if *(i.Action) != "open" {
return
return resp
}
assignee := ""
strLabels := ""
Expand All @@ -118,11 +128,7 @@ func handleIssueEvent(i *gitee.IssueEvent) {
c := gitee_utils.NewClient(getToken)

if len(issueInit) == 0 {
res := c.CreateGiteeIssueComment(org, repo, issueNum, issueTemp)
if res != nil {
fmt.Println(res.Error())
return
}
resp = append(resp, issueTemp)
var labelsToAdd []string

labelMatches := labelRegex.FindAllStringSubmatch(issueBody, -1)
Expand Down Expand Up @@ -180,7 +186,7 @@ func handleIssueEvent(i *gitee.IssueEvent) {
rese := c.AssignGiteeIssue(org, repo, labelsToAdd_str, issueNum, assignee)
if rese != nil {
fmt.Println(rese.Error())
return
return resp
}

issueBody = strings.Replace(issueBody, " ", "", -1)
Expand Down Expand Up @@ -237,19 +243,11 @@ func handleIssueEvent(i *gitee.IssueEvent) {

helloWord = strings.Replace(helloWord, "{"+"label"+"}", fmt.Sprintf("%v", strLabels), -1)

resLabel := c.CreateGiteeIssueComment(org, repo, issueNum, helloWord)
if resLabel != nil {
fmt.Println(resLabel.Error())
return
}
resp = append(resp, helloWord)
}
} else {

res := c.CreateGiteeIssueComment(org, repo, issueNum, issueTemp)
if res != nil {
fmt.Println(res.Error())
return
}
resp = append(resp, issueTemp)

var labelFindTemp []string
for _, label := range issueInit {
Expand All @@ -265,7 +263,7 @@ func handleIssueEvent(i *gitee.IssueEvent) {
rese := c.AssignGiteeIssue(org, repo, "", issueNum, assignee)
if rese != nil {
fmt.Println(rese.Error())
return
return nil
}
} else {
assignTemp := string(assignComment[:])
Expand All @@ -277,11 +275,8 @@ func handleIssueEvent(i *gitee.IssueEvent) {
} else {
assignTemp = strings.Replace(assignTemp, "@{"+"assignee"+"}", fmt.Sprintf("%v", "自己"), -1)
}
rs := c.CreateGiteeIssueComment(org, repo, issueNum, assignTemp)
if rs != nil {
fmt.Println(res.Error())
return
}

resp = append(resp, assignTemp)
}
}

Expand All @@ -292,29 +287,31 @@ func handleIssueEvent(i *gitee.IssueEvent) {
assigneeStr = ""
}
Temp := "hello, @" + issueMaker + assigneeStr + " " + decisionTemp + "\n"
res := c.CreateGiteeIssueComment(org, repo, issueNum, Temp)
if res != nil {
fmt.Println(res.Error())
return
}

resp = append(resp, Temp)
}
}
return resp
}

func handleCommentEvent(i *gitee.NoteEvent) {
func handleCommentEvent(i *gitee.NoteEvent) []string {
switch *(i.NoteableType) {
case "Issue":
go handleIssueCommentEvent(i)
// go handleIssueCommentEvent(i)
var resp []string = handleIssueCommentEvent(i)
return resp
//case "PullRequest":
// go handlePRCommentEvent(i)
default:
return
return nil
}
}

func handleIssueCommentEvent(i *gitee.NoteEvent) {
func handleIssueCommentEvent(i *gitee.NoteEvent) []string {
var resp []string

if *(i.Action) != "comment" {
return
return resp
}
assignee := ""
labelsToAddStr := ""
Expand All @@ -340,18 +337,15 @@ func handleIssueCommentEvent(i *gitee.NoteEvent) {
c := gitee_utils.NewClient(getToken)
labelMatches := labelRegex.FindAllStringSubmatch(noteBody, -1)
if len(labelMatches) == 0 {
return
return nil
}
var labelsToAdd []string
labelsToAdd = getLabelsFromREMatches(labelMatches)

if strings.Contains(noteBody, "good-first-issue") {
astr := "如果您是第一次贡献社区，可以参考我们的贡献指南：https://www.openeuler.org/zh/community/contribution/"
res := c.CreateGiteeIssueComment(org, repo, issueNum, astr)
if res != nil {
fmt.Println(res.Error())
return
}

resp = append(resp, astr)
}

if len(labelStrs) != 0 {
Expand All @@ -372,20 +366,18 @@ func handleIssueCommentEvent(i *gitee.NoteEvent) {
rese := c.AssignGiteeIssue(org, repo, labelsToAddStr, issueNum, assignee)
if rese != nil {
fmt.Println(rese.Error())
return
return nil
}

for _, label := range labelsToAdd {
if label == "kind/decision" {
Temp := "hello, @" + issueMaker + assigneeStr + " " + decisionTemp + "\n"
res := c.CreateGiteeIssueComment(org, repo, issueNum, Temp)
if res != nil {
fmt.Println(res.Error())
return
}

resp = append(resp, Temp)
}
}
}
return resp
}

func handlePRCommentEvent(i *gitee.NoteEvent) {
Expand Down Expand Up @@ -524,6 +516,21 @@ func getRecommendation(c gitee_utils.Client, labels []string) string {
return participants
}

func responseJson(w http.ResponseWriter, resp []string) {
resp_map := make(map[int]string)
for i, content := range resp {
resp_map[i] = content
}
resp_json, err := json.Marshal(resp_map)
if err != nil {
fmt.Println("json.Marshal failed:", err)
return
}
w.Header().Set("content-type", "text/json")
w.WriteHeader(200)
w.Write(resp_json)
}

func loadFile(path, fileType string) error {
jsonFile, err := os.Open(path)
if err != nil {
Expand Down Expand Up @@ -557,14 +564,14 @@ func loadFile(path, fileType string) error {
}

func configFile() {
loadFile("src/data/mentor.json", "json")
loadFile("src/data/issueComTemplate.md", "issue")
loadFile("src/data/decisionTemplate.md", "decision")
loadFile("src/data/prComTemplate.md", "pr")
loadFile("src/data/partiTemplate.md", "parti")
loadFile("src/data/partiTemplate_ai.md", "partiAI")
loadFile("src/data/token.md", "token")
loadFile("src/data/assignTemplate.md", "assign")
loadFile("../src/data/mentor.json", "json")
loadFile("../src/data/issueComTemplate.md", "issue")
loadFile("../src/data/decisionTemplate.md", "decision")
loadFile("../src/data/prComTemplate.md", "pr")
loadFile("../src/data/partiTemplate.md", "parti")
loadFile("../src/data/partiTemplate_ai.md", "partiAI")
loadFile("../src/data/token.md", "token")
loadFile("../src/data/assignTemplate.md", "assign")
}

func main() {
Expand Down