mirror of
https://github.com/yggdrasil-network/yggdrasil-go.git
synced 2025-04-29 14:45:07 +03:00
Migrated to REST backend RIVM-31
This commit is contained in:
parent
609ba1a3cd
commit
d68d3db702
9 changed files with 237 additions and 313 deletions
|
@ -2,25 +2,14 @@ package main
|
|||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"net"
|
||||
"net/url"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/RiV-chain/RiV-mesh/src/defaults"
|
||||
"github.com/hjson/hjson-go"
|
||||
"github.com/webview/webview"
|
||||
"golang.org/x/text/encoding/unicode"
|
||||
|
||||
"github.com/RiV-chain/RiV-mesh/src/admin"
|
||||
"github.com/RiV-chain/RiV-mesh/src/defaults"
|
||||
"github.com/docopt/docopt-go"
|
||||
)
|
||||
|
||||
|
@ -32,7 +21,7 @@ Usage:
|
|||
mesh-ui -v | --version
|
||||
|
||||
Options:
|
||||
<index> Index file name [default: index.html].
|
||||
<index> Index file name [default: http://localhost:19019].
|
||||
-c --console Show debug console window.
|
||||
-h --help Show this screen.
|
||||
-v --version Show version.`
|
||||
|
@ -55,224 +44,20 @@ func main() {
|
|||
defer w.Destroy()
|
||||
w.SetTitle("RiV-mesh")
|
||||
w.SetSize(690, 920, webview.HintFixed)
|
||||
/*1. Create ~/.riv-mesh folder if not existing
|
||||
*2. Create ~/.riv-mesh/mesh.conf if not existing
|
||||
*3. If the file exists read Peers.
|
||||
*3.1 Invoke add peers for each record
|
||||
*/
|
||||
mesh_folder := ".riv-mesh"
|
||||
mesh_conf := "mesh.conf"
|
||||
user_home := get_user_home_path()
|
||||
mesh_settings_folder := filepath.Join(user_home, mesh_folder)
|
||||
err := os.MkdirAll(mesh_settings_folder, os.ModePerm)
|
||||
if err != nil {
|
||||
fmt.Printf("Unable to create folder: %v", err)
|
||||
}
|
||||
mesh_settings_path := filepath.Join(user_home, mesh_folder, mesh_conf)
|
||||
if _, err := os.Stat(mesh_settings_path); os.IsNotExist(err) {
|
||||
err := ioutil.WriteFile(mesh_settings_path, []byte(""), 0750)
|
||||
if err != nil {
|
||||
fmt.Printf("Unable to write file: %v", err)
|
||||
}
|
||||
} else {
|
||||
//read peers from mesh.conf
|
||||
conf, _ := ioutil.ReadFile(mesh_settings_path)
|
||||
var dat map[string]interface{}
|
||||
if err := hjson.Unmarshal(conf, &dat); err != nil {
|
||||
fmt.Printf("Unable to parse mesh.conf file: %v", err)
|
||||
} else {
|
||||
if dat["Peers"] != nil {
|
||||
peers := dat["Peers"].([]interface{})
|
||||
remove_peers()
|
||||
for _, u := range peers {
|
||||
log.Printf("Unmarshaled: %v", u.(string))
|
||||
add_peers(u.(string))
|
||||
}
|
||||
} else {
|
||||
fmt.Printf("Warning: Peers array not loaded from mesh.conf file")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if confui.IndexHtml == "" {
|
||||
confui.IndexHtml = "index.html"
|
||||
confui.IndexHtml = getEndpoint()
|
||||
}
|
||||
//Check is it URL already
|
||||
indexUrl, err := url.ParseRequestURI(confui.IndexHtml)
|
||||
if err != nil || len(indexUrl.Scheme) < 2 { // handling no scheme at all and windows c:\ as scheme detection
|
||||
confui.IndexHtml, err = filepath.Abs(confui.IndexHtml)
|
||||
if err != nil {
|
||||
panic(errors.New("Index file not found: " + err.Error()))
|
||||
}
|
||||
if stat, err := os.Stat(confui.IndexHtml); err != nil {
|
||||
panic(errors.New(fmt.Sprintf("Index file %v not found or permissians denied: %v", confui.IndexHtml, err.Error())))
|
||||
} else if stat.IsDir() {
|
||||
panic(errors.New(fmt.Sprintf("Index file %v not found", confui.IndexHtml)))
|
||||
}
|
||||
path_prefix := ""
|
||||
if indexUrl != nil && len(indexUrl.Scheme) == 1 {
|
||||
path_prefix = "/"
|
||||
}
|
||||
indexUrl, err = url.ParseRequestURI("file://" + path_prefix + strings.ReplaceAll(confui.IndexHtml, "\\", "/"))
|
||||
if err != nil {
|
||||
panic(errors.New("Index file URL parse error: " + err.Error()))
|
||||
}
|
||||
if confui.IndexHtml == "" {
|
||||
confui.IndexHtml = "http://localhost:19019"
|
||||
}
|
||||
|
||||
w.Bind("savePeers", func(peer_list string) {
|
||||
//log.Println("peers saved ", peer_list)
|
||||
var peers []string
|
||||
_ = json.Unmarshal([]byte(peer_list), &peers)
|
||||
log.Printf("Unmarshaled: %v", peers)
|
||||
remove_peers()
|
||||
for _, u := range peers {
|
||||
log.Printf("Unmarshaled: %v", u)
|
||||
add_peers(u)
|
||||
}
|
||||
//add peers to ~/mesh.conf
|
||||
dat := make(map[string]interface{})
|
||||
dat["Peers"] = peers
|
||||
bs, _ := hjson.Marshal(dat)
|
||||
e := ioutil.WriteFile(mesh_settings_path, bs, 0750)
|
||||
if e != nil {
|
||||
fmt.Printf("Unable to write file: %v", e)
|
||||
}
|
||||
})
|
||||
w.Bind("ping", func(peer_list string) {
|
||||
go ping(w, peer_list)
|
||||
})
|
||||
log.Printf("Opening: %v", indexUrl)
|
||||
w.Navigate(indexUrl.String())
|
||||
log.Printf("Opening: %v", confui.IndexHtml)
|
||||
w.Navigate(confui.IndexHtml)
|
||||
w.Run()
|
||||
}
|
||||
|
||||
func ping(w webview.WebView, peer_list string) {
|
||||
var peers []string
|
||||
_ = json.Unmarshal([]byte(peer_list), &peers)
|
||||
log.Printf("Unmarshaled: %v", peers)
|
||||
for _, u := range peers {
|
||||
log.Printf("Unmarshaled: %v", u)
|
||||
ping_time := check(u)
|
||||
log.Printf("ping: %d", ping_time)
|
||||
setPingValue(w, u, strconv.FormatInt(ping_time, 10))
|
||||
}
|
||||
}
|
||||
|
||||
func check(peer string) int64 {
|
||||
u, e := url.Parse(peer)
|
||||
if e != nil {
|
||||
return -1
|
||||
}
|
||||
t := time.Now()
|
||||
_, err := net.DialTimeout("tcp", u.Host, 5*time.Second)
|
||||
if err != nil {
|
||||
return -1
|
||||
}
|
||||
d := time.Since(t)
|
||||
return d.Milliseconds()
|
||||
}
|
||||
|
||||
func add_peers(uri string) {
|
||||
_, err := run_command_with_arg("addpeers", "uri="+uri)
|
||||
if err != nil {
|
||||
log.Println("Error in the add_peers() call:", err)
|
||||
}
|
||||
}
|
||||
|
||||
func remove_peers() {
|
||||
run_command("removepeers")
|
||||
}
|
||||
|
||||
func setPingValue(p webview.WebView, peer string, value string) {
|
||||
p.Dispatch(func() {
|
||||
p.Eval("setPingValue('" + peer + "','" + value + "');")
|
||||
})
|
||||
}
|
||||
|
||||
func run_command(command string) []byte {
|
||||
data, err := run_command_with_arg(command, "")
|
||||
if err != nil {
|
||||
log.Println("Error in the "+command+" call:", err)
|
||||
}
|
||||
return data
|
||||
}
|
||||
|
||||
func run_command_with_arg(command string, arg string) ([]byte, error) {
|
||||
logbuffer := &bytes.Buffer{}
|
||||
logger := log.New(logbuffer, "", log.Flags())
|
||||
|
||||
wrapErr := func(err error) error {
|
||||
logger.Println("Error:", err)
|
||||
return errors.New(fmt.Sprintln(logbuffer))
|
||||
}
|
||||
|
||||
endpoint := getEndpoint(logger)
|
||||
|
||||
var conn net.Conn
|
||||
u, err := url.Parse(endpoint)
|
||||
d := net.Dialer{Timeout: 5000 * time.Millisecond}
|
||||
if err == nil {
|
||||
switch strings.ToLower(u.Scheme) {
|
||||
case "unix":
|
||||
logger.Println("Connecting to UNIX socket", endpoint[7:])
|
||||
conn, err = d.Dial("unix", endpoint[7:])
|
||||
case "tcp":
|
||||
logger.Println("Connecting to TCP socket", u.Host)
|
||||
conn, err = d.Dial("tcp", u.Host)
|
||||
default:
|
||||
logger.Println("Unknown protocol or malformed address - check your endpoint")
|
||||
err = errors.New("protocol not supported")
|
||||
}
|
||||
} else {
|
||||
logger.Println("Connecting to TCP socket", u.Host)
|
||||
conn, err = d.Dial("tcp", endpoint)
|
||||
}
|
||||
if err != nil {
|
||||
return nil, wrapErr(err)
|
||||
}
|
||||
|
||||
logger.Println("Connected")
|
||||
defer conn.Close()
|
||||
|
||||
decoder := json.NewDecoder(conn)
|
||||
encoder := json.NewEncoder(conn)
|
||||
send := &admin.AdminSocketRequest{}
|
||||
recv := &admin.AdminSocketResponse{}
|
||||
send.Name = command
|
||||
args := map[string]string{}
|
||||
switch {
|
||||
case len(arg) > 0:
|
||||
tokens := strings.SplitN(arg, "=", 2)
|
||||
args[tokens[0]] = tokens[1]
|
||||
default:
|
||||
}
|
||||
|
||||
if send.Arguments, err = json.Marshal(args); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := encoder.Encode(&send); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
logger.Printf("Request sent")
|
||||
//js, _ := json.Marshal(send)
|
||||
//fmt.Println("sent:", string(js))
|
||||
if err := decoder.Decode(&recv); err != nil {
|
||||
return nil, wrapErr(err)
|
||||
}
|
||||
if recv.Status == "error" {
|
||||
if err := recv.Error; err != "" {
|
||||
return nil, wrapErr(errors.New("Admin socket returned an error:" + err))
|
||||
} else {
|
||||
return nil, wrapErr(errors.New("Admin socket returned an error but didn't specify any error text"))
|
||||
}
|
||||
}
|
||||
if json, err := json.MarshalIndent(recv.Response, "", " "); err == nil {
|
||||
return json, nil
|
||||
}
|
||||
return nil, wrapErr(err)
|
||||
}
|
||||
|
||||
func getEndpoint(logger *log.Logger) string {
|
||||
func getEndpoint() string {
|
||||
if config, err := os.ReadFile(defaults.GetDefaults().DefaultConfigFile); err == nil {
|
||||
if bytes.Equal(config[0:2], []byte{0xFF, 0xFE}) ||
|
||||
bytes.Equal(config[0:2], []byte{0xFE, 0xFF}) {
|
||||
|
@ -280,25 +65,16 @@ func getEndpoint(logger *log.Logger) string {
|
|||
decoder := utf.NewDecoder()
|
||||
config, err = decoder.Bytes(config)
|
||||
if err != nil {
|
||||
return defaults.GetDefaults().DefaultAdminListen
|
||||
return ""
|
||||
}
|
||||
}
|
||||
var dat map[string]interface{}
|
||||
if err := hjson.Unmarshal(config, &dat); err != nil {
|
||||
return defaults.GetDefaults().DefaultAdminListen
|
||||
return ""
|
||||
}
|
||||
if ep, ok := dat["AdminListen"].(string); ok && (ep != "none" && ep != "") {
|
||||
logger.Println("Found platform default config file", defaults.GetDefaults().DefaultConfigFile)
|
||||
logger.Println("Using endpoint", ep, "from AdminListen")
|
||||
if ep, ok := dat["HttpAddress"].(string); ok && (ep != "none" && ep != "") {
|
||||
return ep
|
||||
} else {
|
||||
logger.Println("Configuration file doesn't contain appropriate AdminListen option")
|
||||
logger.Println("Falling back to platform default", defaults.GetDefaults().DefaultAdminListen)
|
||||
return defaults.GetDefaults().DefaultAdminListen
|
||||
}
|
||||
} else {
|
||||
logger.Println("Can't open config file from default location", defaults.GetDefaults().DefaultConfigFile)
|
||||
logger.Println("Falling back to platform default", defaults.GetDefaults().DefaultAdminListen)
|
||||
return defaults.GetDefaults().DefaultAdminListen
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue