github.go

package githubbot

import (
	"fmt"
	"strconv"
	"time"

	"github.com/cpalone/gohook"
)

func (s *Session) hookServer(port int, secret string, sendReplyChan chan PacketEvent) {
	// spin off github server
	gServer := gohook.NewServer(port, secret, "/postreceive")
	s.logger.Info("Starting github server...")
	gServer.GoListenAndServe()

	// Wait for github hook event
	for {
		et := <-gServer.EventAndTypes
		s.logger.Infof("Received hook event of type '%s'.", et.Type)
		switch et.Type {
		case gohook.CommitCommentEventType:
			payload, ok := et.Event.(*gohook.CommitCommentEvent)
			if !ok {
				panic("Malformed *CommitCommentEvent.")
			}
			// TODO: can we get the branch here?
			msg := fmt.Sprintf("[ %s ] Comment on commit: %s (%s)",
				payload.Repository.Name,
				payload.Comment.Body,
				payload.Comment.HTMLURL,
			)
			s.sendMessage(msg, "", strconv.Itoa(s.msgID))
			s.msgID++
		case gohook.CreateEventType:
			payload, ok := et.Event.(*gohook.CreateEvent)
			if !ok {
				panic("Malformed *CreateEvent.")
			}
			msg := fmt.Sprintf("[ %s | Branch/Tag: %s] Created.",
				payload.Repository.Name,
				payload.RefType,
			)
			s.sendMessage(msg, "", strconv.Itoa(s.msgID))
			s.msgID++
		case gohook.DeleteEventType:
			payload, ok := et.Event.(*gohook.DeleteEvent)
			if !ok {
				panic("Malformed *DeleteEvent.")
			}
			msg := fmt.Sprintf("[ %s | Branch/Tag: %s] Deleted.",
				payload.Repository,
				payload.RefType,
			)
			s.sendMessage(msg, "", strconv.Itoa(s.msgID))
			s.msgID++
		case gohook.IssueCommentEventType:
			payload, ok := et.Event.(*gohook.IssueCommentEvent)
			if !ok {
				panic("Malformed *CommitCommentEvent.")
			}
			msg := fmt.Sprintf("[ %s | Issue: %s ] Comment: %s (%s)",
				payload.Repository.Name,
				payload.Issue.Title,
				payload.Comment.Body,
				payload.Comment.HTMLURL,
			)
			s.sendMessage(msg, "", strconv.Itoa(s.msgID))
			s.msgID++
		case gohook.IssuesEventType:
			payload, ok := et.Event.(*gohook.IssuesEvent)
			if !ok {
				panic("Malformed *IssuesEvent.")
			}
			msg := fmt.Sprintf("[ %s | Issue: %s ] Action: %s. (%s)",
				payload.Repository.Name,
				payload.Issue.Title,
				payload.Action,
				payload.Issue.HTMLURL,
			)
			s.sendMessage(msg, "", strconv.Itoa(s.msgID))
			s.msgID++
		case gohook.PullRequestEventType:
			payload, ok := et.Event.(*gohook.PullRequestEvent)
			if !ok {
				panic("Malformed *PullRequestEvent.")
			}
			action := payload.Action
			if action == "synchronize" {
				action = "New commits made to synced branch."
			}
			msg := fmt.Sprintf(":pencil: [ %s | PR: %s ] %s (%s)",
				payload.Repository.Name,
				payload.PullRequest.Title,
				action,
				payload.PullRequest.HTMLURL,
			)
			s.msgID++
			t := strconv.Itoa(int(time.Now().Unix()))
			s.waiting = true
			s.sendMessage(msg, "", t)
			var reply PacketEvent
			for s.waiting {
				reply = <-sendReplyChan
				if reply.ID == t {
					s.waiting = false
				}
			}
			srPayload, err := reply.Payload()
			if err != nil {
				s.logger.Fatalln(err)
			}

			// need send-reply for msgID to reply to
			data, ok := srPayload.(*SendReply)
			if !ok {
				s.logger.Fatalln("Could not assert *SendReply as such.")
			}
			s.commitParent[payload.PullRequest.Head.SHA] = data.ID
		case gohook.PullRequestReviewCommentEventType:
			payload, ok := et.Event.(*gohook.PullRequestReviewCommentEvent)
			if !ok {
				panic("Malformed *PullRequestReviewCommentEvent.")
			}
			msg := fmt.Sprintf(":speech_balloon: [ %s | PR: %s ] Comment: %s: %s (%s)",
				payload.Repository.Name,
				payload.PullRequest.Title,
				payload.Sender.Login,
				payload.Comment.Body,
				payload.PullRequest.HTMLURL,
			)
			s.sendMessage(msg, "", strconv.Itoa(s.msgID))
			s.msgID++
		case gohook.RepositoryEventType:
			payload, ok := et.Event.(*gohook.RepositoryEvent)
			if !ok {
				panic("Malformed *RepositoryEvent.")
			}
			msg := fmt.Sprintf("[ Repository: %s ] Action: created. (%s) ",
				payload.Repository.Name,
				payload.Repository.HTMLURL,
			)
			s.sendMessage(msg, "", strconv.Itoa(s.msgID))
			s.msgID++
		case gohook.PushEventType:
			s.logger.Info("Entering PushEventType case.")
			payload, ok := et.Event.(*gohook.PushEvent)
			if !ok {
				panic("Malformed *PushEvent.")
			}
			if payload.HeadCommit.Message == "" {
				continue
			}
			var msg string
			if len(payload.Commits) > 1 {
				msg = fmt.Sprintf(":repeat: [ %s | Branch: %s ] %v Commits: %s (%s)",
					payload.Repository.Name,
					payload.Ref[11:], // this discards "refs/heads/"
					len(payload.Commits),
					payload.HeadCommit.Message,
					payload.Compare,
				)
			} else {
				msg = fmt.Sprintf(":repeat: [ %s | Branch: %s ] Commit: %s (%s)",
					payload.Repository.Name,
					payload.Ref[11:], // this discards "refs/heads/"
					payload.HeadCommit.Message,
					payload.HeadCommit.URL,
				)
			}
			t := strconv.Itoa(int(time.Now().Unix()))
			s.waiting = true
			s.sendMessage(msg, "", t)
			var reply PacketEvent
			for s.waiting {
				reply = <-sendReplyChan
				if reply.ID == t {
					s.waiting = false
				}
			}
			srPayload, err := reply.Payload()
			if err != nil {
				s.logger.Fatalln(err)
			}

			// need send-reply for msgID to reply to
			data, ok := srPayload.(*SendReply)
			if !ok {
				s.logger.Fatalln("Could not assert *SendReply as such.")
			}
			s.commitParent[payload.HeadCommit.ID] = data.ID
		case gohook.PingEventType:
			//payload, ok := et.Event.(*gohook.PingEvent)
			//if !ok {
			//	panic("Malformed *PingEvent")
			//}
			s.sendMessage("[ Github Webhook | ping ]", "", strconv.Itoa(s.msgID))
			s.msgID++
		}

	}
}