main.go

package main

import (
	"fmt"
	"log"
	"os"

	"github.com/jmartin127/linear-autolabeler/linear"
	"github.com/jmartin127/linear-autolabeler/sla"
)

// TODO: Put all of the constants into a config object
var issueStatesToIgnore = [...]string{"Done", "Canceled"}

const (
	pageSize = 50
	teamID   = "99dea3d2-59ff-4273-b8a1-379d36bb1678" // TODO load the team ID from the team name
	timeZone = "America/Denver"
)

func main() {
	fmt.Println("Starting...")

	// initialize the linear client
	if len(os.Args) < 2 {
		log.Fatal("No auth token was provided.\nUsage: go run main.go <auth-token>")
	}
	authToken := os.Args[1]
	lc := &linear.LinearClient{
		Token: authToken,
	}

	// initialize the SLA client
	slaClient, err := sla.NewSLA(lc, timeZone)
	if err != nil {
		log.Fatal(err)
	}

	// find the "ExceedsSLA" label
	fmt.Println("Finding Exceeds SLA label...")
	exceedsSLALabelID, err := lc.FindLabelIDWithName(teamID, "ExceedsSLA")
	if err != nil {
		log.Fatal(err)
	}

	var totalIssues int
	pagination := fmt.Sprintf("first:%d", pageSize)
	for true {
		fmt.Printf("Loading issues for team %s and page %s\n", teamID, pagination)
		response, err := lc.GetIssuesForTeam(teamID, pagination)
		if err != nil {
			log.Fatal(err)
		}

		for _, v := range response.Team.Issues.Edges {
			if shouldIgnoreState(v.IssueNode.State.Name) {
				continue
			}

			exceeds, durationExceeding, sla := slaClient.ExceedsSLA(&v.IssueNode)
			ticketNumber := linear.TicketNumber(&v.IssueNode)
			if exceeds {
				addedLabel, err := lc.AddLabelToTicket(ticketNumber, exceedsSLALabelID)
				if err != nil {
					log.Fatal(err)
				}
				if addedLabel {
					comment := fmt.Sprintf("Uh oh!  This ticket is in the %s state, and exceeds the SLA by %s!  FYI, the SLA is %+v (in business hours).", v.IssueNode.State.Name, durationExceeding, sla)
					log.Printf("Ticket: %s, Adding Comment: %s\n", ticketNumber, comment)
					if err := lc.AddCommentToTicket(v.IssueNode.ID, comment); err != nil {
						log.Fatal(err)
					}
				}
			} else {
				if err := lc.RemoveLabelFromTicket(ticketNumber, exceedsSLALabelID); err != nil {
					log.Fatal(err)
				}
			}
			totalIssues++
		}

		// pagination
		pagination = fmt.Sprintf(`first:%d after:"%s"`, pageSize, response.Team.Issues.PageInfo.EndCursor)
		if response.Team.Issues.PageInfo.HasNextPage == false {
			break
		}
	}

	fmt.Printf("Total issues: %d\n", totalIssues)
}

func shouldIgnoreState(state string) bool {
	for _, ignoredState := range issueStatesToIgnore {
		if state == ignoredState {
			return true
		}
	}

	return false
}