fix chain suffix

marigold-dev · Nov 21, 2023 · 35837e0 · 35837e0
1 parent 3e91456
commit 35837e0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/pkg/snapshot/header.go b/pkg/snapshot/header.go
@@ -2,6 +2,7 @@ package snapshot
 
 import (
 	"encoding/json"
+	"strings"
 )
 
 type SnapshotHeader struct {
@@ -26,3 +27,7 @@ func SnapshotHeaderFromJson(snapshotHeaderOutput string) (*SnapshotHeader, error
 
 	return &snapshotHeader.SnapshotHeader, nil
 }
+
+func (s *SnapshotHeader) SanitizeChainName() string {
+	return strings.ToLower(strings.Split(strings.Split(s.ChaiName, "-")[0], "_")[1])
+}
diff --git a/pkg/store/storage.go b/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.")