From 4dadc14d3bbb95bb8bb0a0ed5e086e37b30c8cbc Mon Sep 17 00:00:00 2001
From: xprism1 <xprism01@gmail.com>
Date: Sat, 24 Jun 2023 23:56:56 +0800
Subject: [PATCH] utils: cdn2cia: fix reading list of tmds

---
 utils.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/utils.py b/utils.py
index 6a57a29..623128b 100644
--- a/utils.py
+++ b/utils.py
@@ -523,7 +523,7 @@ def cdn2cia(path, out='', title_ver='', cdn_dev=0, cia_dev=0):
     tmd = ''
     tik = ''
     for i in os.listdir('.'):
-        if i.startswith('tmd'):
+        if i.startswith('tmd.'):
             tmds.append(i)
         elif i == 'cetk':
             tik = i
@@ -545,7 +545,8 @@ def cdn2cia(path, out='', title_ver='', cdn_dev=0, cia_dev=0):
         regen_sig = 'dev'
 
     t = TMDReader(tmd)
-    out = f'{name}.{t.hdr.title_ver}.cia'
+    if out == '':
+        out = f'{name}.{t.hdr.title_ver}.cia'
     if tik == '':
         tikBuilder(titleID=t.titleID, title_ver=t.hdr.title_ver, regen_sig=regen_sig, out='tik')
         tik = 'tik'