diff --git a/src/popup/index.js b/src/popup/index.js index 5c98a60..0a93870 100644 --- a/src/popup/index.js +++ b/src/popup/index.js @@ -2,7 +2,12 @@ // Loads state, initializes views, triggers first render. const { DEBUG } = require("../shared/constants"); -const { state, saveState, loadState } = require("../shared/state"); +const { + state, + saveState, + loadState, + currentNetwork, +} = require("../shared/state"); const { refreshPrices } = require("../shared/prices"); const { refreshBalances } = require("../shared/balances"); const { $, showView } = require("./views/helpers"); @@ -167,18 +172,25 @@ function fallbackView() { } async function init() { - if (DEBUG) { + await loadState(); + applyTheme(state.theme); + + const net = currentNetwork(); + if (DEBUG || net.isTestnet) { const banner = document.createElement("div"); banner.id = "debug-banner"; - banner.textContent = "DEBUG / INSECURE"; + if (DEBUG && net.isTestnet) { + banner.textContent = "DEBUG / INSECURE [TESTNET]"; + } else if (net.isTestnet) { + banner.textContent = "[TESTNET]"; + } else { + banner.textContent = "DEBUG / INSECURE"; + } banner.style.cssText = "background:#c00;color:#fff;text-align:center;font-size:10px;padding:1px 0;font-family:monospace;position:sticky;top:0;z-index:9999;"; document.body.prepend(banner); } - await loadState(); - applyTheme(state.theme); - // Auto-default active address if ( state.activeAddress === null && diff --git a/src/popup/views/helpers.js b/src/popup/views/helpers.js index fe8441c..6560ee5 100644 --- a/src/popup/views/helpers.js +++ b/src/popup/views/helpers.js @@ -59,10 +59,18 @@ function showView(name) { clearFlash(); state.currentView = name; saveState(); - if (DEBUG) { + const net = currentNetwork(); + if (DEBUG || net.isTestnet) { const banner = document.getElementById("debug-banner"); if (banner) { - banner.textContent = "DEBUG / INSECURE (" + name + ")"; + if (DEBUG && net.isTestnet) { + banner.textContent = + "DEBUG / INSECURE [TESTNET] (" + name + ")"; + } else if (net.isTestnet) { + banner.textContent = "[TESTNET]"; + } else { + banner.textContent = "DEBUG / INSECURE (" + name + ")"; + } } } }