From fc0e951491dd6bf4fb0ae6675811f77113b55b75 Mon Sep 17 00:00:00 2001
From: LoveEevee <LoveEevee@users.noreply.github.com>
Date: Thu, 11 Oct 2018 02:59:17 +0300
Subject: [PATCH] Add preview time for tja files

---
 app.py | 39 +++++++++++++++++++++++++++++++++------
 1 file changed, 33 insertions(+), 6 deletions(-)

diff --git a/app.py b/app.py
index 458a7c7..216b357 100644
--- a/app.py
+++ b/app.py
@@ -60,6 +60,26 @@ def get_osu_key(osu, section, key, default=None):
     return default
 
 
+def get_tja_preview(tja):
+    tja_lines = open(tja, 'r').read().replace('\x00', '').split('\n')
+    
+    for line in tja_lines:
+        line = line.strip()
+        if ':' in line:
+            name, value = line.split(':', 1)
+            if name.lower() == 'demostart':
+                value = value.strip()
+                try:
+                    value = float(value)
+                except ValueError:
+                    pass
+                else:
+                    return int(value * 1000)
+        elif line.lower() == '#start':
+            break
+    return 0
+
+
 @app.teardown_appcontext
 def close_connection(exception):
     db = getattr(g, '_database', None)
@@ -85,12 +105,19 @@ def route_api_songs():
         categories[cat[0]] = {'title': cat[1], 'title_en': cat[2]}
     songs_out = []
     for song in songs:
-        osus = [osu for osu in os.listdir('public/songs/%s' % song[0]) if osu in ['easy.osu', 'normal.osu', 'hard.osu', 'oni.osu']]
-        if osus:
-            osud = parse_osu('public/songs/%s/%s' % (song[0], osus[0]))
-            preview = int(get_osu_key(osud, 'General', 'PreviewTime', 0))
+        type = song[9]
+        if type == "tja":
+            if os.path.isfile('public/songs/%s/main.tja' % song[0]):
+                preview = get_tja_preview('public/songs/%s/main.tja' % song[0])
+            else:
+                preview = 0
         else:
-            preview = 0
+            osus = [osu for osu in os.listdir('public/songs/%s' % song[0]) if osu in ['easy.osu', 'normal.osu', 'hard.osu', 'oni.osu']]
+            if osus:
+                osud = parse_osu('public/songs/%s/%s' % (song[0], osus[0]))
+                preview = int(get_osu_key(osud, 'General', 'PreviewTime', 0))
+            else:
+                preview = 0
         category_out = categories[song[8]] if song[8] in categories else def_category
         
         songs_out.append({
@@ -103,7 +130,7 @@ def route_api_songs():
             'preview': preview,
             'category': category_out['title'],
             'category_en': category_out['title_en'],
-            'type': song[9],
+            'type': type,
             'offset': song[10]
         })