wip

Author: joelhawksley

app/apis/api.rb

@@ -1,6 +1,10 @@
 class Api
+  def url
+    Timeframe::Application.config.local["#{self.class.name.underscore}_url"]
+  end
+
   def fetch
-    response = HTTParty.get(Timeframe::Application.config.local["#{self.class.name.underscore}_url"], headers: headers)
+    response = HTTParty.get(url, headers: headers)
 
     return if response.code != 200
 

app/apis/home_assistant_calendar_api.rb

@@ -0,0 +1,73 @@
+class HomeAssistantCalendarApi < Api
+  def initialize(config = Timeframe::Application.config.local)
+    @config = config
+  end
+
+  def prepare_response(response)
+    start_time = (Time.now - 1.day).utc.iso8601
+    end_time = (Time.now + 5.days).utc.iso8601
+
+    out = []
+
+    response.each do |calendar|
+      res = HTTParty.get("#{url}/#{calendar["entity_id"]}?start=#{start_time}&end=#{end_time}", headers: headers)
+      config = Timeframe::Application.config.local["calendars"].find { _1["entity_id"] == calendar["entity_id"] }
+
+      res.map! do |event|
+        event["calendar_entity_id"] = calendar["entity_id"]
+        event["calendar_name"] = calendar["name"]
+
+        event["starts_at"] =
+          if event["start"]["date"]
+            event["start"]["date"]
+          else
+            event["start"]["dateTime"]
+          end
+
+        event["ends_at"] =
+          if event["end"]["date"]
+            event["end"]["date"]
+          else
+            event["end"]["dateTime"]
+          end
+
+        if config
+          event["icon"] = config["icon"]
+          event["letter"] = config["letter"]
+        end
+
+        event["id"] = event["uid"]
+
+        event.delete("uid")
+        event.delete("start")
+        event.delete("end")
+        event.delete("recurrence_id")
+        event.delete("rrule")
+        event.delete("calendar_entity_id")
+        event.delete("calendar_name")
+
+        event
+      end
+
+      out.concat(res)
+    end
+
+    out
+  end
+
+  def headers
+    {
+      Authorization: "Bearer #{@config["home_assistant_token"]}",
+      "content-type": "application/json"
+    }
+  end
+
+  def data
+    @data ||=
+      if super.empty?
+        []
+      else
+        super.map { CalendarEvent.new(**_1.symbolize_keys!) }
+      end
+  end
+end

app/models/display_content.rb

@@ -5,6 +5,7 @@ def call(
     google_calendar_api: GoogleCalendarApi.new,
     calendar_feed: CalendarFeed.new,
     home_assistant_api: HomeAssistantApi.new,
+    home_assistant_calendar_api: HomeAssistantCalendarApi.new,
     birdnet_api: BirdnetApi.new,
     air_now_api: AirNowApi.new
   )
@@ -95,6 +96,8 @@ def call(
       out[:status_icons_with_labels] << ["triangle-exclamation", "Google Calendar"]
     end
 
+    raw_events << home_assistant_calendar_api.data
+
     # :nocov:
 
     out[:day_groups] =