read-write-utility.go

package main

import (
	"encoding/json"
	"fmt"
	"github.com/bradfitz/latlong"
	"io/ioutil"
	"os"
	"time"
	//"log"
)

func createImageDirectories(directories ...string) (err error) {
	for _, directory := range directories {
		var fileExist bool
		if fileExist, err = exists(directory); err != nil {
			return
		} else if fileExist {
			continue
		}

		if err = os.Mkdir(directory, os.ModePerm); err != nil {
			return
		}
	}
	return
}

func (t *Terminal) getTZ() (err error) {
	tz := latlong.LookupZoneName(
		(*t).Location.Latitude,
		(*t).Location.Longitude)

	if len(tz) == 0 {
		err = fmt.Errorf("latlong.LookupZoneName returned empty string for Terminal %v\n", (*t))
	}

	(*t).Timezone, err = time.LoadLocation(
		tz)

	//Debug. Used to TZ export
	(*t).TimezoneTitle = tz
	return
}

func readTerminalArrayToMap(terminalArray []Terminal) (terminalMap map[string]Terminal) {
	//set key to v.Title and set v.Index
	terminalMap = make(map[string]Terminal)
	for _, v := range terminalArray {
		terminalMap[v.Title] = v
	}

	return
}

func readTerminalArrayFromFiles(filenames ...string) (keywordsArray []Terminal, err error) {
	for _, filename := range filenames {
		var locationsRaw []byte
		locationsRaw, err = ioutil.ReadFile(filename)
		if err != nil {
			return
		}

		var tmp []Terminal
		if err = json.Unmarshal(locationsRaw, &tmp); err != nil {
			return
		}

		//Get Timezone for every terminal
		//log.Println("Looking up timezone for terminals.")
		for i, _ := range tmp {
			if err = tmp[i].getTZ(); err != nil {
				return
			}
			//fmt.Println(tmp[i].Timezone, tmp[i].Title, tmp[i].Location)
		}

		//log.Printf("%v locations in file %v", len(tmp), filenames)

		keywordsArray = append(keywordsArray, tmp...)
	}
	/*
	 * Export TZ offset list to json file for preset terminals based on computed TZ
	 */
	/*
	type TerminalTZOffset struct {
		Title          string `json:"title"`
		TimezoneOffset int    `json:"tzOffset"`
	}
	 */

	var terminalTitleTZOffsetArray []Terminal
	for _, v := range keywordsArray {
		t := time.Now()
		t = t.In(v.Timezone)
		_, offset := t.Zone()
		v.TimezoneOffset = offset

		terminalTitleTZOffsetArray = append(terminalTitleTZOffsetArray, v)
	}

	if DEBUG_EXPORT_TERMINAL_TZ {
		output, _ := json.MarshalIndent(terminalTitleTZOffsetArray, "", "	")
		ioutil.WriteFile("tz_export.json", output, 0644)
		fmt.Println("TZ exported")
	}
	
	return
}