Shit

Author: tjayrush

app/app.go

@@ -3,28 +3,30 @@ package app
 import (
 	"context"
 	"encoding/json"
+	"os"
+	"path/filepath"
 	"sync"
 
 	"github.com/TrueBlocks/trueblocks-browse/pkg/daemons"
 	"github.com/TrueBlocks/trueblocks-browse/pkg/messages"
 	"github.com/TrueBlocks/trueblocks-browse/pkg/types"
 	"github.com/TrueBlocks/trueblocks-core/src/apps/chifra/pkg/base"
 	coreConfig "github.com/TrueBlocks/trueblocks-core/src/apps/chifra/pkg/config"
-	configTypes "github.com/TrueBlocks/trueblocks-core/src/apps/chifra/pkg/configtypes"
+	"github.com/TrueBlocks/trueblocks-core/src/apps/chifra/pkg/file"
 	"github.com/TrueBlocks/trueblocks-core/src/apps/chifra/pkg/logger"
 	"github.com/TrueBlocks/trueblocks-core/src/apps/chifra/pkg/output"
 	coreTypes "github.com/TrueBlocks/trueblocks-core/src/apps/chifra/pkg/types"
-	coreUtils "github.com/TrueBlocks/trueblocks-core/src/apps/chifra/pkg/utils"
+	"github.com/TrueBlocks/trueblocks-core/src/apps/chifra/pkg/utils"
 	sdk "github.com/TrueBlocks/trueblocks-sdk/v3"
 	"github.com/wailsapp/wails/v2/pkg/runtime"
 )
 
 type App struct {
 	sdk.Globals `json:",inline"`
 
-	ctx  context.Context
-	cfg  configTypes.Config
-	meta coreTypes.MetaData
+	ctx   context.Context
+	meta  coreTypes.MetaData
+	dirty bool
 
 	renderCtxs map[base.Address][]*output.RenderCtx
 
@@ -35,10 +37,15 @@ type App struct {
 	abis      types.AbiContainer
 	indexes   types.IndexContainer
 	manifests types.ManifestContainer
-	settings  types.SettingsGroup
-	configs   types.ConfigContainer
 	status    types.StatusContainer
-	sessions  types.SessionContainer
+	settings  types.SettingsGroup
+	session   types.SessionContainer
+	config    types.ConfigContainer
+
+	// Memory caches
+	HistoryCache *types.HistoryMap `json:"historyCache"`
+	EnsCache     *sync.Map         `json:"ensCache"`
+	BalanceCache *sync.Map         `json:"balanceCache"`
 
 	// Controllers
 	ScraperController *daemons.DaemonScraper
@@ -50,9 +57,12 @@ func NewApp() *App {
 	a := App{
 		renderCtxs: make(map[base.Address][]*output.RenderCtx),
 	}
+	a.EnsCache = &sync.Map{}
+	a.BalanceCache = &sync.Map{}
+	a.HistoryCache = &types.HistoryMap{}
 	a.names.NamesMap = make(map[base.Address]coreTypes.Name)
-	a.projects = types.NewProjectContainer("Untitled.tbx", &types.HistoryMap{}, &sync.Map{}, &sync.Map{})
-	a.sessions.LastSub = make(map[string]string)
+	a.projects = types.NewProjectContainer("", []types.HistoryContainer{})
+	a.session.LastSub = make(map[string]string)
 
 	return &a
 }
@@ -65,14 +75,21 @@ func (a *App) String() string {
 func (a *App) Startup(ctx context.Context) {
 	a.ctx = ctx
 	a.loadSession()
-
-	go a.loadHistory(a.GetAddress(), nil, nil)
+	a.Chain = a.session.LastChain
 
 	a.FreshenController = daemons.NewFreshen(a, "freshen", 3000, a.IsShowing("freshen"))
 	a.ScraperController = daemons.NewScraper(a, "scraper", 7000, a.IsShowing("scraper"))
 	a.IpfsController = daemons.NewIpfs(a, "ipfs", 10000, a.IsShowing("ipfs"))
 	go a.startDaemons()
 
+	fn := filepath.Join(a.session.LastFolder, a.session.LastFile)
+	if file.FileExists(fn) {
+		a.LoadFile(fn)
+		a.dirty = false
+	} else {
+		a.dirty = true
+	}
+
 	logger.Info("Starting freshen process...")
 	_ = a.Refresh()
 }
@@ -83,12 +100,12 @@ func (a *App) DomReady(ctx context.Context) {
 	runtime.WindowSetSize(a.ctx, win.Width, win.Height)
 	runtime.WindowShow(a.ctx)
 
-	if path, err := coreUtils.GetConfigFn("", "trueBlocks.toml"); err != nil {
+	if path, err := utils.GetConfigFn("", "trueBlocks.toml"); err != nil {
 		messages.EmitMessage(a.ctx, messages.Error, &messages.MessageMsg{
 			String1: err.Error(),
 		})
 	} else {
-		if err := coreConfig.ReadToml(path, &a.cfg); err != nil {
+		if err := coreConfig.ReadToml(path, &a.config.Config); err != nil {
 			messages.EmitMessage(a.ctx, messages.Error, &messages.MessageMsg{
 				String1: err.Error(),
 			})
@@ -101,18 +118,26 @@ func (a *App) Shutdown(ctx context.Context) {
 }
 
 func (a *App) saveSession() {
-	a.sessions.Window.X, a.sessions.Window.Y = runtime.WindowGetPosition(a.ctx)
-	a.sessions.Window.Width, a.sessions.Window.Height = runtime.WindowGetSize(a.ctx)
-	a.sessions.Window.Y += 38 // TODO: This is a hack to account for the menu bar - not sure why it's needed
-	_ = a.sessions.Save()
+	if !isTesting {
+		a.session.Window.X, a.session.Window.Y = runtime.WindowGetPosition(a.ctx)
+		a.session.Window.Width, a.session.Window.Height = runtime.WindowGetSize(a.ctx)
+		a.session.Window.Y += 38 // TODO: This is a hack to account for the menu bar - not sure why it's needed
+	}
+	_ = a.session.Save()
 }
 
 func (a *App) loadSession() {
-	_ = a.sessions.Load()
-	a.sessions.CleanWindowSize(a.ctx)
-	a.Chain = a.sessions.LastChain
+	_ = a.session.Load()
+	a.session.CleanWindowSize(a.ctx)
+	a.Chain = a.session.LastChain
 }
 
 func (a *App) Logger(msg string) {
 	logger.Info(msg)
 }
+
+var isTesting bool
+
+func init() {
+	isTesting = os.Getenv("TB_TEST_MODE") == "true"
+}

app/app_getters.go

@@ -11,23 +11,23 @@ import (
 )
 
 func (a *App) IsShowing(which string) bool {
-	return a.sessions.Toggles.IsOn(which)
+	return a.session.Toggles.IsOn(which)
 }
 
 func (a *App) GetConfig() *configTypes.Config {
-	return &a.cfg
+	return &a.config.Config
 }
 
 func (a *App) GetSession() *coreTypes.Session {
-	return &a.sessions.Session
+	return &a.session.Session
 }
 
 func (a *App) GetContext() context.Context {
 	return a.ctx
 }
 
 func (a *App) GetWindow() *coreTypes.Window {
-	return &a.sessions.Window
+	return &a.session.Window
 }
 
 func (a *App) GetEnv(key string) string {
@@ -39,24 +39,24 @@ func (a *App) GetMeta() coreTypes.MetaData {
 }
 
 func (a *App) GetAppTitle() string {
-	return a.sessions.Window.Title
+	return a.session.Window.Title
 }
 
 func (a *App) GetRoute() string {
 	if !a.IsConfigured() {
 		return "/wizard"
 	}
 
-	route := a.sessions.LastRoute
-	if len(a.sessions.LastSub) > 0 {
-		route += "/" + a.sessions.LastSub[route]
+	route := a.session.LastRoute
+	if len(a.session.LastSub) > 0 {
+		route += "/" + a.session.LastSub[route]
 	}
 
 	return route
 }
 
 func (a *App) GetAddress() base.Address {
-	addr := a.sessions.LastSub["/history"]
+	addr := a.session.LastSub["/history"]
 	return base.HexToAddress(addr)
 }
 
@@ -66,7 +66,7 @@ func (a *App) GetChain() string {
 
 func (a *App) GetChains() []string {
 	ret := []string{}
-	for _, chain := range a.cfg.Chains {
+	for _, chain := range a.GetConfig().Chains {
 		ret = append(ret, chain.Chain)
 	}
 	sort.Strings(ret)

app/app_setters.go

@@ -8,7 +8,7 @@ import (
 )
 
 func (a *App) SetShowing(which string, onOff bool) {
-	a.sessions.Toggles.SetState(which, onOff)
+	a.session.Toggles.SetState(which, onOff)
 	a.saveSession()
 }
 
@@ -17,14 +17,14 @@ func (a *App) SetEnv(key, value string) {
 }
 
 func (a *App) SetRoute(route, subRoute string) {
-	a.sessions.SetRoute(route, subRoute)
+	a.session.SetRoute(route, subRoute)
 	a.saveSession()
 }
 
 func (a *App) SetChain(chain string, address base.Address) {
 	a.CancelAllContexts() // cancel what's happening on the old chain
 	a.Chain = chain
-	a.sessions.LastChain = chain
+	a.session.LastChain = chain
 	a.saveSession()
 	a.Reload(address)
 	a.monitors = types.MonitorContainer{}

app/data_abi.go

@@ -1,3 +1,5 @@
+// This file is auto-generated. Edit only code inside
+// of ExistingCode markers (if any).
 package app
 
 // EXISTING_CODE
@@ -81,6 +83,7 @@ func (a *App) loadAbis(wg *sync.WaitGroup, errorChan chan error) error {
 		a.abis.Summarize()
 		messages.EmitMessage(a.ctx, messages.Info, &messages.MessageMsg{String1: "Loaded abis"})
 	}
+
 	return nil
 }