refactor: move the actor map into a function

nobe4 · May 18, 2024 · b6ac240 · b6ac240
1 parent 0b7cb53
commit b6ac240
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git 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
@@ -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)
 }