treeview.go

package gui

import (
	"encoding/json"
	"io/ioutil"
	"os"

	. "github.com/lxn/walk/declarative"
	"github.com/tinycedar/lily/common"
	"github.com/tinycedar/lily/conf"
	"github.com/tinycedar/lily/core"
	"github.com/tinycedar/lily/model"
)

func newTreeView() TreeView {
	treeModel := conf.Config.HostConfigModel
	return TreeView{
		AssignTo: &(context.treeView),
		Model:    treeModel,
		Font: Font{
			Family:    "MS Shell Dlg 2",
			PointSize: 11,
			Bold:      false,
		},
		// click
		OnCurrentItemChanged: func() {
			context.deleteButton.SetEnabled(true)
			current := context.treeView.CurrentItem().(*model.HostConfigItem)
			if bytes, err := ioutil.ReadFile("conf/hosts/" + current.Text() + ".hosts"); err == nil {
				context.hostConfigText.SetText(string(bytes))
			} else {
				context.hostConfigText.SetText("")
			}
		},
		StretchFactor: 1,
		// double click
		OnItemActivated: func() {
			current := context.treeView.CurrentItem().(*model.HostConfigItem)
			for i := 0; i < treeModel.RootCount(); i++ {
				item := treeModel.RootAt(i).(*model.HostConfigItem)
				if item == current {
					conf.Config.CurrentHostIndex = i
					item.Icon = common.IconMap[common.ICON_OPEN]
				} else {
					item.Icon = common.IconMap[common.ICON_NEW]
				}
				treeModel.PublishItemChanged(item)
			}
			configJSON, err := json.Marshal(conf.Config)
			if err != nil {
				common.Error("Error marshal json: %v", err)
			} else {
				ioutil.WriteFile("conf/config.json", configJSON, os.ModeExclusive)
			}
			if core.FireHostsSwitch() {
				context.notifyIcon.ShowCustom("切换"+current.Text()+"成功！", "Best hosts manager ever")
			} else {
				context.notifyIcon.ShowError("切换"+current.Text()+"失败！", "Best hosts manager ever")
			}
		},
	}
}