whispervis/network_selector.go

package main

import (
	"bytes"
	"fmt"

	"github.com/gopherjs/vecty"
	"github.com/gopherjs/vecty/elem"
	"github.com/gopherjs/vecty/event"
	"github.com/status-im/whispervis/widgets"
)

// NetworkSelector represents widget for choosing or uploading network topology
// to be used for visualization.
type NetworkSelector struct {
	vecty.Core

	current  *Network
	isCustom bool
	networks map[string]*Network

	upload *widgets.UploadWidget

	handler func(*Network) // executed on network change
}

// NewNetworkSelector creates new NetworkSelector.
func NewNetworkSelector(handler func(*Network)) *NetworkSelector {
	current := &Network{}
	networks, err := LoadNetworks()
	if err != nil {
		fmt.Println("No networks loaded:", err)
	} else {
		current = networks["net100.json"]
	}

	ns := &NetworkSelector{
		networks: networks,
		current:  current,
		handler:  handler,
	}
	ns.upload = widgets.NewUploadWidget(ns.onUpload)
	ns.setCurrentNetwork(current)
	return ns
}

// Render implements the vecty.Component interface.
func (n *NetworkSelector) Render() vecty.ComponentOrHTML {
	return elem.Div(
		elem.Heading3(vecty.Text("Choose network:")),
		elem.Select(
			vecty.Markup(
				event.Change(n.onChange),
			),
			n.networkOptions(),
			elem.Option(
				vecty.Markup(
					vecty.Property("value", "upload"),
					vecty.Property("selected", n.isCustom),
				),
				vecty.Text("Upload custom..."),
			),
		),
		n.descriptionBlock(),
		vecty.If(n.isCustom, n.upload),
	)
}

// descriptionBlock renders the block with network description.
func (n *NetworkSelector) descriptionBlock() *vecty.HTML {
	return elem.Div(
		vecty.Markup(
			vecty.Style("padding", "7px"),
			vecty.Style("font-style", "italic"),
			vecty.Style("color", "blue"),
		),
		vecty.Text(n.current.Description),
	)
}

// LoadNetworks imports preloaded neworks from the directory with JSON files.
func LoadNetworks() (map[string]*Network, error) {
	files, err := AssetDir("data")
	if err != nil {
		return nil, err
	}

	networks := map[string]*Network{}
	for _, file := range files {
		network, err := LoadNetwork("data/" + file)
		if err != nil {
			return nil, fmt.Errorf("load network: %v", err)
		}

		networks[file] = network
	}
	return networks, nil
}

// networkOptions renders 'option' elements for network 'select' input tag.
func (n *NetworkSelector) networkOptions() vecty.List {
	var options vecty.List
	for name := range n.networks {
		options = append(options, elem.Option(
			vecty.Markup(
				vecty.Property("value", name),
				vecty.Property("selected", n.current.Name == "data/"+name), // TODO(divan): get rid of "data"
			),
			vecty.Text(name),
		))
	}
	return options
}

// onChange implements handler for select input changed value
func (n *NetworkSelector) onChange(e *vecty.Event) {
	value := e.Target.Get("value").String()

	if value == "upload" {
		n.isCustom = true
		vecty.Rerender(n)
		return
	}

	n.isCustom = false

	net := n.networks[value]
	n.setCurrentNetwork(net)

	vecty.Rerender(n)
}

// onUpload implements callback for "Upload" button clicked event.
func (n *NetworkSelector) onUpload(json []byte) {
	r := bytes.NewReader(json)
	net, err := LoadNetworkFromReader(r)
	if err != nil {
		fmt.Printf("[ERROR] Load network: %v", err)
	}

	net.Name = fmt.Sprintf("Uploaded (%d nodes)", net.NodesCount())

	n.networks[net.Name] = net
	n.setCurrentNetwork(net)

	vecty.Rerender(n)
}

// setCurrentNetwork changes current network and runs needed update handlers.
func (n *NetworkSelector) setCurrentNetwork(net *Network) {
	n.current = net

	if n.handler != nil {
		go n.handler(n.current)
	}
}