fix chain suffix

pkg/snapshot/header.go

@@ -2,6 +2,7 @@ package snapshot
 
 import (
 	"encoding/json"
+	"strings"
 )
 
 type SnapshotHeader struct {
@@ -26,3 +27,16 @@ func SnapshotHeaderFromJson(snapshotHeaderOutput string) (*SnapshotHeader, error
 
 	return &snapshotHeader.SnapshotHeader, nil
 }
+
+func (s *SnapshotHeader) SanitizeChainName() string {
+	result := s.ChaiName
+	if strings.Contains(result, "_") {
+		result = strings.Split(result, "_")[1]
+	}
+
+	if strings.Contains(result, "-") {
+		result = strings.Split(result, "-")[0]
+	}
+
+	return strings.ToLower(result)
+}

pkg/store/storage.go

@@ -140,7 +140,7 @@ func (s *SnapshotStorage) GetSnapshotItems(ctx context.Context) []snapshot.Snaps
 			Filename:        filename,
 			Filesize:        util.FileSize(size),
 			FilesizeBytes:   size,
-			ChainName:       RemoveTezosPrefix(snapshotHeader.ChaiName),
+			ChainName:       snapshotHeader.SanitizeChainName(),
 			Date:            date,
 			BlockTimestamp:  snapshotHeader.Timestamp,
 			URL:             obj.MediaLink,
@@ -169,13 +169,6 @@ func (s *SnapshotStorage) GetSnapshotItems(ctx context.Context) []snapshot.Snaps
 	return items
 }
 
-func RemoveTezosPrefix(filename string) string {
-	if strings.HasPrefix(filename, "TEZOS_") {
-		return strings.ToLower(filename[6:])
-	}
-	return filename
-}
-
 func (s *SnapshotStorage) DeleteExpiredSnapshots(ctx context.Context, maxDays int, maxMonths int) {
 	log.Println("Deleting expired snapshots in the Google Cloud Storage.")