media.go

package main

import (
	"encoding/json"
	"fmt"
	"io/ioutil"
	"os"
	"strings"
)

// getLocalMedia reads media from local directory
func getLocalMedia() (media []Media) {
	files, err := ioutil.ReadDir("./media")
	if err != nil {
		log.Info("Error reading directory", err)
	}
	for _, file := range files {
		if strings.Contains(file.Name(), ".jpg") {
			m := Media{}
			m.LocalFile = file.Name()
			media = append(media, m)
		}
	}
	return media
}

// getRemoteMedia reads media from json file
func getRemoteMedia() (media []Media) {
	// check if file exists, return empty
	// likely scenario would be first run
	if _, err := os.Stat("media.json"); os.IsNotExist(err) {
		if !setup { // dont alert about missing file when known init
			log.Debug("media.json does not exist, first run?")
		}
		return media
	}

	file, err := ioutil.ReadFile("media.json")
	if err != nil {
		log.Warn("Error reading media.json, permissions?", err)
	} else {
		if err := json.Unmarshal(file, &media); err != nil {
			log.Warn("Error parsing JSON from media.json", err)
		}
	}
	return media
}

func compareMedia(local, remote []Media) (media []Media) {
	for _, m := range local {
		exists := false
		for _, r := range remote {
			if m.LocalFile == r.LocalFile {
				exists = true
				log.Debug("Skipping ", m.LocalFile)
			}
		}
		if !exists {
			media = append(media, m)
		}
	}
	return media
}

func uploadMediaItems(media []Media) (uploadedMedia []Media) {
	for _, m := range media {
		if confirmPrompt(fmt.Sprintf("Upload %s, Continue (y/N)? ", m.LocalFile)) {
			upm, err := uploadMedia(m)
			if err == nil {
				upm.LocalFile = m.LocalFile
				log.Info(fmt.Sprintf("Uploaded: %s %s", m.LocalFile, upm.URL))
				uploadedMedia = append(uploadedMedia, upm)
			} else {
				log.Warn("Upload Error", err)
			}
		}
	}
	return uploadedMedia
}

// writeRemoteMedia
func writeRemoteMedia(media []Media) {
	if len(media) == 0 {
		log.Info("No new media to upload.")
		return
	}
	// append new post json
	existingMedia := getRemoteMedia()
	existingMedia = append(existingMedia, media...)

	// write file
	json, err := json.Marshal(existingMedia)
	if err != nil {
		log.Warn("Encoding Error", err)
	} else {
		err = ioutil.WriteFile("media.json", json, 0644)
		if err != nil {
			log.Warn("Error writing media.json", err)
		} else {
			log.Warn("media.json written")
		}
	}
}