invitecreateaction.java

用jxse开发的一个p2p通讯软件 有聊天 文件共享 视频3大功能

/*
*  Copyright (c) 2001 Sun Microsystems, Inc.  All rights
*  reserved.
*
*  Redistribution and use in source and binary forms, with or without
*  modification, are permitted provided that the following conditions
*  are met:
*
*  1. Redistributions of source code must retain the above copyright
*  notice, this list of conditions and the following disclaimer.
*
*  2. Redistributions in binary form must reproduce the above copyright
*  notice, this list of conditions and the following disclaimer in
*  the documentation and/or other materials provided with the
*  distribution.
*
*  3. The end-user documentation included with the redistribution,
*  if any, must include the following acknowledgment:
*  "This product includes software developed by the
*  Sun Microsystems, Inc. for Project JXTA."
*  Alternately, this acknowledgment may appear in the software itself,
*  if and wherever such third-party acknowledgments normally appear.
*
*  4. The names "Sun", "Sun Microsystems, Inc.", "JXTA" and "Project JXTA"
*  must not be used to endorse or promote products derived from this
*  software without prior written permission. For written
*  permission, please contact Project JXTA at http://www.jxta.org.
*
*  5. Products derived from this software may not be called "JXTA",
*  nor may "JXTA" appear in their name, without prior written
*  permission of Sun.
*
*  THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
*  WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
*  OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
*  DISCLAIMED.  IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR
*  ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
*  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
*  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
*  USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
*  ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
*  OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
*  OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
*  SUCH DAMAGE.
*  ====================================================================
*
*  This software consists of voluntary contributions made by many
*  individuals on behalf of Project JXTA.  For more
*  information on Project JXTA, please see
*  <http://www.jxta.org/>.
*
*  This license is based on the BSD license adopted by the Apache Foundation.
*
*  $Id: InviteCreateAction.java,v 1.3 2006/05/17 21:23:58 nano Exp $
*/

package net.jxta.myjxta.misc.unused;

import net.jxta.discovery.DiscoveryService;
import net.jxta.document.Advertisement;
import net.jxta.document.StructuredTextDocument;
import net.jxta.myjxta.View;
import net.jxta.peergroup.PeerGroup;
import net.jxta.protocol.PeerGroupAdvertisement;

import javax.swing.*;
import java.awt.event.ActionEvent;
import java.io.*;
import java.util.Enumeration;
import java.util.zip.ZipOutputStream;


/**
 *
 * @version $Id: InviteCreateAction.java,v 1.3 2006/05/17 21:23:58 nano Exp $
 *
 * @author james todd [gonzo at jxta dot org]
 */

public class InviteCreateAction
    extends AbstractInviteAction {

    /** The name of the file in which data are saved */
    private File saveFile = null;


    /**
     * Create a new Invitation
     * 
     * @param name a descriptive name of the action
     * @param frame the frame used to show the file chooser
     * @param view the MyJXTAView instance used to determine the 
     *              selected peer group
     */
    public InviteCreateAction(String name, 
			      JFrame frame,
			      View view){
        super(name, frame,view);
    }

    public void actionPerformed(ActionEvent ae) {
        File f = null;
        JFileChooser fileChooser = null;

	// Create a suggested file name - if this fails, no peer group is selected 
        try {
            f = getFile();
        } catch (IOException ioe) { ioe.printStackTrace();}

        boolean isValid = false;

	// Show the file choose either with the 
        if (f != null) {
            fileChooser = new JFileChooser(getRootDir());

            fileChooser.
            setFileSelectionMode(JFileChooser.FILES_AND_DIRECTORIES);

	    if( saveFile == null){
	       saveFile = f;
	    }
	    fileChooser.setCurrentDirectory(saveFile);
	    fileChooser.setSelectedFile(saveFile);

            if (fileChooser.showSaveDialog(getFrame()) ==
                    JFileChooser.APPROVE_OPTION) {
                ZipOutputStream zos = null;

                try {
                    File sf = fileChooser.getSelectedFile();
		    saveFile = new File( sf.getAbsolutePath() );

                    zos = new ZipOutputStream(new FileOutputStream(sf));
                } catch (FileNotFoundException fne) {}

                if (zos != null) {
                    putZipEntry(zos, DiscoveryService.PEER);
                    putZipEntry(zos, DiscoveryService.GROUP);
                    putZipEntry(zos, DiscoveryService.ADV);

                    try {
                        zos.close();
                    } catch (IOException ioe) {}

                    isValid = true;
                }
            }
        }

        if (isValid) {
            JOptionPane.showMessageDialog(getFrame(),
                                          "Create Invite Succeeded.",
                                          "Success",
                                          JOptionPane.INFORMATION_MESSAGE);
        } else {
            JOptionPane.showMessageDialog(getFrame(),
                                          "An Error Occured During Create Invite.",
                                          "Failure",
                                          JOptionPane.ERROR_MESSAGE);
       }
    }


    /** 
     * Return the suggested name for the invite we want to create
     *
     * @return the suggested file name for the invite
     *  
     * @throws IOException if the file canot be found
     */
    private File getFile()
    throws IOException {
        Advertisement adv = getAdvertisement();
	PeerGroup pg = getPeerGroup();
	File f = null;

	if( adv != null && pg != null){
          f =  new File(((PeerGroupAdvertisement)adv).getName() + "-" +
                        pg.getPeerName() + ".jxta");
	}
	return f;
    }

    /**
     * Writes the entry into the the zip file
     *
     * @param zos the stream to which to append the zip entry
     * @param type the type of advertisement to zip and write
     */
    private void putZipEntry(ZipOutputStream zos, int type) {
        String s = getDocumentAsString(type);

	if( s != null){
            try {
               zos.putNextEntry(getZipEntry(type));
               zos.write(s.getBytes(), 0, s.length());
               zos.closeEntry();
           } catch (IOException ioe) {}
	}
    }

    /**
     * Returns the Advertisment of the desired type converted to a string.
     * 
     * @param type the desired type of the advertisement
     * @return the advertisment in string form or null if the advertisment was not 
     *         found.
     */
    private String getDocumentAsString(int type) {
        String s = null;

        try {
            Advertisement adv = null;

            switch (type) {
            case DiscoveryService.GROUP:
                adv = getAdvertisement();

                break;

            case DiscoveryService.PEER:
                adv = getAdvertisement(DiscoveryService.PEER);

                break;

            case DiscoveryService.ADV:
                adv = getAdvertisement(DiscoveryService.ADV, TALK_PREFIX);

                break;
            }

	    if( adv != null){
               StringWriter sw = new StringWriter();
 
               ((StructuredTextDocument)adv.getDocument(getMIME())).
                sendToWriter(sw);

                s = sw.toString();

                sw.close();
	    }
        } catch (IOException ioe) {}

        return s;
    }

    /**
     * Returns the advertisment  for the 
     * currently selected PeerGroup is returned.
     * 
     * @return the desired Advertisement or null if the advertisment was not found
     */
    private Advertisement getAdvertisement()
    throws IOException {
        return getAdvertisement( -1, "");
    }


    /**
     * Returns the advertisment with type from the
     * local cache. The name value is the  peer name of the peer in 
     * the current peer group. If type is -1, the advertisement for the 
     * currently selected PeerGroup is returned.
     * 
     * @param type the type of advertisement desired. If -1 the current PeerGroup
     *             advertisement is returned
     * @return the desired Advertisement or null if the advertisment was not found
     */
    private Advertisement getAdvertisement(int type)
    throws IOException {
        return getAdvertisement(type, "");
    }

    /**
     * Returns the advertisment with the indicated name and type from the
     * local cache. The peer name of the peer in the current peer group is appended
     * to the name value. If type is -1, the advertisement for the currently selected
     * PeerGroup is returned.
     * 
     * @param type the type of advertisement desired. If -1 the current PeerGroup
     *             advertisement is returned
     * @param name the value of the desired advertisment.
     * @return the desired Advertisement or null if the advertisment was not found
     */
    private Advertisement getAdvertisement(int type, String name)
    throws IOException {
        PeerGroup pg = getPeerGroup();
	Advertisement adv = null;

	if( pg == null) throw new IOException("PeerGroup not found");

	if( type > -1){
	    Enumeration en = pg.getDiscoveryService().
                getLocalAdvertisements(type, "Name", name + pg.getPeerName());
	    if( en.hasMoreElements() ) {
		adv = (Advertisement) en.nextElement();
	    }
	}
	else{
	    adv = pg.getPeerGroupAdvertisement();
	}

	return adv;
    }
}