Dashboard.java

import javax.swing.*;
import javax.swing.border.*;
import java.awt.*;
import java.awt.event.*;

public class Dashboard {

    private final JFrame frame;
    private final JLabel walletABalanceLabel;
    private final JLabel walletBBalanceLabel;
    private final JLabel mempoolLabel;
    private final JLabel difficultyLabel;
    private final JLabel blockHeightLabel;
    private final JLabel rewardLabel;
    private final JLabel statusLabel;
    private final JTextField amountField;
    private final JTextArea logArea;
    private final JComboBox<String> directionCombo;
    private final JComboBox<String> minerCombo;

    public Dashboard() {
        setLookAndFeel();

        frame = new JFrame("NoobChain Dashboard");
        frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        frame.setLayout(new BorderLayout(10, 10));
        frame.getContentPane().setBackground(new Color(18, 20, 26));

        JPanel root = new JPanel(new BorderLayout(10, 10));
        root.setBorder(BorderFactory.createEmptyBorder(14, 14, 14, 14));
        root.setBackground(new Color(18, 20, 26));


        JPanel topRow = new JPanel(new GridLayout(1, 2, 10, 0));
        topRow.setOpaque(false);

        walletABalanceLabel = new JLabel();
        walletBBalanceLabel = new JLabel();
        topRow.add(makeBalanceCard("Wallet A", walletABalanceLabel, new Color(39, 174, 96)));
        topRow.add(makeBalanceCard("Wallet B", walletBBalanceLabel, new Color(52, 152, 219)));


        JPanel statsBar = new JPanel(new GridLayout(1, 4, 8, 0));
        statsBar.setOpaque(false);

        mempoolLabel    = new JLabel("0", SwingConstants.CENTER);
        difficultyLabel = new JLabel("0", SwingConstants.CENTER);
        blockHeightLabel= new JLabel("0", SwingConstants.CENTER);
        rewardLabel     = new JLabel("0", SwingConstants.CENTER);

        statsBar.add(makeStatCard("Mempool",    mempoolLabel,     new Color(243, 156, 18)));
        statsBar.add(makeStatCard("Difficulty", difficultyLabel,  new Color(155, 89, 182)));
        statsBar.add(makeStatCard("Height",     blockHeightLabel, new Color(52, 152, 219)));
        statsBar.add(makeStatCard("Reward",     rewardLabel,      new Color(39, 174, 96)));


        JPanel headerPanel = new JPanel(new BorderLayout(0, 8));
        headerPanel.setOpaque(false);
        headerPanel.add(topRow,   BorderLayout.NORTH);
        headerPanel.add(statsBar, BorderLayout.SOUTH);


        logArea = new JTextArea();
        logArea.setEditable(false);
        logArea.setLineWrap(true);
        logArea.setWrapStyleWord(true);
        logArea.setFont(new Font("Consolas", Font.PLAIN, 12));
        logArea.setBackground(new Color(13, 15, 20));
        logArea.setForeground(new Color(200, 210, 220));
        logArea.setBorder(BorderFactory.createEmptyBorder(10, 10, 10, 10));
        logArea.setCaretColor(new Color(39, 174, 96));

        JScrollPane scroll = new JScrollPane(logArea);
        scroll.setBorder(makeTitledBorder("Activity Log"));
        scroll.setPreferredSize(new Dimension(820, 300));
        scroll.getVerticalScrollBar().setUnitIncrement(16);


        JPanel row1 = new JPanel(new FlowLayout(FlowLayout.LEFT, 8, 0));
        row1.setOpaque(false);

        amountField = new JTextField(10);
        amountField.setFont(new Font("Consolas", Font.PLAIN, 14));
        amountField.setBackground(new Color(28, 32, 40));
        amountField.setForeground(Color.WHITE);
        amountField.setCaretColor(Color.WHITE);
        amountField.setBorder(BorderFactory.createCompoundBorder(
                BorderFactory.createLineBorder(new Color(60, 70, 90)),
                BorderFactory.createEmptyBorder(6, 8, 6, 8)
        ));

        directionCombo = new JComboBox<>(new String[]{"A → B", "B → A"});
        styleCombo(directionCombo);

        JButton sendBtn     = makeButton("Submit to Mempool", new Color(39, 174, 96));
        JButton mineBtn     = makeButton("Mine Block",        new Color(52, 152, 219));
        JButton validateBtn = makeButton("Validate Chain",    new Color(155, 89, 182));
        JButton clearBtn    = makeButton("Clear Log",         new Color(80, 90, 110));

        row1.add(new JLabel(styledLabel("Amount:")));
        row1.add(amountField);
        row1.add(new JLabel(styledLabel("Direction:")));
        row1.add(directionCombo);
        row1.add(sendBtn);

        // Row 2: miner selection + mine + validate + clear
        JPanel row2 = new JPanel(new FlowLayout(FlowLayout.LEFT, 8, 0));
        row2.setOpaque(false);

        minerCombo = new JComboBox<>(new String[]{"Wallet A mines", "Wallet B mines"});
        styleCombo(minerCombo);

        row2.add(new JLabel(styledLabel("Miner:")));
        row2.add(minerCombo);
        row2.add(mineBtn);
        row2.add(validateBtn);
        row2.add(clearBtn);

        // Status bar
        statusLabel = new JLabel("Ready — submit a transaction or mine a block.");
        statusLabel.setFont(new Font("Consolas", Font.PLAIN, 12));
        statusLabel.setForeground(new Color(140, 160, 180));
        statusLabel.setBorder(BorderFactory.createEmptyBorder(6, 2, 0, 0));

        JPanel controlsPanel = new JPanel(new BorderLayout(0, 6));
        controlsPanel.setOpaque(false);
        controlsPanel.setBorder(makeTitledBorder("Controls"));

        JPanel rows = new JPanel(new GridLayout(2, 1, 0, 6));
        rows.setOpaque(false);
        rows.add(row1);
        rows.add(row2);

        controlsPanel.add(rows,        BorderLayout.CENTER);
        controlsPanel.add(statusLabel, BorderLayout.SOUTH);


        root.add(headerPanel,   BorderLayout.NORTH);
        root.add(scroll,        BorderLayout.CENTER);
        root.add(controlsPanel, BorderLayout.SOUTH);
        frame.add(root, BorderLayout.CENTER);


        sendBtn.addActionListener(e -> handleSend());
        mineBtn.addActionListener(e -> handleMine());
        validateBtn.addActionListener(e -> handleValidate());
        clearBtn.addActionListener(e -> logArea.setText(""));


        amountField.addActionListener(e -> handleSend());


        new Timer(1000, e -> refreshAll()).start();

        refreshAll();
        log("INFO", "Dashboard initialized. Blockchain height: " + Noob.blockchain.size());

        frame.pack();
        frame.setMinimumSize(new Dimension(860, 600));
        frame.setLocationRelativeTo(null);
        frame.setVisible(true);
    }



    private void handleSend() {
        if (!isReady()) return;
        try {
            float amount = Float.parseFloat(amountField.getText().trim());
            if (amount <= 0f) { showError("Amount must be > 0."); return; }

            boolean aToB = directionCombo.getSelectedIndex() == 0;
            Wallet sender    = aToB ? Noob.walletA : Noob.walletB;
            Wallet recipient = aToB ? Noob.walletB : Noob.walletA;
            String senderName    = aToB ? "A" : "B";
            String recipientName = aToB ? "B" : "A";

            // sendFunds() now auto-submits to mempool
            Transaction tx = sender.sendFunds(recipient.publicKey, (long)(amount * 100), (long)(Noob.miningReward * 10)); // fee = 10x reward for demo purposes

            if (tx == null) {
                showError("Insufficient funds — Wallet " + senderName + " has "
                        + String.format("%.2f", sender.getBalance()));
                return;
            }

            log("TX", String.format("Wallet %s → Wallet %s : %.2f  |  Mempool: %d pending",
                    senderName, recipientName, amount, Noob.mempool.size()));
            setStatus("Transaction pending in mempool. Click Mine Block to confirm.");
            amountField.setText("");
            refreshAll();

        } catch (NumberFormatException ex) {
            showError("Invalid amount — enter a number like 10 or 25.5");
        }
    }

    private void handleMine() {
        if (!isReady()) return;

        boolean aIsMinier = minerCombo.getSelectedIndex() == 0;
        Wallet miner     = aIsMinier ? Noob.walletA : Noob.walletB;
        String minerName = aIsMinier ? "A" : "B";

        if (Noob.mempool.isEmpty()) {
            log("MINE", "Mempool empty — mining reward-only block for Wallet " + minerName);
        } else {
            log("MINE", "Mining " + Noob.mempool.size() + " pending TX(s) | Miner: Wallet " + minerName);
        }

        String prevHash = Noob.blockchain.get(Noob.blockchain.size() - 1).hash;

        try {

            Block mined = Noob.mineNextBlock(prevHash, miner);

            if (mined != null) {
                log("MINE", String.format(
                        "Block #%d mined | hash: %s... | difficulty: %d | reward: %.1f → Wallet %s",
                        Noob.blockchain.size() - 1,
                        mined.hash.substring(0, 12),
                        mined.difficulty,
                        Noob.miningReward,
                        minerName
                ));
                setStatus("Block mined. Difficulty now: " + Noob.difficulty);
            }
            refreshAll();

        } catch (Exception ex) {
            showError("Mining failed: " + ex.getMessage());
        }
    }

    private void handleValidate() {
        if (!isReady()) return;
        log("VALIDATE", "Validating chain of " + Noob.blockchain.size() + " blocks...");
        boolean valid = Noob.isChainValid();
        if (valid) {
            log("VALID", "All " + Noob.blockchain.size() + " blocks passed validation.");
            setStatus("Blockchain is valid ✓");
        } else {
            log("INVALID", "Chain validation FAILED — see console for details.");
            setStatus("Chain validation FAILED.");
        }
    }



    private void refreshAll() {
        // Balances
        String a = Noob.walletA != null ? StringUtil.toCoins(Noob.walletA.getBalance()): "-";
        String b = Noob.walletB != null ? StringUtil.toCoins(Noob.walletB.getBalance()) : "—";
        walletABalanceLabel.setText(balanceHtml(a));
        walletBBalanceLabel.setText(balanceHtml(b));

        // Stats
        mempoolLabel.setText(String.valueOf(Noob.mempool.size()));
        difficultyLabel.setText(String.valueOf(Noob.difficulty));
        blockHeightLabel.setText(String.valueOf(Noob.blockchain.size() - 1)); // 0-indexed
        rewardLabel.setText(StringUtil.toCoins(Noob.miningReward));
    }

    private void log(String tag, String msg) {
        Color color = switch (tag) {
            case "TX"       -> new Color(39,  174, 96);
            case "MINE"     -> new Color(52,  152, 219);
            case "VALID"    -> new Color(39,  174, 96);
            case "INVALID"  -> new Color(231, 76,  60);
            case "VALIDATE" -> new Color(155, 89,  182);
            default         -> new Color(140, 160, 180);
        };


        logArea.append("[" + tag + "] " + msg + "\n");
        logArea.setCaretPosition(logArea.getDocument().getLength());
    }

    private void showError(String msg) {
        log("ERROR", msg);
        setStatus("Error: " + msg);
        JOptionPane.showMessageDialog(frame, msg, "Error", JOptionPane.ERROR_MESSAGE);
    }

    private void setStatus(String msg) {
        statusLabel.setText(msg);
    }

    private boolean isReady() {
        if (Noob.walletA == null || Noob.walletB == null) {
            showError("Wallets not initialized."); return false;
        }
        if (Noob.blockchain == null || Noob.blockchain.isEmpty()) {
            showError("Blockchain empty."); return false;
        }
        return true;
    }

    private String balanceHtml(String value) {
        return "<html><span style='font-family:Consolas;font-size:26px;" +
                "font-weight:bold;color:#E8F4FD;'>" + value + "</span></html>";
    }

    private String styledLabel(String text) {
        return "<html><span style='color:#8899aa;font-family:Consolas;" +
                "font-size:12px;'>" + text + "</span></html>";
    }


    private JPanel makeBalanceCard(String title, JLabel valueLabel, Color accent) {
        JPanel card = new JPanel(new BorderLayout(0, 6));
        card.setBackground(new Color(24, 28, 36));
        card.setBorder(BorderFactory.createCompoundBorder(
                BorderFactory.createMatteBorder(0, 3, 0, 0, accent),
                BorderFactory.createCompoundBorder(
                        BorderFactory.createLineBorder(new Color(45, 52, 65)),
                        BorderFactory.createEmptyBorder(14, 16, 14, 16)
                )
        ));

        JLabel titleLabel = new JLabel(title);
        titleLabel.setFont(new Font("Consolas", Font.PLAIN, 12));
        titleLabel.setForeground(new Color(120, 140, 160));

        valueLabel.setText(balanceHtml("—"));

        card.add(titleLabel,  BorderLayout.NORTH);
        card.add(valueLabel,  BorderLayout.CENTER);
        return card;
    }

    private JPanel makeStatCard(String title, JLabel valueLabel, Color accent) {
        JPanel card = new JPanel(new BorderLayout(0, 4));
        card.setBackground(new Color(24, 28, 36));
        card.setBorder(BorderFactory.createCompoundBorder(
                BorderFactory.createLineBorder(new Color(45, 52, 65)),
                BorderFactory.createEmptyBorder(10, 12, 10, 12)
        ));

        JLabel titleLabel = new JLabel(title, SwingConstants.CENTER);
        titleLabel.setFont(new Font("Consolas", Font.PLAIN, 11));
        titleLabel.setForeground(new Color(100, 120, 140));

        valueLabel.setFont(new Font("Consolas", Font.BOLD, 22));
        valueLabel.setForeground(accent);
        valueLabel.setHorizontalAlignment(SwingConstants.CENTER);

        card.add(titleLabel,  BorderLayout.NORTH);
        card.add(valueLabel,  BorderLayout.CENTER);
        return card;
    }

    private JButton makeButton(String text, Color bg) {
        JButton btn = new JButton(text);
        btn.setFont(new Font("Consolas", Font.BOLD, 12));
        btn.setBackground(bg);
        btn.setForeground(Color.WHITE);
        btn.setFocusPainted(false);
        btn.setBorderPainted(false);
        btn.setCursor(Cursor.getPredefinedCursor(Cursor.HAND_CURSOR));
        btn.setBorder(BorderFactory.createEmptyBorder(8, 14, 8, 14));

        btn.addMouseListener(new MouseAdapter() {
            final Color base = bg;
            public void mouseEntered(MouseEvent e) { btn.setBackground(base.brighter()); }
            public void mouseExited(MouseEvent e)  { btn.setBackground(base); }
        });
        return btn;
    }

    private void styleCombo(JComboBox<String> combo) {
        combo.setFont(new Font("Consolas", Font.PLAIN, 13));
        combo.setBackground(new Color(28, 32, 40));
        combo.setForeground(Color.WHITE);
    }

    private TitledBorder makeTitledBorder(String title) {
        TitledBorder tb = BorderFactory.createTitledBorder(
                BorderFactory.createLineBorder(new Color(50, 60, 80)), title
        );
        tb.setTitleFont(new Font("Consolas", Font.PLAIN, 11));
        tb.setTitleColor(new Color(100, 120, 150));
        return tb;
    }

    private void setLookAndFeel() {
        try { UIManager.setLookAndFeel(UIManager.getSystemLookAndFeelClassName()); }
        catch (Exception ignored) {}
    }

    public static void main(String[] args) {
        SwingUtilities.invokeLater(Dashboard::new);
    }
}