photoprism/importer.go

package photoprism

import (
	"path/filepath"
	"os"
	"log"
	"fmt"
	"github.com/pkg/errors"
	"path"
	"sort"
	"strings"
)

type Importer struct {
	originalsPath string
	removeDotFiles bool
	removeExistingFiles bool
	removeEmptyDirectories bool
}

func NewImporter(originalsPath string) *Importer {
	instance := &Importer{
		originalsPath: originalsPath,
		removeDotFiles: true,
		removeExistingFiles: true,
		removeEmptyDirectories: true,
	}

	return instance
}

func (i *Importer) ImportPhotosFromDirectory(importPath string) {
	var directories []string

	err := filepath.Walk(importPath, func(filename string, fileInfo os.FileInfo, err error) error {
		if err != nil {
			// log.Print(err.Error())
			return nil
		}

		if fileInfo.IsDir() {
			if filename != importPath {
				directories = append(directories, filename)
			}
			return nil
		}

		if i.removeDotFiles && strings.HasPrefix(filepath.Base(filename), ".") {
			os.Remove(filename)
			return nil
		}

		mediaFile := NewMediaFile(filename)

		if !mediaFile.Exists() || !mediaFile.IsPhoto() {
			return nil
		}

		relatedFiles, masterFile, _ := mediaFile.GetRelatedFiles()

		for _, relatedMediaFile := range relatedFiles {
			if destinationFilename, err := i.GetDestinationFilename(masterFile, relatedMediaFile); err == nil {
				os.MkdirAll(path.Dir(destinationFilename), os.ModePerm)
				log.Printf("Moving file %s to %s", relatedMediaFile.GetFilename(), destinationFilename)
				relatedMediaFile.Move(destinationFilename)
			} else if i.removeExistingFiles {
				relatedMediaFile.Remove()
				log.Printf("Deleted %s (already exists)", relatedMediaFile.GetFilename())
			}
		}

		// mediaFile.Move(i.originalsPath)

		return nil
	})

	sort.Slice(directories, func(i, j int) bool {
		return len(directories[i]) > len(directories[j])
	})

	if i.removeEmptyDirectories {
		// Remove empty directories from import path
		for _, directory := range directories {
			if directoryIsEmpty(directory) {
				os.Remove(directory)
				log.Printf("Deleted empty directory %s", directory)
			}
		}
	}

	if err != nil {
		log.Print(err.Error())
	}
}

func (i *Importer) GetDestinationFilename(masterFile *MediaFile, mediaFile *MediaFile) (string, error) {
	canonicalName := masterFile.GetCanonicalName()
	fileExtension := mediaFile.GetExtension()
	dateCreated := masterFile.GetDateCreated()

	//	Mon Jan 2 15:04:05 -0700 MST 2006
	pathName := i.originalsPath + "/" + dateCreated.UTC().Format("2006/01")

	iteration := 1

	result := pathName + "/" + canonicalName + fileExtension

	for fileExists(result) {
		if mediaFile.GetHash() == fileHash(result) {
			return result, errors.New("File already exists")
		}

		iteration++

		result = pathName + "/" + canonicalName + "_" + fmt.Sprintf("V%d", iteration) + fileExtension
	}

	return result, nil
}