diff --git a/cmd/gh-not/main.go b/cmd/gh-not/main.go index db3d8da..6b7f8cd 100644 --- a/cmd/gh-not/main.go +++ b/cmd/gh-not/main.go @@ -27,18 +27,12 @@ func main() { fmt.Println(allNotifications.ToString()) - actorsMap := map[string]actors.Actor{ - "debug": &actors.DebugActor{}, - "print": &actors.PrintActor{}, - "hide": &actors.HideActor{}, - } - config, err := config.New("config.json") if err != nil { panic(err) } - allNotifications, err = config.Apply(allNotifications, actorsMap) + allNotifications, err = config.Apply(allNotifications, actors.Map()) if err != nil { panic(err) } diff --git a/internal/actors/actors.go b/internal/actors/actors.go index b96ea8c..4226e93 100644 --- a/internal/actors/actors.go +++ b/internal/actors/actors.go @@ -6,6 +6,16 @@ import ( "github.com/nobe4/gh-not/internal/notifications" ) +type ActorsMap map[string]Actor + +func Map() ActorsMap { + return map[string]Actor{ + "debug": &DebugActor{}, + "print": &PrintActor{}, + "hide": &HideActor{}, + } +} + type Actor interface { Run(notifications.Notification) (notifications.Notification, error) }