arozos/system.go

package main

import (
	"encoding/json"
	"io/ioutil"
	"log"
	"net/http"
	"path/filepath"
	"strings"

	uuid "github.com/satori/go.uuid"
	"imuslab.com/arozos/mod/common"
	fs "imuslab.com/arozos/mod/filesystem"
)

/*
	System Identification API

	This module handles cross cluster scanning, responses and more that related
	to functions that identifiy this as a ArOZ Online device
*/
func SystemIDInit() {
	//Initialize device UUID if not exists
	systemIdGenerateSystemUUID()

	//Register as a system setting
	registerSetting(settingModule{
		Name:     "ArozOS",
		Desc:     "System Information",
		IconPath: "SystemAO/info/img/small_icon.png",
		Group:    "About",
		StartDir: "SystemAO/info/about.html",
	})

	//Handle the about page
	http.HandleFunc("/system/id/requestInfo", systemIdHandleRequest)

	//Handle ArOZ Online Beta search methods
	if *enable_beta_scanning_support {
		http.HandleFunc("/AOB/hb.php", systemIdResponseBetaScan)
		http.HandleFunc("/AOB/", func(w http.ResponseWriter, r *http.Request) {
			http.Redirect(w, r, "../index.html", 307)
		})
		http.HandleFunc("/AOB/SystemAOB/functions/info/version.inf", systemIdServeVersonNumber)
		http.HandleFunc("/AOB/SystemAOB/functions/system_statistic/getDriveStat.php", systemIdGetDriveStates)
	}

	//Handle license info
	registerSetting(settingModule{
		Name:     "Open Source",
		Desc:     "License from the Open Source Community",
		IconPath: "SystemAO/info/img/small_icon.png",
		Group:    "About",
		StartDir: "SystemAO/info/license.html",
	})

	registerSetting(settingModule{
		Name:     "License",
		Desc:     "License of ArozOS",
		IconPath: "SystemAO/info/img/small_icon.png",
		Group:    "About",
		StartDir: "SystemAO/info/srcLicense.html",
	})

	//Register vendor information
	if fs.FileExists("web/SystemAO/vendor/index.html") {
		registerSetting(settingModule{
			Name:     "Vendor",
			Desc:     "Vendor Notes",
			IconPath: "SystemAO/info/img/small_icon.png",
			Group:    "About",
			StartDir: "SystemAO/vendor/index.html",
		})
	}

	http.HandleFunc("/system/info/license", systemHandleListLicense)

	//Handle health check ping
	http.HandleFunc("/system/id/ping", systemIdHandlePing)

}

/*
	Ping function. This function handles the request
*/
func systemIdHandlePing(w http.ResponseWriter, r *http.Request) {
	w.Header().Set("Access-Control-Allow-Origin", "*")
	w.Header().Set("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE")
	w.Header().Set("Access-Control-Allow-Headers", "Accept, Content-Type, Content-Length, Accept-Encoding, X-CSRF-Token, Authorization")
	js, _ := json.Marshal(struct {
		Status string
	}{
		"OK",
	})
	common.SendJSONResponse(w, string(js))
}

func systemIdGenerateSystemUUID() {
	if !fs.FileExists("./system/dev.uuid") {
		//UUID not exist. Create one
		thisuuid := uuid.NewV4().String()
		if *system_uuid != "" {
			//User has defined the uuid. Use user defined one instead.
			thisuuid = *system_uuid
		}
		err := ioutil.WriteFile("./system/dev.uuid", []byte(thisuuid), 0755)
		if err != nil {
			log.Fatal(err)
		}
		deviceUUID = thisuuid
	} else {
		thisuuid, err := ioutil.ReadFile("./system/dev.uuid")
		if err != nil {
			log.Fatal("Failed to read system uuid file (system/dev.uuid).")
		}
		deviceUUID = string(thisuuid)
	}
}

func systemIdGetSystemUUID() string {
	fileUUID, err := ioutil.ReadFile("./system/dev.uuid")
	if err != nil {
		systemWideLogger.PrintAndLog("Storage", "Unable to read system UUID from dev.uuid file", err)
		log.Fatal(err)
	}

	return string(fileUUID)
}

func systemHandleListLicense(w http.ResponseWriter, r *http.Request) {
	licenses, _ := filepath.Glob("./web/SystemAO/info/license/*.txt")
	results := [][]string{}
	for _, file := range licenses {
		fileName := filepath.Base(file)
		name := strings.TrimSuffix(fileName, filepath.Ext(fileName))
		content, _ := ioutil.ReadFile(file)
		results = append(results, []string{name, string(content)})
	}

	js, _ := json.Marshal(results)
	common.SendJSONResponse(w, string(js))
}

func systemIdHandleRequest(w http.ResponseWriter, r *http.Request) {
	//Check if user has logged in
	if authAgent.CheckAuth(r) == false {
		common.SendErrorResponse(w, "User not logged in")
		return
	}

	//Group everything required to show into one json string
	type returnStruct struct {
		SystemUUID string
		IpAddress  string
		Vendor     string
		Build      string
		Version    string
		Model      string
		VendorIcon string
	}

	//thisDevIP := network_info_GetOutboundIP().String()
	thisDevIP := ""

	jsonString, _ := json.Marshal(returnStruct{
		SystemUUID: systemIdGetSystemUUID(),
		IpAddress:  thisDevIP,
		Vendor:     deviceVendor,
		Build:      build_version,
		Version:    internal_version,
		Model:      deviceModel,
		VendorIcon: iconVendor,
	})

	common.SendJSONResponse(w, string(jsonString))
}

func systemIdResponseBetaScan(w http.ResponseWriter, r *http.Request) {
	//Handle beta scanning method
	uuid := systemIdGetSystemUUID()
	IPAddress := r.Header.Get("X-Real-Ip")
	if IPAddress == "" {
		IPAddress = r.Header.Get("X-Forwarded-For")
	}
	if IPAddress == "" {
		IPAddress = r.RemoteAddr
	}
	IPAddress = IPAddress[:strings.LastIndex(IPAddress, ":")]
	resp := *host_name + ",Alive," + uuid + "," + IPAddress
	w.Header().Set("Access-Control-Allow-Origin", "*")
	w.Header().Set("Access-Control-Request-Headers", "*")
	w.Header().Set("Access-Control-Allow-Headers", "Content-Type")
	w.Write([]byte(resp))
}

func systemIdServeVersonNumber(w http.ResponseWriter, r *http.Request) {
	if build_version == "development" {
		w.Write([]byte("AO-DEV_v" + internal_version))
	} else {
		w.Write([]byte("AO-REL_v" + internal_version))
	}
}

func systemIdGetDriveStates(w http.ResponseWriter, r *http.Request) {
	results := [][]string{}
	results = append(results, []string{
		"user",
		"User",
		"-1B/-1B",
	})
	jsonString, _ := json.Marshal(results)
	common.SendJSONResponse(w, string(jsonString))
}