TabManager.java

/*
 * Copyright (c) 2014 Magnet Systems, Inc.
 * All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you
 * may not use this file except in compliance with the License. You
 * may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
 * implied. See the License for the specific language governing
 * permissions and limitations under the License.
 */
package com.magnet.plugin.r2m.ui.tab;

import com.beust.jcommander.internal.Lists;
import com.intellij.openapi.project.Project;
import com.magnet.plugin.common.Logger;
import com.magnet.plugin.r2m.listeners.TabRemoveListener;
import com.magnet.plugin.r2m.messages.R2MMessages;
import com.magnet.plugin.r2m.ui.AddControllerForm;

import javax.swing.*;
import javax.swing.event.ChangeEvent;
import javax.swing.event.ChangeListener;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

/**
 * Manage the method tabs in form
 */
public class TabManager {
    private final Project project;
    private final AddControllerForm form;

    private final JTabbedPane tabPanel;

    private final List<MethodTabPanel> tabs;

    public TabManager(Project project, AddControllerForm form, JTabbedPane tabPanel) {
        this.project = project;
        this.form = form;
        this.tabPanel = tabPanel;
        this.tabs = new ArrayList<MethodTabPanel>();

        init();

    }

    /**
     * Initialize the tabs
     */
    private void init() {
        MethodTabPanel customTab = new MethodTabPanel(project, form, tabPanel);
        customTab.setTabRemoveListener(tabRemoveListener);
        customTab.setIndex(0);
        tabs.add(customTab);
        tabPanel.addTab(R2MMessages.getMessage("METHOD_N", 1), customTab);
        tabPanel.addTab(R2MMessages.getMessage("PLUS_TAB"), new JLabel(""));
        tabPanel.addChangeListener(tabListener);
    }

    public MethodTabPanel addNewTab(int index) {
        MethodTabPanel customTab = new MethodTabPanel(project, form, tabPanel);
        customTab.setTabRemoveListener(tabRemoveListener);
        customTab.setIndex(index);

        tabs.add(customTab);

        // now add new tab in panel
        // clean up first
        // remove change listener
        tabPanel.removeChangeListener(tabListener);

        // remove "+" tab
        tabPanel.remove(index);
        // add "method N" tab
        tabPanel.addTab(R2MMessages.getMessage("METHOD_N", index + 1), customTab);
        // add "+" tab
        tabPanel.addTab(R2MMessages.getMessage("PLUS_TAB"), new JLabel(""));
        // set selected index
        tabPanel.setSelectedIndex(index);

        // add listener back
        tabPanel.addChangeListener(tabListener);
        updateRemoveButtons();
        return customTab;
    }

    public void removeAllTabs() {
        tabPanel.removeChangeListener(tabListener);
        for (MethodTabPanel tab : getTabs() /* use a copy */) {
            removeTab(tab);
        }
        tabPanel.addChangeListener(tabListener);
    }

    public void removeTab(MethodTabPanel methodTabPanel) {
        tabs.remove(methodTabPanel);
        tabPanel.setSelectedIndex(0);
        tabPanel.remove(methodTabPanel);
        tabPanel.invalidate();
    }

    public void selectTab(int index) {
      if(index >= 0 && index < tabs.size()) {
        tabPanel.setSelectedIndex(index);
      }
    }

    private ChangeListener tabListener = new ChangeListener() {
        @Override
        public void stateChanged(ChangeEvent e) {
            int lastSelected = tabPanel.getSelectedIndex();
            Logger.info(getClass(), "" + lastSelected);
            if (lastSelected == tabPanel.getTabCount() - 1) {  // the last index i.e. "+" button
                addNewTab(lastSelected);
                updateRemoveButtons();
                tabPanel.revalidate();

            }
        }
    };

    private TabRemoveListener tabRemoveListener = new TabRemoveListener() {
        @Override
        public void removeCurrentPanel(MethodTabPanel methodTabPanel) {
            if (tabPanel.getTabCount() > 1) {
                removeTab(methodTabPanel);
                // do not delete the method example file.
                // Users will be asked whether they want to keep it on or not at generation time.
                updateRemoveButtons();
            }
        }
    };


    public void updateRemoveButtons() {
        for (int i = 0; i < tabs.size(); i++) {
            tabs.get(i).setIndex(i);
        }
        for (MethodTabPanel panel : tabs) {
            panel.enableRemoveButton(tabs.size() > 0);
        }
    }

    /**
     * @return an immutable copy of the tabs list
     */
    public List<MethodTabPanel> getTabs() {
        return Collections.unmodifiableList(Lists.newArrayList(tabs));
    }

    public JTabbedPane getTabPanel() {
        return tabPanel;
    }



}