package model import ( "fmt" "sync" "github.com/davecgh/go-spew/spew" "github.com/strongdm/comply/internal/config" ) var tsPluginsMu sync.Mutex var tsPlugins = make(map[TicketSystem]TicketPlugin) var tsConfigureOnce sync.Once // TicketSystem is the type of ticket database. type TicketSystem string const ( // Jira from Atlassian. Jira = TicketSystem("jira") // GitHub from GitHub. GitHub = TicketSystem("github") // NoTickets indicates no ticketing system integration. NoTickets = TicketSystem("none") ) type TicketLinks struct { ProcedureOpen string ProcedureAll string AuditOpen string AuditAll string } // TicketPlugin models support for ticketing systems. type TicketPlugin interface { Get(ID string) (*Ticket, error) FindOpen() ([]*Ticket, error) FindByTag(name, value string) ([]*Ticket, error) FindByTagName(name string) ([]*Ticket, error) Create(ticket *Ticket, labels []string) error Configure(map[string]interface{}) error Prompts() map[string]string Links() TicketLinks LinkFor(ticket *Ticket) string Configured() bool } // GetPlugin loads the ticketing database. func GetPlugin(ts TicketSystem) TicketPlugin { tsPluginsMu.Lock() defer tsPluginsMu.Unlock() tp, ok := tsPlugins[ts] if !ok { panic("Unknown ticket system: " + ts) } if config.Exists() { tsConfigureOnce.Do(func() { ticketsMap := config.Config().Tickets hasTickets := true cfg, ok := ticketsMap[string(ts)] if !ok { hasTickets = false } if hasTickets { cfgTyped, ok := cfg.(map[interface{}]interface{}) if !ok { spew.Dump(cfg) panic(fmt.Sprintf("malformatted ticket configuration block `%s` in project YAML", string(ts))) } cfgStringed := make(map[string]interface{}) for k, v := range cfgTyped { kS, ok := k.(string) if !ok { spew.Dump(cfgStringed) panic(fmt.Sprintf("malformatted key in configuration block `%s` in project YAML", string(ts))) } cfgStringed[kS] = v } tp.Configure(cfgStringed) } }) } return tp } // Register ticketing system plugin. func Register(ts TicketSystem, plugin TicketPlugin) { tsPluginsMu.Lock() defer tsPluginsMu.Unlock() _, ok := tsPlugins[ts] if ok { panic("Duplicate ticketing system registration: " + ts) } tsPlugins[ts] = plugin }