rootpanel.java

用java写的ftp服务器程序

// $Id: RootPanel.java 306734 2005-09-24 10:38:20 +0530 (Sat, 24 Sep 2005) rana_b $
/*
 * Copyright 2004 The Apache Software Foundation
 *
 * 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 org.apache.ftpserver.gui;

import java.awt.BorderLayout;
import java.awt.Color;
import java.awt.Dimension;
import java.awt.FlowLayout;
import java.awt.Font;
import java.awt.GridBagConstraints;
import java.awt.GridBagLayout;
import java.awt.Insets;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;

import org.apache.ftpserver.FtpConfigImpl;
import org.apache.ftpserver.FtpServer;
import org.apache.ftpserver.config.PropertiesConfiguration;
import org.apache.ftpserver.config.XmlConfigurationHandler;
import org.apache.ftpserver.ftplet.Configuration;
import org.apache.ftpserver.interfaces.IFtpConfig;
import org.apache.ftpserver.util.IoUtils;

import javax.swing.ButtonGroup;
import javax.swing.JButton;
import javax.swing.JFileChooser;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JRadioButton;
import javax.swing.JScrollPane;
import javax.swing.JTextArea;
import javax.swing.JTextField;
import javax.swing.JToggleButton;

/**
 * This is the tree root panel. It is responsible to load 
 * configuration file and start/stop the server. 
 * 
 * @author <a href="mailto:[e-mail]">Rana Bhattacharyya</a>
 */
public 
class RootPanel extends PluginPanel {
    
    private static final long serialVersionUID = 6048031185986894615L;

    private JTextField m_cfgFileTxt;
    
    private JRadioButton m_cfgPropRadio;
    private JRadioButton m_cfgXmlRadio;
    
    private JRadioButton m_cfgFileRadio;
    private JRadioButton m_cfgClasspathRadio;
    
    private JButton m_browseButton;
    private JButton m_startButton;
    private JButton m_stopButton;
    private JToggleButton m_suspendButton; 
    
    private FtpServer m_server;
    
    
    /**
     * Constructor - create all UI components.
     */
    public RootPanel(PluginPanelContainer container) {
        super(container);
        initComponents();
    }
    
    
    /**
     * Initialize UI components.
     */
    public void initComponents() {
        
        setLayout(new BorderLayout());
        
        // top panel
        JPanel topPnl = new JPanel(new GridBagLayout());
        add(topPnl, BorderLayout.NORTH);
        GridBagConstraints gc = new GridBagConstraints();
        int yindex = -1; 
        gc.insets = new Insets(0, 0, 0, 10);
        
        // configuration label
        JLabel cfgLab = new JLabel("Configuration", JLabel.CENTER);
        cfgLab.setForeground(Color.darkGray);
        cfgLab.setFont(new Font(null, Font.BOLD, 14));
        gc.gridx = 0;
        gc.gridy = ++yindex;
        gc.gridwidth = 4;
        gc.fill = GridBagConstraints.HORIZONTAL;
        topPnl.add(cfgLab, gc);
        
        // file label
        JLabel cfgFileLab = new JLabel("File :: ", JLabel.RIGHT);
        cfgFileLab.setFont(new Font(null, Font.BOLD, 12));
        gc.gridx = 0;
        gc.gridy = ++yindex;
        gc.gridwidth = 1;
        gc.fill = GridBagConstraints.HORIZONTAL;
        topPnl.add(cfgFileLab, gc);
        
        // file text field
        m_cfgFileTxt = new JTextField();
        m_cfgFileTxt.setText("./res/conf/ftpd.properties");
        gc.fill = GridBagConstraints.HORIZONTAL;
        gc.gridx = 1;
        gc.gridy = yindex;
        gc.gridwidth = 2;
        topPnl.add(m_cfgFileTxt, gc);
        
        // browse button
        m_browseButton = new JButton("Browse");
        m_browseButton.setPreferredSize(new Dimension(80, 22));
        m_browseButton.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent evt) {
                File currFile = new File(m_cfgFileTxt.getText());
                JFileChooser fileChoose = new JFileChooser(currFile.getParentFile());
                fileChoose.setFileSelectionMode(JFileChooser.FILES_ONLY);
                int returnVal = fileChoose.showOpenDialog(RootPanel.this);
                if (returnVal == JFileChooser.APPROVE_OPTION) {
                    String path = fileChoose.getSelectedFile().getAbsolutePath();
                    m_cfgFileTxt.setText(path);
                }
            }
        });
        gc.fill = GridBagConstraints.HORIZONTAL;
        gc.gridx = 3;
        gc.gridy = yindex;
        gc.gridwidth = 1;
        topPnl.add(m_browseButton, gc);
        
        // source label
        JLabel cfgSrcLab = new JLabel("Source :: ", JLabel.RIGHT);
        cfgSrcLab.setFont(new Font(null, Font.BOLD, 12));
        gc.fill = GridBagConstraints.HORIZONTAL;
        gc.gridx = 0;
        gc.gridy = ++yindex;
        gc.gridwidth = 1;
        topPnl.add(cfgSrcLab, gc);
        
        // file system radio button
        m_cfgFileRadio = new JRadioButton("File System");
        m_cfgFileRadio.setFont(new Font(null, Font.BOLD, 12));
        m_cfgFileRadio.setSelected(true);
        m_cfgFileRadio.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent evt) {
                m_browseButton.setEnabled(m_cfgFileRadio.isSelected());
            }
        });
        gc.gridx = 1;
        gc.gridy = yindex;
        gc.gridwidth = 1;
        topPnl.add(m_cfgFileRadio, gc);
        
        // classpath radio button
        m_cfgClasspathRadio = new JRadioButton("Classpath");
        m_cfgClasspathRadio.setFont(new Font(null, Font.BOLD, 12));
        m_cfgClasspathRadio.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent evt) {
                m_browseButton.setEnabled(!m_cfgClasspathRadio.isSelected());
            }
        });
        gc.gridx = 2;
        gc.gridy = yindex;
        gc.gridwidth = 1;
        topPnl.add(m_cfgClasspathRadio, gc);
        
        // type label
        JLabel cfgTypLab = new JLabel("Type :: ", JLabel.RIGHT);
        cfgTypLab.setFont(new Font(null, Font.BOLD, 12));
        gc.fill = GridBagConstraints.HORIZONTAL;
        gc.gridx = 0;
        gc.gridy = ++yindex;
        gc.gridwidth = 1;
        topPnl.add(cfgTypLab, gc);
        
        // properties radio button
        m_cfgPropRadio = new JRadioButton("Properties");
        m_cfgPropRadio.setFont(new Font(null, Font.BOLD, 12));
        m_cfgPropRadio.setSelected(true);
        gc.gridx = 1;
        gc.gridy = yindex;
        gc.gridwidth = 1;
        topPnl.add(m_cfgPropRadio, gc);
        
        // xml radio button
        m_cfgXmlRadio = new JRadioButton("XML");
        m_cfgXmlRadio.setFont(new Font(null, Font.BOLD, 12));
        gc.gridx = 2;
        gc.gridy = yindex;
        gc.gridwidth = 1;
        topPnl.add(m_cfgXmlRadio, gc);
        
        // button panel
        JPanel btnPnl = new JPanel(new FlowLayout(FlowLayout.CENTER));
        gc.gridx = 0;
        gc.gridy = ++yindex;
        gc.gridwidth = 4;
        topPnl.add(btnPnl, gc);
        
        // start button
        m_startButton = new JButton("Start");
        m_startButton.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent evt) {
                startServer();
            }
        });
        btnPnl.add(m_startButton);
        
        // stop button
        m_stopButton = new JButton("Stop");
        m_stopButton.setEnabled(false);
        m_stopButton.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent evt) {
                stopServer();
            }
        });
        btnPnl.add(m_stopButton);
        
        // suspend button
        m_suspendButton = new JToggleButton("Suspend");
        m_suspendButton.setEnabled(false);
        m_suspendButton.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent evt) {
                suspendServer();
            }
        });
        btnPnl.add(m_suspendButton);
        
        // bottom license text area
        JTextArea txtArea = new JTextArea();
        txtArea.setFont(new Font("Monospaced", Font.PLAIN, 10));
        txtArea.setEditable(false);
        
        // load start.html
        InputStream is = null;
        InputStreamReader rd = null;
        try {
            is = getClass().getClassLoader().getResourceAsStream("org/apache/ftpserver/gui/LICENSE.txt");
            if (is != null) {
                rd = new InputStreamReader(is);
                txtArea.read(rd, null);
            }
        }
        catch(IOException ex) {
        }
        finally {
            IoUtils.close(rd);
            IoUtils.close(is);
        }
        
        // editor scroll pane
        JScrollPane editorScrollPane = new JScrollPane(txtArea);
        add(editorScrollPane, BorderLayout.CENTER);
        
        // type button group 
        ButtonGroup typeButtonGroup = new ButtonGroup();
        typeButtonGroup.add(m_cfgPropRadio);
        typeButtonGroup.add(m_cfgXmlRadio);
        
        // source button group
        ButtonGroup sourceButtonGroup = new ButtonGroup();
        sourceButtonGroup.add(m_cfgFileRadio);
        sourceButtonGroup.add(m_cfgClasspathRadio);
    }
    
    
    /**
     * Refresh the panel with the new ftp config.
     */
    public void refresh(IFtpConfig ftpCOnfig) {
    }
    
    
    /**
     * This panel can always be displayed.
     */
    public boolean canBeDisplayed() {
        return true;
    }
    
    
    /**
     * Start the server.
     */
    public void startServer() {
        
        // already started - no need to do it again
        if(m_server != null) {
            return;
        }
        
        InputStream in = null;
        try {
            
            // get appropriate stream
            String path = m_cfgFileTxt.getText();
            if(m_cfgFileRadio.isSelected()) {
                in = new FileInputStream(path);
            }
            else {
                // strip the first slash (/) if exists
                if( (path.length() > 0) && (path.charAt(0) == '/') ) {
                    path = path.substring(1);
                }
                in = getClass().getClassLoader().getResourceAsStream(path);
                if (in == null) {
                    GuiUtils.showErrorMessage(RootPanel.this,
                                    "Configuration file " + path
                                  + " not found in classpath"
                                  + " nor inside any of classpath JARs.");
                    return;
                }
            }
            
            // create configuration object
            Configuration config = null;
            if(m_cfgPropRadio.isSelected()) {
                config = new PropertiesConfiguration(in);
            }
            else {
                XmlConfigurationHandler xmlHandler = new XmlConfigurationHandler(in);
                config = xmlHandler.parse();
            }
            
            // create ftp configuration object
            IFtpConfig fconfig = new FtpConfigImpl(config);
            
            // start server
            m_server = new FtpServer(fconfig);
            m_server.start();
            
            // enabled/disable components
            m_cfgFileTxt.setEnabled(false);
            m_cfgPropRadio.setEnabled(false);
            m_cfgXmlRadio.setEnabled(false);
            m_cfgFileRadio.setEnabled(false);
            m_cfgClasspathRadio.setEnabled(false);
            m_browseButton.setEnabled(false);
            m_startButton.setEnabled(false);
            m_stopButton.setEnabled(true);
            m_suspendButton.setEnabled(true);
            m_suspendButton.setSelected(false);
            
            // refresh the container
            getContainer().refresh(m_server.getFtpConfig());
        }
        catch(Exception ex) {
            m_server = null;
            ex.printStackTrace();
            GuiUtils.showErrorMessage(RootPanel.this, "Cannot start FTP server.");
        }
        finally {
            IoUtils.close(in);
        }
    }
    
    /**
     * Stop the server.
     */
    public void stopServer() {
        
        // not running - no need to stop it.
        if(m_server == null) {
            return;
        }
        
        m_server.stop();
        m_server = null;
    
        // enable/disable components
        m_cfgFileTxt.setEnabled(true);
        m_cfgPropRadio.setEnabled(true);
        m_cfgXmlRadio.setEnabled(true);
        m_cfgFileRadio.setEnabled(true);
        m_cfgClasspathRadio.setEnabled(true);
        m_browseButton.setEnabled(true);
        m_startButton.setEnabled(true);
        m_stopButton.setEnabled(false);
        m_suspendButton.setEnabled(false);
        m_suspendButton.setSelected(false);
        
        // refresh the container
        getContainer().refresh(null);
    }
    
    
    /**
     * Suspend server.
     */
    private void suspendServer() {
        
        // not running - no need to do anything.
        if(m_server == null) {
            return;
        }
        
        if(m_suspendButton.isSelected()) {
            m_server.suspend();
        }
        else {
            m_server.resume();
        }
    }
    
    
    /**
     * Get the panel name.
     */
    public String toString() {
        return "FTP";
    }
}