Philipp Häfelfinger
5a4c133f27
moved piwigo login from app.go to piwigoContext.go Updated visibility of some types Added first elements of the local data store including configuration on startup and flag yea yea... I know, the commit contains to many changes ;-)
101 lines
3.1 KiB
Go
101 lines
3.1 KiB
Go
package app
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"git.haefelfinger.net/piwigo/PiwigoDirectoryUploader/internal/pkg/localFileStructure"
|
|
"git.haefelfinger.net/piwigo/PiwigoDirectoryUploader/internal/pkg/piwigo"
|
|
"github.com/sirupsen/logrus"
|
|
"path/filepath"
|
|
"sort"
|
|
)
|
|
|
|
func getAllCategoriesFromServer(context *appContext) (map[string]*piwigo.PiwigoCategory, error) {
|
|
logrus.Debugln("Starting GetAllCategories")
|
|
categories, err := piwigo.GetAllCategories(context.piwigo)
|
|
return categories, err
|
|
}
|
|
|
|
func synchronizeCategories(context *appContext, filesystemNodes map[string]*localFileStructure.FilesystemNode, existingCategories map[string]*piwigo.PiwigoCategory) error {
|
|
logrus.Infoln("Synchronizing categories...")
|
|
|
|
missingCategories := findMissingCategories(filesystemNodes, existingCategories)
|
|
|
|
if len(missingCategories) == 0 {
|
|
logrus.Infof("No categories missing!")
|
|
return nil
|
|
}
|
|
|
|
return createMissingCategories(context, missingCategories, existingCategories)
|
|
}
|
|
|
|
func findMissingCategories(fileSystem map[string]*localFileStructure.FilesystemNode, existingCategories map[string]*piwigo.PiwigoCategory) []string {
|
|
missingCategories := make([]string, 0, len(fileSystem))
|
|
|
|
for _, file := range fileSystem {
|
|
if !file.IsDir {
|
|
continue
|
|
}
|
|
|
|
_, exists := existingCategories[file.Key]
|
|
|
|
if exists {
|
|
logrus.Debugf("Found existing category %s", file.Key)
|
|
} else {
|
|
logrus.Infof("Missing category detected %s", file.Key)
|
|
missingCategories = append(missingCategories, file.Key)
|
|
}
|
|
}
|
|
|
|
return missingCategories
|
|
}
|
|
|
|
func createMissingCategories(context *appContext, missingCategories []string, existingCategories map[string]*piwigo.PiwigoCategory) error {
|
|
// we sort them to make sure the categories gets created
|
|
// in the right order and we have the parent available while creating the children
|
|
sort.Strings(missingCategories)
|
|
|
|
logrus.Infof("Creating %d categories", len(missingCategories))
|
|
|
|
for _, categoryKey := range missingCategories {
|
|
logrus.Infof("Creating category %s", categoryKey)
|
|
|
|
name, parentId, err := getNameAndParentId(categoryKey, existingCategories)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// create category on piwigo
|
|
id, err := piwigo.CreateCategory(context.piwigo, parentId, name)
|
|
if err != nil {
|
|
return errors.New(fmt.Sprintf("Could not create category on piwigo: %s", err))
|
|
}
|
|
|
|
newCategory := piwigo.PiwigoCategory{Id: id, Name: name, ParentId: parentId, Key: categoryKey}
|
|
logrus.Println(newCategory)
|
|
existingCategories[newCategory.Key] = &newCategory
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func getNameAndParentId(category string, categories map[string]*piwigo.PiwigoCategory) (string, int, error) {
|
|
parentKey := filepath.Dir(category)
|
|
_, name := filepath.Split(category)
|
|
if name == category {
|
|
logrus.Debugf("The category %s is a root category, there is no parent", name)
|
|
return name, 0, nil
|
|
}
|
|
|
|
logrus.Debugf("Looking up parent with key %s", parentKey)
|
|
parent, exists := categories[parentKey]
|
|
if !exists {
|
|
return "", 0, errors.New(fmt.Sprintf("could not find parent with key %s", parentKey))
|
|
}
|
|
|
|
parentId := parent.Id
|
|
logrus.Debugf("Found parent %s with id %d", parentKey, parentId)
|
|
|
|
return name, parentId, nil
|
|
}
|