✨ Iconize

Author: oschrenk

Sources/Cli/Main.swift

@@ -6,7 +6,8 @@ enum Main {
     start: Date,
     end: Date,
     opts: SharedOptions,
-    eventSelector: EventSelectorI
+    selector: EventSelectorI,
+    transformer: EventTransformer
   ) {
     Log.setDebug(opts.debug)
 
@@ -35,7 +36,9 @@ enum Main {
       eventFilter: eventFilter
     )
 
-    let events = eventSelector.select(events: unsortedEvents)
+    let events = selector
+      .select(events: unsortedEvents)
+      .map { transformer.transform(event: $0) }
     if opts.templatePath.isEmpty {
       events.printAsJson()
     } else {

Sources/Cli/Next.swift

@@ -34,6 +34,10 @@ struct Next: ParsableCommand {
     ]
     )
 
-    Main.run(start: start, end: end, opts: opts, eventSelector: eventSelector)
+    Main.run(
+      start: start, end: end, opts: opts,
+      selector: eventSelector,
+      transformer: EventTransformer(rules: Loader.readConfig()?.iconize ?? [])
+    )
   }
 }

Sources/Cli/Today.swift

@@ -27,6 +27,10 @@ struct Today: ParsableCommand {
     ]
     )
 
-    Main.run(start: start, end: end, opts: opts, eventSelector: eventSelector)
+    Main.run(
+      start: start, end: end, opts: opts,
+      selector: eventSelector,
+      transformer: EventTransformer(rules: Loader.readConfig()?.iconize ?? [])
+    )
   }
 }

Sources/Model/EventTransformer.swift

@@ -2,14 +2,18 @@ import Foundation
 import swift_lens
 
 class EventTransformer {
-  let rule: Rule
+  let rules: [Rule]
 
   init(rule: Rule) {
-    self.rule = rule
+    rules = [rule]
+  }
+
+  init(rules: [Rule]) {
+    self.rules = rules
   }
 
   func transform(event: Event) -> Event {
-    if event.title.label.range(
+    for rule in rules where event.title.label.range(
       of: rule.regex, options: .regularExpression
     ) != nil {
       return event |> (Event.titleLens * Title.iconLens) *~ rule.icon