downloadfriendremarkshell.java

类似于MSN

/*
* LumaQQ - Java QQ Client
*
* Copyright (C) 2004 luma <[e-mail]>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
package edu.tsinghua.lumaqq.shells;

import org.eclipse.swt.SWT;
import org.eclipse.swt.events.SelectionAdapter;
import org.eclipse.swt.events.SelectionEvent;
import org.eclipse.swt.events.ShellAdapter;
import org.eclipse.swt.events.ShellEvent;
import org.eclipse.swt.graphics.Rectangle;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.layout.GridLayout;
import org.eclipse.swt.widgets.Button;
import org.eclipse.swt.widgets.Display;
import org.eclipse.swt.widgets.Label;
import org.eclipse.swt.widgets.ProgressBar;
import org.eclipse.swt.widgets.Shell;

import edu.tsinghua.lumaqq.IconHolder;
import edu.tsinghua.lumaqq.LumaQQ;
import edu.tsinghua.lumaqq.models.FriendModel;
import edu.tsinghua.lumaqq.models.GroupModel;
import edu.tsinghua.lumaqq.qq.QQ;
import edu.tsinghua.lumaqq.qq.beans.FriendRemark;
import edu.tsinghua.lumaqq.qq.events.QQEvent;
import edu.tsinghua.lumaqq.qq.events.QQListener;
import edu.tsinghua.lumaqq.qq.packets.in.FriendRemarkOpReplyPacket;
import edu.tsinghua.lumaqq.qq.packets.out.FriendRemarkOpPacket;
import edu.tsinghua.lumaqq.utils.RemarkUtils;
import edu.tsinghua.lumaqq.xml.remarks.Remark;
import edu.tsinghua.lumaqq.xml.remarks.RemarkImpl;

/**
 * <pre>
 * 批量下载好友备注信息进度窗口
 * </pre>
 * 
 * @author 马若劼
 */
public class DownloadFriendRemarkShell extends ShellAdapter implements QQListener {
	private IconHolder icons = IconHolder.getInstance();
	private Label lblHint, lblProgress;
	private ProgressBar bar;
	private Button btn;
	private Shell shell;
	private MainShell main;
	private Display display;
	// 当前组的组索引和组内的好友数
	private int tabIndex, itemCount;
	// 当前正在下载的好友索引
	private int curItem;
	// 已经下载了的好友总数
	private int itemDownloaded;
	// 好友总数
	private int totalItem;
	// 当前的好友model
	private FriendModel f;
	// 下载是否已经完成
	private boolean finished;
	// 备注工具类
	private RemarkUtils remarks;
	
	/**
	 * @param parent
	 * @param style
	 */
	public DownloadFriendRemarkShell(MainShell main) {		
		this.main = main;
		this.display = main.display;
	    // create shell
		shell = new Shell(this.display, SWT.BORDER | SWT.MIN | SWT.CLOSE | SWT.MODELESS);
		shell.setText(LumaQQ.getResourceString("op.download.remark.title"));
		shell.setSize(450, 140);
		// set title and image
		shell.setImage(icons.getResource(IconHolder.icoQQ));
		// add listener
		shell.addShellListener(this);
		
		// 初始化变量
		remarks = RemarkUtils.getInstance();
		finished = false;
		tabIndex = 0;
		itemCount = main.model.getItemCount(tabIndex);
		curItem = -1;
		itemDownloaded = 0;
		totalItem = 0;
		int tabCount = main.model.getTabCount();
		for(int i = 0; i < tabCount; i++)
		    totalItem += main.model.getItemCount(i);
		
		// init child controls
		initLayout();
	}
	
	/**
	 * 在下载完成后刷新界面
	 */
	private void finished() {
	    lblHint.setText(LumaQQ.getResourceString("download.remark.label.success"));
	    btn.setText(LumaQQ.getResourceString("download.remark.button.close"));
	    main.client.removeQQListener(this);
	    main.working = false;
	    finished = true;
	    remarks.save();
	}
	
	/**
	 * 开始发送得到下一个好友备注的请求
	 * @return true表示下载已完成，false表示没有完成
	 */
	private boolean getNext() {
	    if(totalItem == itemDownloaded)
	        return true;
	    
	    while(++curItem < itemCount) {
	        Object obj = main.model.getItem(tabIndex, curItem);
	        if(obj instanceof FriendModel) {
	            f = (FriendModel)obj;
	            break;
	        }
	    }
	    if(curItem < itemCount) {
	        main.client.downloadFriendRemark(f.getQQ());
	        return false;
	    } else {
	        curItem = 0;
	        for(tabIndex++; ; tabIndex++) {
		        GroupModel g = (GroupModel)main.model.getTab(tabIndex);
		        if(g == null)
		            return true;
		        else if(g.isCluster())
		            continue;
		        else {
		            itemCount = main.model.getItemCount(tabIndex);
		            if(itemCount == 0)
		                continue;
		            else {
		                Object obj = main.model.getItem(tabIndex, 0);
		                if(obj instanceof FriendModel) {
		                    f = (FriendModel)obj;
		                    main.client.downloadFriendRemark(f.getQQ());
		                    return false;
		                } else
		                    continue;
		            }
		        }	            
	        }
	    }
	}
	
	/**
	 * 打开对话框
	 */
	public void open() {
		// layout
		shell.layout();
		// set dialog to center of screen
		Rectangle dialogRect = shell.getBounds();
		Rectangle displayRect = display.getBounds();
		shell.setLocation((displayRect.width-dialogRect.width)/2, (displayRect.height-dialogRect.height)/2);
		// open
		shell.open();
		// 把自己加为QQListener
		main.client.addQQListener(this);
		// 开始得到好友备注
		if(getNext())
		    finished();
	}

	/**
	 * @param dialog
	 */
	private void initLayout() {
	    GridLayout layout = new GridLayout();
	    layout.marginHeight = layout.marginWidth = 15;
	    layout.verticalSpacing = 15;
	    layout.numColumns = 2;
		shell.setLayout(layout);
		// 提示标签
		lblHint = new Label(shell, SWT.NONE);
		lblHint.setText(LumaQQ.getResourceString("download.remark.label.hint"));
		lblHint.setLayoutData(new GridData());
		// 进度标签
		lblProgress = new Label(shell, SWT.NONE);
		lblProgress.setText("0/" + totalItem);
		lblProgress.setLayoutData(new GridData(GridData.FILL_BOTH | GridData.HORIZONTAL_ALIGN_BEGINNING));
		// 进度条
		bar = new ProgressBar(shell, SWT.HORIZONTAL | SWT.SMOOTH);
		GridData gd = new GridData(GridData.FILL_BOTH);
		gd.horizontalSpan = 2;
		bar.setLayoutData(gd);
		bar.setMaximum(totalItem);
		bar.setSelection(itemDownloaded);		
		// 取消按钮或确定按钮
		btn = new Button(shell, SWT.PUSH);
		btn.setText(LumaQQ.getResourceString("download.remark.button.stop"));
		gd = new GridData(GridData.HORIZONTAL_ALIGN_END);
		gd.horizontalSpan = 2;
		gd.widthHint = 100;
		btn.setLayoutData(gd);
		btn.addSelectionListener(new SelectionAdapter() {
            public void widgetSelected(SelectionEvent e) {
                close();
            }
		});
	}
	
	/**
	 * 设置窗口标题
	 * @param title
	 */
	public void setTitle(String title) {
	    shell.setText(title);
	}
	
	/**
	 * 设置提示消息
	 * @param hint
	 */
	public void setHint(String hint) {
	    lblHint.setText(hint);
	}
	
	/**
	 * 关闭窗口
	 */
	public void close() {
		shell.close();
	}
	
	/* (non-Javadoc)
	 * @see org.eclipse.swt.events.ShellListener#shellClosed(org.eclipse.swt.events.ShellEvent)
	 */
	public void shellClosed(ShellEvent e) {
	    if(!finished)
	        finished();
	}

	/**
	 * @return true 如果对话框已经被释放
	 */
	public boolean isDisposed() {
		return (shell == null || shell.isDisposed());
	}

    /* (non-Javadoc)
     * @see edu.tsinghua.lumaqq.qq.events.QQListener#qqEvent(edu.tsinghua.lumaqq.qq.events.QQEvent)
     */
    public void qqEvent(QQEvent e) {
        switch(e.type) {
            case QQEvent.QQ_DOWNLOAD_FRIEND_REMARK_SUCCESS:
                processDownloadFriendRemarkSuccess(e);
            	break;
            case QQEvent.QQ_OPERATION_TIMEOUT:
                if(e.operation == QQ.QQ_CMD_FRIEND_REMARK_OP)
                    processDownloadFriendRemarkTimeout(e);
                break;
        }
    }

    /**
     * 处理下载好友备注超时事件
     * @param e
     */
    private void processDownloadFriendRemarkTimeout(QQEvent e) {
        FriendRemarkOpPacket packet = (FriendRemarkOpPacket)e.getSource();
        if(packet.getType() == QQ.QQ_DOWNLOAD_FRIEND_REMARK) {
            main.display.syncExec(new Runnable() {
                public void run() {
                    if(getNext())
                        finished();
                }
            });
        }
    }

    /**
     * 处理下载好友备注成功事件
     * @param e
     */
    private void processDownloadFriendRemarkSuccess(QQEvent e) {
        FriendRemarkOpReplyPacket packet = (FriendRemarkOpReplyPacket)e.getSource();
        if(packet.hasRemark) {
	        final Remark remark = createRemarkElement(packet.remark);
	        remarks.addRemark(remark);
			main.display.syncExec(new Runnable() {
				public void run() {
				    bar.setSelection(++itemDownloaded);
				    lblProgress.setText("" + itemDownloaded + '/' + totalItem);
					f.addProperty("realName", remark.getName());				
					if(getNext())
					    finished();
				}
			});            
        } else {
            main.display.syncExec(new Runnable() {
                public void run() {
				    bar.setSelection(++itemDownloaded);
				    lblProgress.setText("" + itemDownloaded + '/' + totalItem);
					if(getNext())
					    finished();
                }
            });
        }
    }
    
	/**
	 * 从FriendRemark中创建Remark对象
	 * @param fr
	 * @return
	 */
	private Remark createRemarkElement(FriendRemark fr) {
		Remark remark = new RemarkImpl();
		remark.setName(fr.name);
		remark.setZipcode(fr.zipcode);
		remark.setTelephone(fr.telephone);
		remark.setMobile(fr.mobile);
		remark.setEmail(fr.email);
		remark.setAddress(fr.address);
		remark.setNote(fr.note);
		remark.setQqNum(((Integer)f.getProperty("qq")).toString());
		return remark;
	}
}