package org.sl.util;
import org.apache.commons.net.telnet.TelnetClient;
import java.io.*;
import java.nio.ByteBuffer;
public class TelnetUtil {
    String charset = null;
    byte[] buff = new byte[2048];
    TelnetClient telnetClient = new TelnetClient();
    BufferedReader telnetReader = null;
    BufferedWriter telnetWirter = null;
    InputStream telnetIn = null;
    OutputStream telnetOut = null;

    public TelnetUtil() {
        telnetClient = new TelnetClient();
    }

    /**
     * 連接至服務器
     * @param ip
     * @param port
     * @throws UnsupportedEncodingException
     * @throws IOException
     */
    public void connect(String ip, int port) throws UnsupportedEncodingException,IOException {
        telnetClient.connect(ip,port);
        setIOStream();
    }

    /**
     * 連接至服務器
      * @param ip
     * @throws UnsupportedEncodingException
     * @throws IOException
     */
    public void connect(String ip) throws UnsupportedEncodingException,IOException {
        telnetClient.connect(ip);
        setIOStream();
    }

    void setIOStream() throws UnsupportedEncodingException {
        telnetIn = telnetClient.getInputStream();
        telnetOut = telnetClient.getOutputStream();
        if(null==charset){
            telnetReader = new BufferedReader(new InputStreamReader(telnetIn));
            telnetWirter = new BufferedWriter(new OutputStreamWriter(telnetOut));
        }else{
            telnetReader = new BufferedReader(new InputStreamReader(telnetIn, charset));
            telnetWirter = new BufferedWriter(new OutputStreamWriter(telnetOut, charset));
        }
    }

    /**
     * 登錄
     * @param user
     * @param passwd
     * @return 是否登錄成功.
     * @throws IOException
     */
    public boolean login(String user,String passwd) throws IOException {
        String read = readString();
        for(int i=0; ; i++){
            if(-1==read.indexOf("login")){
                read = readString();
            }else{
                break;
            }
        }
        writeText(user);

        read = readString();
        for(int i=0; ; i++){
            if(-1==read.indexOf("Password")){
                read = readString();
            }else{
                break;
            }
        }
        writeText(passwd);

        for(;;){
            read = readString();
            //System.out.println("last:"+read);
            if(-1!=read.indexOf("Last")){
                return true;
            }else if(-1!=read.indexOf("incorrect")){
                return false;
            }
        }
    }

    /**
     * 這是一個測試方法,隨便寫。
     * @throws IOException
     */
    public void show() throws IOException {
//        System.out.println(readString());
//        System.out.println(readString());
//        ByteBuffer tmp = ByteBuffer.allocate(1024);
//        byte[] buff = new byte[1024];
//        while(telnetIn.available()>0){
//            int readLen = readBytes(buff,0,1024);
//            tmp.put(buff,0,readLen);
//        }

//        System.out.println(new String(tmp.array()));
        System.out.println("1 "+readString());
        System.out.println("2 "+readString());
        System.out.println("3 "+readString());
        writeText("root");
        System.out.println("4 " + readString());
        writeText("123456");
        System.out.println("5 "+readString());
//        System.out.println("6 "+readString());
//        System.out.println("7 "+readString());

    }

    public int readBytes(byte[] buff, int offset, int len) throws IOException {
        return telnetIn.read(buff,offset,len);
    }

    /**
     * 讀取字符串<br/>
     * 相當于readByte()轉為字符串
     * @return
     * @throws IOException
     */
    public String readString() throws IOException {
        int readLen = readBytes(this.buff, 0, this.buff.length);
        if(0<readLen)
            return new String(buff,0,readLen).trim();
        else
            return "";
    }

    /**
     * 讀取一行<br/>
     * 如果服務器與客戶端不是同一種操作系統，可能導致此方法計行失敗。
     * @return
     * @throws IOException
     */
    public String readLine() throws IOException {
        String read = telnetReader.readLine();
        return null==read?"":read.trim();
    }

    public void writeBytes(byte[] buff, int offset, int len) throws IOException {
        telnetOut.write(buff,offset,len);
    }

    /**
     * 向服務器寫字符串
     * @param text
     * @throws IOException
     */
    public void writeText(String text) throws IOException {
        telnetWirter.write(text);
        telnetWirter.write('/r');
        telnetWirter.write('/n');
        telnetWirter.flush();
    }

    /**
     * 執行命令，并返回結果<br/>
     * 相當于: <br>
     * writeText();  <br/>
     * return readString();
     * @param cmd
     * @return
     * @throws IOException
     */
    public String exec(String cmd) throws IOException {
        writeText(cmd);
        return readString();
    }

    String login1(String user,String passwd) throws IOException {
        String read = null;
        readString();
        readString();
        read = readString();

        if(-1!=read.indexOf("login")){
            writeText(user);
        }

        read = readString();
        if(-1!=read.indexOf("Password")){
            writeText(passwd);
        }

        read  = readString();
        read += readString();
        return read;

//        StringBuffer sb = new StringBuffer();
//        while(null!= (read = readString())){
//            sb.append(read);
//        }
//
//        return sb.toString();
    }

    /**
     * 關閉
     */
    public void close(){
        try{
            writeText("exit");
            writeText("exit");
            writeText("exit");
        }catch(Exception ex){
        }

        try {
            if(null!=telnetIn) telnetIn.close();
        } catch (Exception e) {
        }

        try {
            if(null!=telnetOut) telnetOut.close();
        } catch (Exception e) {
        }

        try {
            if(null!=telnetClient)telnetClient.disconnect();
        } catch (Exception e) {
        }
    }

    /**
     * 設置telnet通信時的字符集<br/>
     * 注:此字符集與服務器端字符集沒有必然關系<br/>
     * 此方法需在connect()前調用
     * @param charset
     */
    public void setCharset(String charset ){
        this.charset = charset;
    }

    /**
     * 重新設置buff大小,默認為2048字節.
     * @param size
     */
    public void setBufferSize(int size){
        this.buff = new byte[size];
    }
}

static void t4(){
        TelnetUtil tu = new TelnetUtil();
        try {
            tu.connect("192.168.2.154");
            System.out.println(tu.login("root", "123456"));
            //tu.show();
            //System.out.println(tu.readString());
            //System.out.println(tu.exec("pwd"));

            System.out.println(tu.exec("echo /"123456789/">1.txt"));
            System.out.println(tu.exec("cat 1.txt"));
        } catch (IOException e) {
            e.printStackTrace();
        }

        tu.close();
    }

    static void t1(){
        TelnetUtil tu = new TelnetUtil();
        try {
            tu.connect("192.168.2.154");
            System.out.println(tu.login("sl1", "coffee8215"));
            //tu.show();
            //System.out.println(tu.readString());
            System.out.println(tu.exec("pwd"));

        } catch (IOException e) {
            e.printStackTrace();
        }

        tu.close();
    }