use of platform aliases

1 files changed, 15 insertions, 8 deletions

diff --git a/internal/config/args.go b/internal/config/args.go
index a0fa0ea..d1d8fe8 100644
--- a/internal/config/args.go
+++ b/internal/config/args.go
@@ -8,6 +8,7 @@ import (
 	"time"
 
 	"codeberg.org/snonux/gos/internal/colour"
+	"codeberg.org/snonux/gos/internal/platforms"
 )
 
 var validPlatforms = []string{"mastodon", "linkedin"}
@@ -16,7 +17,7 @@ type Args struct {
 	GosDir            string
 	CacheDir          string
 	DryRun            bool
-	Platforms         map[string]int // Platform name and post size limits
+	Platforms         map[platforms.Platform]int // Platform and post size limits
 	Target            int
 	MinQueued         int
 	MaxDaysQueued     int
@@ -27,20 +28,26 @@ type Args struct {
 	OAuth2Browser     string
 }
 
-func (a *Args) ParsePlatforms(platforms string) error {
-	for _, platform := range strings.Split(platforms, ",") {
+func (a *Args) ParsePlatforms(platformStrs string) error {
+	a.Platforms = make(map[platforms.Platform]int)
+
+	for _, platformStr := range strings.Split(platformStrs, ",") {
 		// E.g. Mastodon:500
-		parts := strings.Split(platform, ":")
-		var err error
+		parts := strings.Split(platformStr, ":")
+		platform, err := platforms.New(parts[0])
+		if err != nil {
+			return err
+		}
+
 		// E.g. args.Platform["mastodon"] = 500
 		if len(parts) > 1 {
-			a.Platforms[parts[0]], err = strconv.Atoi(parts[1])
+			a.Platforms[platform], err = strconv.Atoi(parts[1])
 			if err != nil {
 				return err
 			}
 		} else {
 			colour.Infoln("No message length specified for", platform, "so assuming 500")
-			a.Platforms[parts[0]] = 500
+			a.Platforms[platform] = 500
 		}
 	}
 	return nil
@@ -48,7 +55,7 @@ func (a *Args) ParsePlatforms(platforms string) error {
 
 func (a *Args) Validate() error {
 	for platform := range a.Platforms {
-		if !slices.Contains(validPlatforms, strings.ToLower(platform)) {
+		if !slices.Contains(validPlatforms, platform.String()) {
 			return fmt.Errorf("Platform %s not supported", platform)
 		}
 	}