ffrs-fw-dl/main.go

291 lines
6.4 KiB
Go
Raw Normal View History

2020-05-10 16:45:29 +00:00
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)
}