From 071cd29f4ddb6db3ac9b04dde8abdf41adc5cea3 Mon Sep 17 00:00:00 2001
From: Mike Lang <mikelang3000@gmail.com>
Date: Tue, 19 Sep 2023 00:51:59 +1000
Subject: [PATCH] sheetsync: Implement Streamlog middleware

---
 sheetsync/sheetsync/streamlog.py | 76 +++++++++++++++++++++++++++++++-
 1 file changed, 75 insertions(+), 1 deletion(-)

diff --git a/sheetsync/sheetsync/streamlog.py b/sheetsync/sheetsync/streamlog.py
index f31a473..a4717ca 100644
--- a/sheetsync/sheetsync/streamlog.py
+++ b/sheetsync/sheetsync/streamlog.py
@@ -3,7 +3,9 @@ import json
 
 import requests
 
-class StreamLog():
+from common.dateutil import parse_utc_only
+
+class StreamLogClient():
 	"""Client for Stream Log server"""
 
 	def __init__(self, url, event_name, auth_token):
@@ -33,3 +35,75 @@ class StreamLog():
 		if content:
 			return json.loads(content)
 		return None
+
+
+class StreamLogMiddleware:
+	def __init__(self, client, bustime_start):
+		self.client = client
+		self.bustime_start = bustime_start
+		# Maps DB column names to streamlog fields.
+		self.column_map = {
+			'event_start': 'start_time',
+			'event_end': 'end_time',
+			'category': 'entry_type',
+			'description': 'description',
+			'submitter_winner': 'submitter_or_winner',
+			'poster_moment': 'poster_moment',
+			'image_links': 'media_link',
+			'notes': 'notes_to_editor',
+			'tags': 'tags',
+			'video_link': 'video_link',
+			'state': 'video_state',
+			'edit_link': 'editor_link',
+			'error': 'video_errors',
+			'id': 'id',
+		}
+		# Maps DB column names to a decode function to convert from streamlog format to internal.
+		# Omitted columns act as the identity function.
+		self.column_decode = {
+			'event_start': parse_utc_only,
+			'event_end': parse_utc_only,
+			'category': lambda v: v["name"],
+			'image_links': lambda v: [link.strip() for link in v.split()] if v.strip() else [],
+			'state': lambda v: v.upper(),
+		}
+		# Maps DB column names to an encode function to convert from internal format to streamlog.
+		# Omitted columns act as the identity function.
+		self.column_encode = {
+			'state': lambda v: v[0].upper() + v[1:].lower(), # Titlecase
+		}
+
+	def pick_worksheets(self):
+		# We don't have a concept of seperate worksheets, so just use a generic name
+		return "streamlog"
+
+	def get_rows(self):
+		for row in self.client.get_rows():
+			yield self.parse_row(row)
+
+	def parse_row(self, row):
+		output = {}
+		for column, key in self.column_map.items():
+			value = row[key]
+			if column in self.column_decode:
+				value = self.column_decode[column](value)
+			output[column] = value
+
+		# Implicit tags
+		day = dt_to_bustime(self.bustime_start, output['event_start']) // 86400
+		output['tags'] += [
+			output['category'],
+			"Day %d" % (day + 1) if day >= 0 else "Tech Test & Preshow",
+		]
+		if output["poster_moment"]:
+			output['tags'] += 'Poster Moment'
+
+		return output
+
+	def write_value(self, worksheet, row, key, value):
+		if key in self.column_encode:
+			value = self.column_encode[key](value)
+		self.client.write_value(row["id"], key, value)
+
+	def mark_modified(self, worksheet):
+		pass # not a concept we have