From ba30e5c2d86ac0df13cf92322d47f40b263f5b04 Mon Sep 17 00:00:00 2001
From: Felipe Martin <me@fmartingr.com>
Date: Sun, 19 Feb 2023 22:35:36 +0100
Subject: [PATCH] wip

---
 internal/cli/cli.go                       |  2 +
 pkg/providers/worldofwarcraft/provider.go | 73 +++++++++++++++++++++++
 2 files changed, 75 insertions(+)
 create mode 100644 pkg/providers/worldofwarcraft/provider.go

diff --git a/internal/cli/cli.go b/internal/cli/cli.go
index 80853b1..7039d72 100644
--- a/internal/cli/cli.go
+++ b/internal/cli/cli.go
@@ -12,6 +12,7 @@ import (
 	"github.com/fmartingr/games-screenshot-manager/pkg/providers/playstation4"
 	"github.com/fmartingr/games-screenshot-manager/pkg/providers/retroarch"
 	"github.com/fmartingr/games-screenshot-manager/pkg/providers/steam"
+	"github.com/fmartingr/games-screenshot-manager/pkg/providers/worldofwarcraft"
 	"github.com/fmartingr/games-screenshot-manager/pkg/providers/xbox_game_bar"
 	"github.com/fmartingr/games-screenshot-manager/pkg/registry"
 	"github.com/sirupsen/logrus"
@@ -36,6 +37,7 @@ func Start() {
 	registry.Register(xbox_game_bar.Name, xbox_game_bar.NewXboxGameGarProvider)
 	registry.Register(steam.Name, steam.NewSteamProvider)
 	registry.Register(retroarch.Name, retroarch.NewRetroArchProvider)
+	registry.Register(worldofwarcraft.Name, worldofwarcraft.NewWorldOfWarcraftProvider)
 
 	options := models.Options{
 		ProcessBufferSize: 32,
diff --git a/pkg/providers/worldofwarcraft/provider.go b/pkg/providers/worldofwarcraft/provider.go
new file mode 100644
index 0000000..c7d8fc4
--- /dev/null
+++ b/pkg/providers/worldofwarcraft/provider.go
@@ -0,0 +1,73 @@
+package worldofwarcraft
+
+import (
+	"fmt"
+	"io/ioutil"
+	"path/filepath"
+	"runtime"
+	"strings"
+	"time"
+
+	"github.com/fmartingr/games-screenshot-manager/internal/models"
+	"github.com/fmartingr/games-screenshot-manager/pkg/helpers"
+	"github.com/sirupsen/logrus"
+)
+
+const (
+	Name           = "worldofwarcraft"
+	platformName   = "PC"
+	dateTimeLayout = "010206_150405"
+	filenamePrefix = "WoWScrnShot_"
+	extension      = ".jpg"
+)
+
+type WorldOfWarcraftProvider struct {
+	logger *logrus.Entry
+}
+
+func (p *WorldOfWarcraftProvider) FindGames(options models.ProviderOptions) ([]*models.Game, error) {
+	game := models.NewGame("wow", "World of Warcraft", platformName, Name)
+	var path string
+
+	if options.InputPath != "" {
+		path = helpers.ExpandUser(options.InputPath)
+	} else if runtime.GOOS == "windows" {
+		path = "C:/Program Files (x86)/World of Warcraft/_retail_/Screenshots"
+	} else if runtime.GOOS == "darwin" {
+		path = "/Applications/World of Warcraft/_retail_/Screenshots"
+	}
+
+	files, err := ioutil.ReadDir(path)
+	if err != nil {
+		return nil, fmt.Errorf("error reading from path %s: %s", options.InputPath, err)
+	}
+
+	for _, file := range files {
+		fullPath := filepath.Join(path, file.Name())
+
+		if strings.Contains(file.Name(), extension) {
+			var destinationName string
+
+			parts := strings.SplitN(file.Name(), "_", 2)
+			dateStr := strings.Replace(parts[1], extension, "", 1)
+
+			filenameDate, err := time.Parse(dateTimeLayout, dateStr)
+			if err != nil {
+				p.logger.Warnf("error parsing media creation time for %s: %s", file.Name(), err)
+				continue
+			}
+
+			destinationName = filenameDate.Format(models.DatetimeFormat) + extension
+
+			game.Screenshots = append(game.Screenshots, models.NewScreenshot(fullPath, destinationName))
+		}
+	}
+
+	return []*models.Game{&game}, nil
+}
+
+func NewWorldOfWarcraftProvider(logger *logrus.Logger, cache models.Cache) models.Provider {
+	return &WorldOfWarcraftProvider{
+		logger: logger.WithField("from", "provider."+Name),
+	}
+}