ffrs-fw-dl/main.go

package main

import (
	"fmt"
	"io/ioutil"
	"log"
	"os"
	"path/filepath"
	"regexp"
	"strings"

	"git.nils.zone/nils/prettify"
	"github.com/sergi/go-diff/diffmatchpatch"
	"gopkg.in/yaml.v2"
)

// structs
type firmware struct {
	Hersteller          string
	Modell              string
	Version             string
	Erstinstallation    string
	Entwicklungsstadium string
}

type image struct {
	Name       string `json:"name"`
	Prettyname string `json:"prettyname"`
}

type city struct {
	Name   string  `json:"name"`
	Images []image `json:"images"`
}

type manufacturer struct {
	Name       string `json:"name"`
	Prettyname string `json:"prettyname"`
}

type assignments struct {
	Cities        []city         `json:"cities"`
	Manufacturers []manufacturer `json:"manufacturers"`
}

// global variables
var branches = []string{"stable", "beta", "experimental"}
var installMethod = []string{"sysupgrade", "factory"}
var community string
var domains []string
var prefix = ""
var assign assignments
var currentBranch string // holds info of the current branch
var currentMethod string // holds info of the current install method

func stringInSlice(a string, list []string) bool {
	for _, b := range list {
		if b == a {
			return true
		}
	}
	return false
}

func getFirmwareImages(folder string) {
	// get all files in folder
	files := getFilesOrFolders(folder)

	// check if files exist
	//fmt.Println(files)
	if files != nil {
		getImagePrefix(folder)

		// loop through all files
		for _, file := range files {
			// check if file really is a file
			if !file.IsDir() {
				// check if file starts with gluon
				if strings.HasPrefix(file.Name(), "gluon") {

					//fmt.Println(file.Name())
					/*
						path, err := filepath.Abs(file.Name())
						if err != nil {
							panic(err)
						}
					*/
					//getFirmwareData(path, prefix)
					getFirmwareData(file.Name(), prefix)
				}
			}
		}
	}
}

func getFirmwareData(image string, prefix string) {
	// create firmware variable
	var fw firmware
	//fmt.Println(prefix)
	//fmt.Println(image)

	// set the branch of firmware
	fw.Entwicklungsstadium = currentBranch

	// set install method of firmware
	fw.Erstinstallation = currentMethod

	// cut away the prefix of this folder
	fw.Modell = strings.Split(image, prefix)[1]

	// cut away the extension
	//fw.Modell = strings.TrimSuffix(fw.Modell, filepath.Ext(fw.Modell)) // this has issues with img.gz use below until we have an image with a "." in it
	fw.Modell = strings.Split(fw.Modell, ".")[0]

	for _, manufacturer := range assign.Manufacturers {
		// check if image is from manufacturer
		if strings.HasPrefix(fw.Modell, manufacturer.Name) {
			// image is from manufacturer
			fw.Hersteller = manufacturer.Name
			break
		}
	}

	// debug output message
	var out bool

	// check if manufacturer was not found
	if fw.Hersteller == "" {
		// guess the manufacturer (word(s) before first "-")
		fw.Hersteller = strings.Split(fw.Modell, "-")[0]
		log.Printf("I had to guess the manufacturer, sorry! I think it is >%v<\nIf you think this is wrong please change assignments.yml on our git repository\n\n", fw.Hersteller)

		// set debug output message to true
		out = true
	}

	// assemble new image name (cut away manufacturer)
	fw.Modell = strings.Split(fw.Modell, fw.Hersteller+"-")[1]
	//fmt.Println(fw.Modell)

	// compile regex for hw revisions
	r, err := regexp.Compile("(v[0-9]{1,2}$)|([-]*rev-\\w*$)|(-[a-zA-Z][0-9]{0,1}$)")
	if err != nil {
		panic(err)
	}

	// find hardware revision in image name
	fw.Version = r.FindString(fw.Modell)

	if fw.Version != "" {
		// cut hardware version from image name
		fw.Modell = strings.Split(fw.Modell, fw.Version)[0]
	} else {
		fw.Version = "Alle"
	}

	// check if hardware version starts with a dash
	if strings.HasPrefix(fw.Version, "-") {
		// cut away "-"
		fw.Version = fw.Version[1:]
	}

	//out, err := yaml.Marshal(fw)
	//if err != nil {
	//	panic(err)
	//}
	//fmt.Println(len(out))
	//fmt.Println(string(out))

	//fmt.Println(image)
	/*
		os.Exit(0)
		// compile regex for branches
		r, err := regexp.Compile("(stable\\b|beta\\b|experimental\\b)")
		if err != nil {
			panic(err)
		}
		branch := r.Find([]byte(image))
		fmt.Println(string(branch))

		os.Exit(0)
	*/
	// check if out was set and if so print out all info we have on the device
	if out {
		log.Println(image)
		prettify.Print(fw)
	}
}

func getSubdirectory(folder string) {
	subFolders := getFilesOrFolders(folder)

	// go through all subFolder
	for _, f := range subFolders {

		// check if entry is a directory
		if f.IsDir() {
			// check if we are at root
			if folder == "./images" {
				community = f.Name()
				//fmt.Println(community)
			}
			//fmt.Println(f.Name())
			// check if we are a branch
			if stringInSlice(f.Name(), branches) {
				//fmt.Println(f.Name())
				currentBranch = f.Name()
				domain := filepath.Base(folder)
				domains = append(domains, domain)
				//fmt.Println("\t" + f.Name())
			} else if stringInSlice(f.Name(), installMethod) {
				// check if we are stable/beta/experimental
				currentMethod = f.Name()
			}

			// recursive call
			getSubdirectory(folder + "/" + f.Name())
		}
	}

	// at the end of the tunnel
	if stringInSlice(filepath.Base(folder), installMethod) {
		//fmt.Println("\t\t" + filepath.Base(folder))
		firmwareFolder, err := filepath.Abs(folder)
		if err != nil {
			panic(err)
		}
		//fmt.Println(firmwareFolder)
		getFirmwareImages(firmwareFolder)
	}
}

func getFilesOrFolders(folder string) []os.FileInfo {
	// get all files in folder
	files, err := ioutil.ReadDir(folder)
	if err != nil {
		log.Fatal(err)
	}

	if len(files) == 0 {
		//fmt.Println(len(files))
		return nil
	}
	return files
}

func getImagePrefix(folder string) string {
	files := getFilesOrFolders(folder)
	if prefix == "" {
		prefix = files[0].Name()
	}
	for _, file := range files {
		prefix = diffStrings(file.Name(), prefix)
	}
	return prefix
}

func diffStrings(text1 string, text2 string) string {
	dmp := diffmatchpatch.New()
	length := dmp.DiffCommonPrefix(text1, text2)
	return text1[0:length]
}

func getAssignments() assignments {
	// declare returning variable
	var a assignments

	// load assignments from file
	file, err := ioutil.ReadFile("assignments.yml")
	if err != nil {
		panic(err)
	}

	// unmarshal file onto assignment
	err = yaml.Unmarshal(file, &a)
	if err != nil {
		panic(err)
	}

	return a
}

func main() {
	// get all assignments from file
	assign = getAssignments()

	// start script in images folder
	getSubdirectory("./images")
	fmt.Println(branches)
}