From 3cdb8e7a81b0c89f30500ef8d841d2c69f511d48 Mon Sep 17 00:00:00 2001
From: Night Space <63776734+NightSpaceC@users.noreply.github.com>
Date: Sun, 2 Apr 2023 21:13:20 +0800
Subject: [PATCH] fix(trainbit): incorrect filename display (#4027)

---
 drivers/trainbit/util.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/drivers/trainbit/util.go b/drivers/trainbit/util.go
index 91f7fb8f..ab9c55cc 100644
--- a/drivers/trainbit/util.go
+++ b/drivers/trainbit/util.go
@@ -113,10 +113,10 @@ func parseRawFileObject(rawObject []any) ([]model.Obj, error) {
 		}
 		isFolder := int64(object["ty"].(float64)) == 1
 		var name string
-		if isFolder {
-			name = object["name"].(string)
+		if object["ext"].(string) != "" {
+			name = strings.Join([]string{object["name"].(string), object["ext"].(string)}, ".")
 		} else {
-			name = object["name"].(string) + object["ext"].(string)
+			name = object["name"].(string)
 		}
 		modified, err := time.Parse("2006/01/02 15:04:05", object["modified"].(string))
 		if err != nil {