package com.bytezone.diskbrowser.gui;

import java.awt.Desktop;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;
import java.util.prefs.Preferences;

import javax.swing.AbstractButton;
import javax.swing.ButtonGroup;
import javax.swing.JCheckBoxMenuItem;
import javax.swing.JMenu;
import javax.swing.JMenuBar;
import javax.swing.JMenuItem;
import javax.swing.JRadioButtonMenuItem;
import javax.swing.KeyStroke;

import com.bytezone.common.FontAction;
import com.bytezone.diskbrowser.applefile.AssemblerProgram;
import com.bytezone.diskbrowser.applefile.BasicProgram;
import com.bytezone.diskbrowser.applefile.HiResImage;
import com.bytezone.diskbrowser.applefile.Palette;
import com.bytezone.diskbrowser.applefile.PaletteFactory;
import com.bytezone.diskbrowser.applefile.VisicalcFile;
import com.bytezone.diskbrowser.disk.DataDisk;
import com.bytezone.diskbrowser.disk.FormattedDisk;
import com.bytezone.diskbrowser.prodos.ProdosDisk;

public class MenuHandler implements DiskSelectionListener, FileSelectionListener,
    QuitListener, SectorSelectionListener
{
  static final String PREFS_LINE_WRAP = "line wrap";
  private static final String PREFS_SHOW_CATALOG = "show catalog";
  private static final String PREFS_SHOW_LAYOUT = "show layout";
  private static final String PREFS_SHOW_FREE_SECTORS = "show free sectors";
  private static final String PREFS_COLOUR_QUIRKS = "colour quirks";
  private static final String PREFS_MONOCHROME = "monochrome";

  private static final String PREFS_SPLIT_REMARKS = "splitRemarks";
  private static final String PREFS_ALIGN_ASSIGN = "alignAssign";
  private static final String PREFS_SHOW_TARGETS = "showTargets";
  private static final String PREFS_ONLY_SHOW_TARGETS = "onlyShowTargets";
  private static final String PREFS_SHOW_HEADER = "showHeader";
  private static final String PREFS_SHOW_CARET = "showCaret";

  private static final String PREFS_SHOW_ASSEMBLER_TARGETS = "showAssemblerTargets";
  private static final String PREFS_SHOW_ASSEMBLER_STRINGS = "showAssemblerStrings";
  private static final String PREFS_SHOW_ASSEMBLER_HEADER = "showAssemblerHeader";

  private static final String PREFS_PRODOS_SORT_DIRECTORIES = "prodosSortDirectories";

  //  private static final String PREFS_DEBUGGING = "debugging";
  private static final String PREFS_PALETTE = "palette";

  FormattedDisk currentDisk;
  private final SaveTempFileAction saveTempFileAction = new SaveTempFileAction ();
  final SaveSectorsAction saveSectorsAction = new SaveSectorsAction ();

  private final BasicPreferences basicPreferences = new BasicPreferences ();
  private final List<BasicPreferencesListener> basicPreferencesListeners =
      new ArrayList<> ();

  private final AssemblerPreferences assemblerPreferences = new AssemblerPreferences ();
  private final List<AssemblerPreferencesListener> assemblerPreferencesListeners =
      new ArrayList<> ();

  private final ProdosPreferences prodosPreferences = new ProdosPreferences ();
  private final List<ProdosPreferencesListener> prodosPreferencesListeners =
      new ArrayList<> ();

  JMenuBar menuBar = new JMenuBar ();
  JMenu fileMenu = new JMenu ("File");
  JMenu formatMenu = new JMenu ("Format");
  JMenu colourMenu = new JMenu ("Colours");
  JMenu applesoftMenu = new JMenu ("Applesoft");
  JMenu assemblerMenu = new JMenu ("Assembler");
  JMenu prodosMenu = new JMenu ("Prodos");
  //  JMenu helpMenu = new JMenu ("Help");

  // File menu items
  final JMenuItem rootItem = new JMenuItem ("Set root folder...");
  final JMenuItem refreshTreeItem = new JMenuItem ("Refresh current tree");
  JMenuItem executeDiskItem;
  final JMenuItem saveDiskItem = new JMenuItem ("Save converted disk as...");
  final JMenuItem saveSectorsItem = new JMenuItem ("Save sectors as...");
  final JMenuItem printItem = new JMenuItem ("Print output panel...");
  final JMenuItem closeTabItem = new JMenuItem ();
  final JMenuItem duplicateItem = new JMenuItem ();
  final FontAction fontAction = new FontAction ();

  // Format menu items
  final JMenuItem lineWrapItem = new JCheckBoxMenuItem ("Line wrap");
  final JMenuItem showLayoutItem = new JCheckBoxMenuItem ("Show layout panel");
  final JMenuItem showCatalogItem = new JCheckBoxMenuItem ("Show catalog panel");
  final JMenuItem showFreeSectorsItem = new JCheckBoxMenuItem ("Show free sectors");

  final JMenuItem sector256Item = new JRadioButtonMenuItem ("256 byte sectors");
  final JMenuItem sector512Item = new JRadioButtonMenuItem ("512 byte blocks");
  final JMenuItem interleave0Item = new JRadioButtonMenuItem (new InterleaveAction (0));
  final JMenuItem interleave1Item = new JRadioButtonMenuItem (new InterleaveAction (1));
  final JMenuItem interleave2Item = new JRadioButtonMenuItem (new InterleaveAction (2));
  final JMenuItem interleave3Item = new JRadioButtonMenuItem (new InterleaveAction (3));

  final JMenuItem colourQuirksItem = new JCheckBoxMenuItem ("Colour quirks");
  final JMenuItem monochromeItem = new JCheckBoxMenuItem ("Monochrome");
  final JMenuItem debuggingItem = new JCheckBoxMenuItem ("Debugging");
  final JMenuItem nextPaletteItem = new JMenuItem ("Next Palette");
  final JMenuItem prevPaletteItem = new JMenuItem ("Previous Palette");

  // Applesoft menu items
  final JMenuItem splitRemarkItem = new JCheckBoxMenuItem ("Split remarks");
  final JMenuItem alignAssignItem = new JCheckBoxMenuItem ("Align assign");
  final JMenuItem showBasicTargetsItem = new JCheckBoxMenuItem ("Show targets");
  final JMenuItem onlyShowTargetLinesItem =
      new JCheckBoxMenuItem ("Only show target lines");
  final JMenuItem showHeaderItem = new JCheckBoxMenuItem ("Show header");
  final JMenuItem showCaretItem = new JCheckBoxMenuItem ("Show caret");

  // Assembler menu items
  final JMenuItem showAssemblerTargetsItem = new JCheckBoxMenuItem ("Show targets");
  final JMenuItem showAssemblerStringsItem = new JCheckBoxMenuItem ("Show strings");
  final JMenuItem showAssemblerHeaderItem = new JCheckBoxMenuItem ("Show header");

  // Prodos menu items
  final JMenuItem prodosSortDirectoriesItem = new JCheckBoxMenuItem ("Sort directories");

  ButtonGroup paletteGroup = new ButtonGroup ();

  public MenuHandler ()
  {
    menuBar.add (fileMenu);
    menuBar.add (formatMenu);
    menuBar.add (colourMenu);
    menuBar.add (applesoftMenu);
    menuBar.add (assemblerMenu);
    menuBar.add (prodosMenu);
    //    menuBar.add (helpMenu);

    fileMenu.add (rootItem);
    fileMenu.addSeparator ();
    fileMenu.add (refreshTreeItem);
    fileMenu.add (saveDiskItem);
    fileMenu.add (saveSectorsItem);

    addLauncherMenu ();

    fileMenu.add (printItem);
    fileMenu.addSeparator ();
    fileMenu.add (closeTabItem);

    JMenuItem fontItem = new JMenuItem (fontAction);
    fileMenu.add (fontItem);
    fontAction.setSampleText ("120  FOR Z = 14 TO 24:\n  VTAB 5:\n  HTAB Z:\n"
        + "  PRINT AB$:\n  FOR TI = 1 TO 50:\n  NEXT :\n  POKE 0,Z + 40:\n"
        + "  POKE 1,9:\n  CALL MU:\n  VTAB 5:\n  HTAB Z:\n"
        + "  PRINT SPC(12):\nNEXT :\nVTAB 5:\nHTAB 24:\nPRINT AB$\n");

    fileMenu.add (duplicateItem);
    fileMenu.add (debuggingItem);

    formatMenu.add (lineWrapItem);
    formatMenu.add (showCatalogItem);
    formatMenu.add (showLayoutItem);
    formatMenu.add (showFreeSectorsItem);

    formatMenu.addSeparator ();

    formatMenu.add (interleave0Item);
    formatMenu.add (interleave1Item);
    formatMenu.add (interleave2Item);
    formatMenu.add (interleave3Item);

    formatMenu.addSeparator ();

    formatMenu.add (sector256Item);
    formatMenu.add (sector512Item);

    // set placeholders for the palettes
    List<Palette> palettes = HiResImage.getPalettes ();
    for (int i = 0; i < palettes.size (); i++)
    {
      JCheckBoxMenuItem menuItem = new JCheckBoxMenuItem ("x");
      paletteGroup.add (menuItem);
      colourMenu.add (menuItem);
    }

    colourMenu.addSeparator ();
    colourMenu.add (colourQuirksItem);
    colourMenu.add (monochromeItem);
    colourMenu.addSeparator ();
    colourMenu.add (nextPaletteItem);
    colourMenu.add (prevPaletteItem);

    applesoftMenu.add (splitRemarkItem);
    applesoftMenu.add (alignAssignItem);
    applesoftMenu.add (showBasicTargetsItem);
    applesoftMenu.add (onlyShowTargetLinesItem);
    applesoftMenu.add (showHeaderItem);
    applesoftMenu.add (showCaretItem);

    assemblerMenu.add (showAssemblerTargetsItem);
    assemblerMenu.add (showAssemblerStringsItem);
    assemblerMenu.add (showAssemblerHeaderItem);

    prodosMenu.add (prodosSortDirectoriesItem);

    ActionListener basicPreferencesAction = new ActionListener ()
    {
      @Override
      public void actionPerformed (ActionEvent e)
      {
        setBasicPreferences ();
        notifyBasicPreferencesListeners ();
      }
    };

    ActionListener assemblerPreferencesAction = new ActionListener ()
    {
      @Override
      public void actionPerformed (ActionEvent e)
      {
        setAssemblerPreferences ();
        notifyAssemblerPreferencesListeners ();
      }
    };

    ActionListener prodosPreferencesAction = new ActionListener ()
    {
      @Override
      public void actionPerformed (ActionEvent e)
      {
        setProdosPreferences ();
        notifyProdosPreferencesListeners ();
      }
    };

    splitRemarkItem.addActionListener (basicPreferencesAction);
    alignAssignItem.addActionListener (basicPreferencesAction);
    showBasicTargetsItem.addActionListener (basicPreferencesAction);
    onlyShowTargetLinesItem.addActionListener (basicPreferencesAction);
    showHeaderItem.addActionListener (basicPreferencesAction);
    showCaretItem.addActionListener (basicPreferencesAction);

    showAssemblerTargetsItem.addActionListener (assemblerPreferencesAction);
    showAssemblerStringsItem.addActionListener (assemblerPreferencesAction);
    showAssemblerHeaderItem.addActionListener (assemblerPreferencesAction);

    prodosSortDirectoriesItem.addActionListener (prodosPreferencesAction);

    //    helpMenu.add (new JMenuItem (new EnvironmentAction ()));

    sector256Item.setActionCommand ("256");
    sector256Item.setAccelerator (KeyStroke.getKeyStroke ("alt 4"));
    sector512Item.setActionCommand ("512");
    sector512Item.setAccelerator (KeyStroke.getKeyStroke ("alt 5"));

    ButtonGroup sectorGroup = new ButtonGroup ();
    ButtonGroup interleaveGroup = new ButtonGroup ();

    sectorGroup.add (sector256Item);
    sectorGroup.add (sector512Item);
    interleaveGroup.add (interleave0Item);
    interleaveGroup.add (interleave1Item);
    interleaveGroup.add (interleave2Item);
    interleaveGroup.add (interleave3Item);

    saveDiskItem.setAction (saveTempFileAction);
    saveSectorsItem.setAction (saveSectorsAction);
  }

  private void setBasicPreferences ()
  {
    basicPreferences.splitRem = splitRemarkItem.isSelected ();
    basicPreferences.alignAssign = alignAssignItem.isSelected ();
    basicPreferences.showCaret = showCaretItem.isSelected ();
    basicPreferences.showHeader = showHeaderItem.isSelected ();
    basicPreferences.showTargets = showBasicTargetsItem.isSelected ();
    basicPreferences.onlyShowTargetLineNumbers = onlyShowTargetLinesItem.isSelected ();
    BasicProgram.setBasicPreferences (basicPreferences);
  }

  void addBasicPreferencesListener (BasicPreferencesListener listener)
  {
    if (!basicPreferencesListeners.contains (listener))
    {
      basicPreferencesListeners.add (listener);
      listener.setBasicPreferences (basicPreferences);
    }
  }

  void notifyBasicPreferencesListeners ()
  {
    for (BasicPreferencesListener listener : basicPreferencesListeners)
      listener.setBasicPreferences (basicPreferences);
  }

  private void setAssemblerPreferences ()
  {
    assemblerPreferences.showTargets = showAssemblerTargetsItem.isSelected ();
    assemblerPreferences.showStrings = showAssemblerStringsItem.isSelected ();
    assemblerPreferences.showHeader = showAssemblerHeaderItem.isSelected ();
    AssemblerProgram.setAssemblerPreferences (assemblerPreferences);
  }

  void addAssemblerPreferencesListener (AssemblerPreferencesListener listener)
  {
    if (!assemblerPreferencesListeners.contains (listener))
    {
      assemblerPreferencesListeners.add (listener);
      listener.setAssemblerPreferences (assemblerPreferences);
    }
  }

  void notifyAssemblerPreferencesListeners ()
  {
    for (AssemblerPreferencesListener listener : assemblerPreferencesListeners)
      listener.setAssemblerPreferences (assemblerPreferences);
  }

  private void setProdosPreferences ()
  {
    prodosPreferences.sortDirectories = prodosSortDirectoriesItem.isSelected ();
    ProdosDisk.setProdosPreferences (prodosPreferences);
  }

  void addProdosPreferencesListener (ProdosPreferencesListener listener)
  {
    if (!prodosPreferencesListeners.contains (listener))
    {
      prodosPreferencesListeners.add (listener);
      listener.setProdosPreferences (prodosPreferences);
    }
  }

  void notifyProdosPreferencesListeners ()
  {
    for (ProdosPreferencesListener listener : prodosPreferencesListeners)
      listener.setProdosPreferences (prodosPreferences);
  }

  //  void addHelpMenuAction (Action action, String functionName)
  //  {
  //    helpMenu.add (new JMenuItem (action));
  //  }

  private void addLauncherMenu ()
  {
    if (!Desktop.isDesktopSupported ())
      return;

    boolean openSupported = false;
    for (Desktop.Action action : Desktop.Action.values ())
      if (action.toString ().equals ("OPEN"))
      {
        openSupported = true;
        break;
      }
    if (!openSupported)
      return;

    executeDiskItem = new JMenuItem (new ExecuteDiskAction (this));
    fileMenu.add (executeDiskItem);
    fileMenu.addSeparator ();
  }

  @Override
  public void quit (Preferences prefs)
  {
    prefs.putBoolean (PREFS_LINE_WRAP, lineWrapItem.isSelected ());
    prefs.putBoolean (PREFS_SHOW_LAYOUT, showLayoutItem.isSelected ());
    prefs.putBoolean (PREFS_SHOW_CATALOG, showCatalogItem.isSelected ());
    prefs.putBoolean (PREFS_SHOW_FREE_SECTORS, showFreeSectorsItem.isSelected ());
    prefs.putBoolean (PREFS_COLOUR_QUIRKS, colourQuirksItem.isSelected ());
    prefs.putBoolean (PREFS_MONOCHROME, monochromeItem.isSelected ());
    //    prefs.putBoolean (PREFS_DEBUGGING, debuggingItem.isSelected ());
    prefs.putInt (PREFS_PALETTE,
        HiResImage.getPaletteFactory ().getCurrentPaletteIndex ());
    fontAction.quit (prefs);

    prefs.putBoolean (PREFS_SPLIT_REMARKS, splitRemarkItem.isSelected ());
    prefs.putBoolean (PREFS_ALIGN_ASSIGN, alignAssignItem.isSelected ());
    prefs.putBoolean (PREFS_SHOW_CARET, showCaretItem.isSelected ());
    prefs.putBoolean (PREFS_SHOW_HEADER, showHeaderItem.isSelected ());
    prefs.putBoolean (PREFS_SHOW_TARGETS, showBasicTargetsItem.isSelected ());
    prefs.putBoolean (PREFS_ONLY_SHOW_TARGETS, onlyShowTargetLinesItem.isSelected ());

    prefs.putBoolean (PREFS_SHOW_ASSEMBLER_TARGETS,
        showAssemblerTargetsItem.isSelected ());
    prefs.putBoolean (PREFS_SHOW_ASSEMBLER_STRINGS,
        showAssemblerStringsItem.isSelected ());
    prefs.putBoolean (PREFS_SHOW_ASSEMBLER_HEADER, showAssemblerHeaderItem.isSelected ());

    prefs.putBoolean (PREFS_PRODOS_SORT_DIRECTORIES,
        prodosSortDirectoriesItem.isSelected ());
  }

  @Override
  public void restore (Preferences prefs)
  {
    lineWrapItem.setSelected (prefs.getBoolean (PREFS_LINE_WRAP, true));
    showLayoutItem.setSelected (prefs.getBoolean (PREFS_SHOW_LAYOUT, true));
    showCatalogItem.setSelected (prefs.getBoolean (PREFS_SHOW_CATALOG, true));
    showFreeSectorsItem.setSelected (prefs.getBoolean (PREFS_SHOW_FREE_SECTORS, false));
    colourQuirksItem.setSelected (prefs.getBoolean (PREFS_COLOUR_QUIRKS, false));
    monochromeItem.setSelected (prefs.getBoolean (PREFS_MONOCHROME, false));

    //    debuggingItem.setSelected (prefs.getBoolean (PREFS_DEBUGGING, false));

    splitRemarkItem.setSelected (prefs.getBoolean (PREFS_SPLIT_REMARKS, false));
    alignAssignItem.setSelected (prefs.getBoolean (PREFS_ALIGN_ASSIGN, true));
    showCaretItem.setSelected (prefs.getBoolean (PREFS_SHOW_CARET, false));
    showHeaderItem.setSelected (prefs.getBoolean (PREFS_SHOW_HEADER, true));
    showBasicTargetsItem.setSelected (prefs.getBoolean (PREFS_SHOW_TARGETS, false));
    onlyShowTargetLinesItem
        .setSelected (prefs.getBoolean (PREFS_ONLY_SHOW_TARGETS, false));

    showAssemblerTargetsItem
        .setSelected (prefs.getBoolean (PREFS_SHOW_ASSEMBLER_TARGETS, true));
    showAssemblerStringsItem
        .setSelected (prefs.getBoolean (PREFS_SHOW_ASSEMBLER_STRINGS, true));
    showAssemblerHeaderItem
        .setSelected (prefs.getBoolean (PREFS_SHOW_ASSEMBLER_HEADER, true));

    prodosSortDirectoriesItem
        .setSelected (prefs.getBoolean (PREFS_PRODOS_SORT_DIRECTORIES, true));

    setBasicPreferences ();
    setAssemblerPreferences ();
    setProdosPreferences ();

    int paletteIndex = prefs.getInt (PREFS_PALETTE, 0);
    PaletteFactory paletteFactory = HiResImage.getPaletteFactory ();
    paletteFactory.setCurrentPalette (paletteIndex);
    Palette palette = paletteFactory.getCurrentPalette ();
    Enumeration<AbstractButton> enumeration = paletteGroup.getElements ();
    while (enumeration.hasMoreElements ())
    {
      JCheckBoxMenuItem item = (JCheckBoxMenuItem) enumeration.nextElement ();
      if (item.getText ().equals (palette.getName ()))
      {
        item.setSelected (true);
        break;
      }
    }

    HiResImage.setDefaultColourQuirks (colourQuirksItem.isSelected ());
    HiResImage.setDefaultMonochrome (monochromeItem.isSelected ());
    VisicalcFile.setDefaultDebug (debuggingItem.isSelected ());

    fontAction.restore (prefs);
  }

  @Override
  public void diskSelected (DiskSelectedEvent event)
  {
    currentDisk = event.getFormattedDisk ();
    adjustMenus (currentDisk);
  }

  @Override
  public void fileSelected (FileSelectedEvent event)
  {
    // This can happen if a file is selected from a dual-dos disk
    if (event.appleFileSource.getFormattedDisk () != currentDisk)
    {
      currentDisk = event.appleFileSource.getFormattedDisk ();
      adjustMenus (currentDisk);
    }
  }

  private void adjustMenus (final FormattedDisk disk)
  {
    if (disk != null)
    {
      sector256Item.setSelected (disk.getDisk ().getBlockSize () == 256);
      sector512Item.setSelected (disk.getDisk ().getBlockSize () == 512);
      interleave0Item.setSelected (disk.getDisk ().getInterleave () == 0);
      interleave1Item.setSelected (disk.getDisk ().getInterleave () == 1);
      interleave2Item.setSelected (disk.getDisk ().getInterleave () == 2);
      interleave3Item.setSelected (disk.getDisk ().getInterleave () == 3);
    }

    boolean isDataDisk = (disk instanceof DataDisk);

    sector256Item.setEnabled (isDataDisk);
    sector512Item.setEnabled (isDataDisk);
    interleave0Item.setEnabled (isDataDisk);
    interleave1Item.setEnabled (isDataDisk);
    interleave2Item.setEnabled (isDataDisk);
    interleave3Item.setEnabled (isDataDisk);

    if (isDataDisk)
    {
      // make this an action too
      ActionListener sectorListener = new ActionListener ()
      {
        @Override
        public void actionPerformed (ActionEvent e)
        {
          int size = Integer.parseInt (e.getActionCommand ());
          disk.getDisk ().setBlockSize (size);
        }
      };

      sector256Item.addActionListener (sectorListener);
      sector512Item.addActionListener (sectorListener);

      ((InterleaveAction) interleave0Item.getAction ()).setDisk (currentDisk);
      ((InterleaveAction) interleave1Item.getAction ()).setDisk (currentDisk);
      ((InterleaveAction) interleave2Item.getAction ()).setDisk (currentDisk);
      ((InterleaveAction) interleave3Item.getAction ()).setDisk (currentDisk);
    }

    saveDiskItem.setEnabled (disk.isTempDisk ());
    saveTempFileAction.setDisk (disk);
  }

  @Override
  public void sectorSelected (SectorSelectedEvent event)
  {
    //    List<DiskAddress> sectors = event.getSectors ();
  }
}