96 lines
2.7 KiB
Go
96 lines
2.7 KiB
Go
package playstation4
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"github.com/rwcarlsen/goexif/exif"
|
|
|
|
"github.com/fmartingr/games-screenshot-mananger/pkg/games"
|
|
)
|
|
|
|
const providerName = "playstation-4"
|
|
const platformName = "PlayStation 4"
|
|
|
|
func addScreenshotToGame(userGames []games.Game, gameName string, screenshot games.Screenshot) []games.Game {
|
|
var foundGame games.Game
|
|
for gameIndex, game := range userGames {
|
|
if game.Name == gameName {
|
|
foundGame = game
|
|
userGames[gameIndex].Screenshots = append(userGames[gameIndex].Screenshots, screenshot)
|
|
}
|
|
}
|
|
|
|
// Game not found
|
|
if foundGame.Name == "" {
|
|
foundGame := games.Game{Name: gameName, ID: gameName, Platform: platformName, Provider: providerName}
|
|
foundGame.Screenshots = append(foundGame.Screenshots, screenshot)
|
|
userGames = append(userGames, foundGame)
|
|
}
|
|
|
|
return userGames
|
|
}
|
|
|
|
func GetGames(cliOptions games.CLIOptions) []games.Game {
|
|
var userGames []games.Game
|
|
|
|
err := filepath.Walk(*cliOptions.InputPath,
|
|
func(filePath string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !info.IsDir() {
|
|
var destinationName string
|
|
gameName := filepath.Base(filepath.Dir(filePath))
|
|
fileName := filepath.Base(filePath)
|
|
extension := filepath.Ext(filepath.Base(filePath))
|
|
layout := "20060102150405"
|
|
|
|
if extension == ".jpg" {
|
|
fileDescriptor, errFileDescriptor := os.Open(filePath)
|
|
if errFileDescriptor != nil {
|
|
log.Printf("[warning] Couldn't open file %s: %s", fileName, errFileDescriptor)
|
|
return nil
|
|
}
|
|
exifData, errExifData := exif.Decode(fileDescriptor)
|
|
if errExifData != nil {
|
|
log.Printf("[Error] Decoding EXIF data from %s: %s", filePath, errExifData)
|
|
return nil
|
|
}
|
|
defer fileDescriptor.Close()
|
|
|
|
exifDateTime, _ := exifData.DateTime()
|
|
destinationName = exifDateTime.Format(games.DatetimeFormat)
|
|
|
|
} else if extension == ".mp4" {
|
|
if len(fileName) >= len(layout)+len(extension) {
|
|
videoDatetime, err := time.Parse(layout, fileName[len(fileName)-len(extension)-len(layout):len(fileName)-len(extension)])
|
|
|
|
if err == nil {
|
|
destinationName = videoDatetime.Format(games.DatetimeFormat)
|
|
} else {
|
|
log.Printf("[Warning] File does not follow datetime convention: %s. (%s) skipping...", fileName, err)
|
|
return nil
|
|
}
|
|
} else {
|
|
log.Printf("[Warning] File does not follow datetime convention: %s, skipping...", fileName)
|
|
return nil
|
|
}
|
|
}
|
|
|
|
screenshot := games.Screenshot{Path: filePath, DestinationName: destinationName + extension}
|
|
userGames = addScreenshotToGame(userGames, gameName, screenshot)
|
|
|
|
}
|
|
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
log.Panic(err)
|
|
}
|
|
return userGames
|
|
}
|