From 7b98d932b2bdbfa76ecdb07e17b95c1250066ee6 Mon Sep 17 00:00:00 2001
From: Evgeny <zobnin@gmail.com>
Date: Wed, 10 Apr 2024 20:01:46 +0400
Subject: [PATCH] [calendar-menu] use calendar colors instead of event colors

---
 main/calendar-menu.lua | 30 ++++++++++++++++++------------
 1 file changed, 18 insertions(+), 12 deletions(-)

diff --git a/main/calendar-menu.lua b/main/calendar-menu.lua
index 5af4289..7cbe4ae 100644
--- a/main/calendar-menu.lua
+++ b/main/calendar-menu.lua
@@ -4,15 +4,18 @@
 -- type = "drawer"
 -- aio_version = "4.7.99"
 -- author = "Evgeny Zobnin"
--- version = "1.0"
+-- version = "1.1"
 
 local fmt = require "fmt"
 
 local have_permission = false
 local events = {}
+local calendars = {}
 
 function on_drawer_open()
     events = calendar:events()
+    calendars = calendar:calendars()
+    add_cal_colors(events, calendars)
 
     if events == "permission_error" then
         calendar:request_permission()
@@ -25,14 +28,25 @@ function on_drawer_open()
         return
     end
 
-    lines = map(events, function(it)
-        local date = fmt.colored(format_date(it.begin), it.color)
+    lines = map(function(it)
+        local date = fmt.colored(format_date(it.begin), it.calendar_color)
         return date..fmt.space(4)..it.title
-    end)
+    end, events)
 
     drawer:show_ext_list(lines)
 end
 
+function add_cal_colors(events, cals)
+    for i, event in ipairs(events) do
+        for _, cal in ipairs(cals) do
+            if event.calendar_id == cal.id then
+                event.calendar_color = cal.color
+                break
+            end
+        end
+    end
+end
+
 function format_date(date)
     if system.format_date_localized then
         return system:format_date_localized("dd.MM", date)
@@ -53,11 +67,3 @@ function on_long_click(idx)
     calendar:open_event(events[idx].id)
 end
 
-function map(tbl, f)
-    local ret = {}
-    for k,v in pairs(tbl) do
-        ret[k] = f(v)
-    end
-    return ret
-end
-