package XXRoom;

import java.awt.*;
import java.awt.event.*;
import javax.swing.*;
import java.util.List;
import java.util.ArrayList;
import javax.swing.table.AbstractTableModel;
import javax.swing.event.*;
import javax.swing.text.*;
import java.net.*;
import java.io.*;
import java.util.*;

public class Client extends JFrame implements Runnable
{	
	JLabel label1;
	JLabel label2;
	JTextField ip;
	JTextField port;
	JButton    connect;
	JButton    quit;

	JTextPane  tp;
	JTextPane  ptp;
	JTextField tf;
	JButton    send;
	JScrollPane jsp;
	JScrollPane pjsp;
	AbstractDocument doc;
	AbstractDocument pdoc;
	SimpleAttributeSet sacBlue;
	SimpleAttributeSet sacRed;
	SimpleAttributeSet sacBlack;

	JButton   file;//傳送文件
	JButton   ok;  //接收文件
	JButton   cancel;//拒絕或者取消文件傳送
	
	JLabel    fileCondition;
	JCheckBox isPersonal;	

	JTable    friends;

	JPanel    link;
	JPanel    chat;
	JPanel    say;

	JPopupMenu     menu;
	JMenuItem clear;

	Socket    skt;
	DatagramSocket dSkt;
	BufferedReader read;
	PrintStream   write;
	//服務(wù)器的UDP端口
	int serverUDPPort;
	String nickName;

	Thread cThread;

	NickNameDlg nickNameDlg;	

	boolean isFileTrans;
	//傳送文件線程
	FileThread ft;	

	public Client()
	{
		super( "XX聊天室" );
		
		nickName = "XX聊天室";

		initComponent();
	
		addListeners();

		reLocate();		

		cThread = null;
		nickNameDlg = new NickNameDlg( this );

		isFileTrans = false;
	}	

	private void initComponent()
	{
		label1 = new JLabel( "服務(wù)器IP" );
		label2 = new JLabel( "端口" );
		ip = new JTextField( 16 );
		port = new JTextField( 4 );
		connect = new JButton( "連接" );
		quit   = new JButton( "退出聊天室" );
		quit.setEnabled( false );

		tp = new JTextPane( );		
		tp.setEditable( false );
		ptp = new JTextPane();
		ptp.setEditable( false );

		tf = new JTextField( 40 );
		send = new JButton( "發(fā)送" );
		tf.setEditable( false );
		send.setEnabled( false );

		file = new JButton();
		ok   = new JButton();
		cancel = new JButton();
		file.setToolTipText( "文件傳送" );
		ok.setToolTipText( "接收" );
		cancel.setToolTipText( "取消" );
		isPersonal = new JCheckBox( "悄悄話" );		
		isPersonal.setBackground( Color.white );
		
		ImageIcon img = createImageIcon( "XXRoom/images/file.png" );
		if( img != null )
			file.setIcon( img );
		else
			file.setText( "傳送文件" );
		img = createImageIcon( "XXRoom/images/ok.png" );
		if( img != null )
			ok.setIcon( img );
		else
			ok.setText( "接收" );
		img = createImageIcon( "XXRoom/images/cancel.png" );
		if( img != null )
			cancel.setIcon( img );
		else
			cancel.setText( "取消" );		
		
		ok.setBackground( Color.white );
		cancel.setBackground( Color.white );
		file.setBackground( Color.white );
		file.setBorder( null );
		ok.setBorder( null );
		cancel.setBorder( null );
		file.setEnabled( false );
		ok.setEnabled( false );
		cancel.setEnabled( false );
		fileCondition = new JLabel( "等待傳送文件" );
		fileCondition.setBackground( Color.white );		
	
		
		
		friends = new JTable( new FriendsTableModel() );		
		friends.setPreferredScrollableViewportSize( new Dimension( 100, 400 ) );
		//令table單選
		friends.setSelectionMode( ListSelectionModel.SINGLE_SELECTION );
		link = new JPanel( new FlowLayout() );
		chat = new JPanel( new BorderLayout() );
		say  = new JPanel( new FlowLayout() );

		link.add( label1 );
		link.add( ip );
		link.add( label2 );
		link.add( port );
		link.add( connect );
		link.add( quit );
		
		jsp = new JScrollPane( tp );
		jsp.setPreferredSize( new Dimension( 100, 300 ) );
		pjsp = new JScrollPane ( ptp );
		pjsp.setPreferredSize( new Dimension( 100, 100 ) );
		JSplitPane sp = new JSplitPane( JSplitPane.VERTICAL_SPLIT, jsp, pjsp );
		sp.setOneTouchExpandable( true );
		
		chat.add( sp, BorderLayout.NORTH);	
		JPanel jp2 = new JPanel();
		jp2.setPreferredSize( new Dimension( 100, 25 ) );
		jp2.setLayout( null );//取消默認(rèn)布局管理器
		jp2.setBackground( Color.white );		
		jp2.add( file );
		jp2.add( ok );
		jp2.add( cancel );
		jp2.add( fileCondition );
		jp2.add( isPersonal );
		file.setBounds( 0, 0, 25, 25 );
		ok.setBounds( 30, 0, 25, 25 );
		cancel.setBounds( 60, 0, 25, 25 );
		fileCondition.setBounds( 100, 0, 300, 25 );
		isPersonal.setBounds( 400, 0, 75, 25 );
		chat.add( jp2, BorderLayout.CENTER );
		JPanel jp1 = new JPanel();
		jp1.add( tf );
		jp1.add( send );
		chat.add( jp1, BorderLayout.SOUTH );
		

		getContentPane().add( link, BorderLayout.NORTH );
		getContentPane().add( chat, BorderLayout.CENTER );
		getContentPane().add( new JScrollPane( friends ), BorderLayout.EAST );
		
		menu = new JPopupMenu();
		clear = new JMenuItem( "Clear" );
		menu.add( clear );

		StyledDocument sd = tp.getStyledDocument();
		if( sd instanceof AbstractDocument ) {
			doc = (AbstractDocument) sd;
		}
		else {
			System.err.println( "can't cast to AbstractDocument" );
			System.exit( 0 );
		}
		sd = ptp.getStyledDocument();
		if( sd instanceof AbstractDocument ) {
			pdoc = (AbstractDocument) sd;
		}
		else {
			System.err.println( "can't cast to AbstractDocument" );
			System.exit( 0 );
		}
		sacBlue = new SimpleAttributeSet();
		StyleConstants.setForeground( sacBlue, Color.blue );
		sacRed = new SimpleAttributeSet();
		StyleConstants.setForeground( sacRed, Color.red );
		sacBlack = new SimpleAttributeSet();
		StyleConstants.setForeground( sacBlack, Color.black );
	}
	
	public ImageIcon createImageIcon( String s ) {			
			
			ImageIcon img = new ImageIcon( s );
			
			return img;
	}

	public void addListeners()
	{
		//連接服務(wù)器
		connect.addActionListener( new ActionListener() {
			public void actionPerformed( ActionEvent e ) {
				try
				{
					String address = ip.getText();
					int    p       = Integer.parseInt( port.getText() );
					try					
					{
						skt = new Socket( address, p );
						read = new BufferedReader( new InputStreamReader( skt.getInputStream() ) );						
						write = new PrintStream( skt.getOutputStream() );	
						//開(kāi)始等待聯(lián)機(jī)線程
						cThread = new Thread( Client.this );						
						cThread.start();
					}
					catch( UnknownHostException err )
					{
						JOptionPane.showMessageDialog( Client.this, "不正確的地址或端口號(hào)，請(qǐng)重新輸入" );
						return;
					}
					catch( IOException err )
					{
						JOptionPane.showMessageDialog( Client.this, "不正確的地址或端口號(hào)，請(qǐng)重新輸入" );
						return;
					}
					//初始化UDP端口
					int i = 1000;
					while( true )
					{
						try
						{
							dSkt = new DatagramSocket( i );
							break;
						}
						catch ( SocketException er )
						{
							++ i;
						}
					}

				//	JOptionPane.showMessageDialog( Client.this, "聯(lián)機(jī)成功" );
					//取得用戶昵稱
					getNickName();					
					
				}
				catch ( NumberFormatException er )
				{
					JOptionPane.showMessageDialog( Client.this, "數(shù)字格式不正確， 請(qǐng)重新輸入" );
					return;
				}
			}			
		});

		quit.addActionListener( new ActionListener() {
			public void actionPerformed( ActionEvent e ) {
				//發(fā)送QUIT消息，收到服務(wù)器確認(rèn)后調(diào)用leave
				sendMessage( "QUIT: " + nickName );
				leave();
			}
		});
		
		this.addWindowListener( new WindowAdapter() {
			public void windowClosing( WindowEvent e ) {
				if( cThread != null ) {
					sendMessage( "QUIT: " + nickName );
					leave();
				}
				System.exit( 0 );
			}
		});
		//先判斷是不是悄悄話
		send.addActionListener( new ActionListener() {
			public void actionPerformed( ActionEvent e ) {
				//發(fā)送消息
				String s = tf.getText();				

				if( s.length() >= 100 ) {
					appendPersonalSystemMsg( "超出消息長(zhǎng)度限制" );
					return;
				}

				if( s.length() != 0 ) {
					if( ! isPersonal.isSelected() )
						sendMessage( nickName + ": " + s );
					else {
						int i = friends.getSelectedRow();
						if( i == -1 ) {
							//沒(méi)有選中任何用戶
							appendPersonalSystemMsg( "請(qǐng)先選中一位好友\n" );
							return;
						}
						else {
							//選中了一位好友
							String name = (String) friends.getValueAt( i , 0 );
							sendMessage( "TALK " + name + " " + nickName + " " + s );
						}
					}
				}
				tf.setText( "" );
			}
		});
		//先判斷是不是悄悄話
		tf.addActionListener( new ActionListener() { 
			public void actionPerformed( ActionEvent e ) {
				//發(fā)送消息
				String s = tf.getText();		
				
				if( s.length() >= 100 ) {
					appendPersonalSystemMsg( "超出消息長(zhǎng)度限制" );
					return;
				}

				if( s.length() != 0 ) {
					if( ! isPersonal.isSelected() )
						sendMessage( nickName + ": " + s );
					else {
						int i = friends.getSelectedRow();
						if( i == -1 ) {
							//沒(méi)有選中任何用戶
							appendPersonalSystemMsg( "請(qǐng)先選中一位好友\n" );
							return;
						}
						else {
							//選中了一位好友
							String name = (String) friends.getValueAt( i , 0 );
							sendMessage( "TALK " + name + " " + nickName + " " + s );
						}
					}
				}
				tf.setText( "" );
			}
		});

		tp.addMouseListener( new MouseAdapter() {
			public void mouseReleased( MouseEvent e ) {
				if( e.isPopupTrigger() ) {
					menu.show( e.getComponent(), e.getX(), e.getY() );
				}
			}
		});

		clear.addActionListener( new ActionListener() {
			public void actionPerformed( ActionEvent e ) {
				try
				{
					doc.remove( 0, doc.getLength() );
				}
				catch ( BadLocationException er )
				{
					er.printStackTrace();
				}
			}
		});
		
		friends.getSelectionModel().addListSelectionListener( new ListSelectionListener() {
			public void valueChanged( ListSelectionEvent e ) {
				if( e.getValueIsAdjusting() )
					return;
				ListSelectionModel lsm = (ListSelectionModel)e.getSource();
				if( lsm.isSelectionEmpty() )
					file.setEnabled( false );
				else
					if( ! isFileTrans )
					file.setEnabled( true );
			}
		});

		//發(fā)送文件
		file.addActionListener( new ActionListener() {
			public void actionPerformed( ActionEvent e ) {
				JFileChooser jfc = new JFileChooser( "." );
				jfc.setMultiSelectionEnabled( false );
				int result = jfc.showOpenDialog( Client.this );
				if( result == JFileChooser.APPROVE_OPTION ) {
					File file = jfc.getSelectedFile();
			//		JOptionPane.showMessageDialog( Client.this, file.getPath() );

					int i = friends.getSelectedRow();
					String s = (String)friends.getValueAt( i, 0 );

					//啟動(dòng)發(fā)送文件線程
					ft = new FileSendThread( Client.this, file.getPath() , s );
					((FileSendThread)ft).start();

					
					try
					{
						sendMessage( "FILE " + s + " " + nickName + " " + InetAddress.getLocalHost().getHostAddress() 
						+ " " + ((FileSendThread)ft).sSkt.getLocalPort() + " " + file.length()
						+ " " + file.getName() ); 
			//			JOptionPane.showMessageDialog( Client.this, "等待對(duì)方回應(yīng)" );
					}
					catch ( UnknownHostException er )
					{
						JOptionPane.showMessageDialog( Client.this, "無(wú)法解析本機(jī)地址" );
						ft = null;
					}					
				}
			}
		});
		//取消文件傳送
		cancel.addActionListener( new ActionListener() {
			public void actionPerformed ( ActionEvent e ) {
				if( ft.getClass() == FileReceiveThread.class ) {
					((FileReceiveThread)ft).stop();
					ft.dispose();
					sendMessage( "INTERRUPT " + ((FileReceiveThread)ft).name + " " + nickName );
				}
				if( ft.getClass() == FileSendThread.class ) {
					((FileSendThread)ft).stop();
					ft.dispose();
					sendMessage( "INTERRUPT " + ((FileSendThread)ft).name + " " + nickName );
				}
			}
		});
	}

	private void reLocate()
	{
		setResizable( false );
		pack();
		//重新設(shè)定窗口位置
		int screenWidth = getToolkit().getScreenSize().width;
		int screenHeight = getToolkit().getScreenSize().height;		
		setLocation( screenWidth/2 - getSize().width/2, screenHeight/2 - getSize().height/2 );		
	}

	public void leave()
	{
		//刪除好友列表
		removeAllFriends();