mirror of
https://github.com/strongdm/comply
synced 2024-08-26 09:16:47 +00:00
90 lines
2.0 KiB
Go
90 lines
2.0 KiB
Go
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")
|
|
)
|
|
|
|
// 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
|
|
}
|
|
|
|
// 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
|
|
|
|
cfg, ok := ticketsMap[string(ts)]
|
|
if !ok {
|
|
spew.Dump(cfg)
|
|
panic(fmt.Sprintf("Missing configuration for plugin system; add `%s` block to project YAML", string(ts)))
|
|
}
|
|
|
|
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
|
|
}
|