package com.instantbank.collections.util;

import java.net.InetAddress;
import java.net.ServerSocket;
import java.net.Socket;
import java.net.UnknownHostException;
import java.io.BufferedInputStream;
import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.PrintWriter;

public class FTPServices {
  Socket ctrlSocket;
  public PrintWriter ctrlOutput;
  public BufferedReader ctrlInput;
  final int CTRLPORT = 21;
  String serverType;


  public FTPServices() { }


  public void closeConnection() throws IOException {
    try {
      ctrlOutput.println("QUIT ");
      ctrlOutput.flush();
      ctrlSocket.close();
    }
    catch(IOException e) {
      throw e;
    }
  }


  public Socket dataConnection(String ctrlcmd) throws Exception {
    String cmd = "PORT ";
    int i;
    Socket dataSocket = null;

    try {
      byte[] address = InetAddress.getLocalHost().getAddress();

      ServerSocket serverDataSocket = new ServerSocket(0, 1);

      for(i = 0; i < 4; ++i) {
        cmd = cmd + (address[i] & 0xff) + ",";
      }
      cmd = cmd + (((serverDataSocket.getLocalPort()) / 256) & 0xff)
        + ","
        + (serverDataSocket.getLocalPort() & 0xff);

      ctrlOutput.println(cmd);
      ctrlOutput.flush();

      ctrlOutput.println(ctrlcmd);
      ctrlOutput.flush();

      serverDataSocket.setSoTimeout(5000);
      dataSocket = serverDataSocket.accept();
      serverDataSocket.close();

    }
    catch(Exception e) {
      throw e;
    }
    return dataSocket;
  }


  public void doAscii() throws Exception {
    try {
      ctrlOutput.println("TYPE A");
      ctrlOutput.flush();
    }
    catch(Exception e) {
      throw e;
    }
  }


  public void doBinary() throws Exception {
    try {
      ctrlOutput.println("TYPE I");
      ctrlOutput.flush();
    }
    catch(Exception e) {
      throw e;
    }
  }


  public void doCd(String dirName) throws Exception {
    try {
      ctrlOutput.println("CWD " + dirName);
      ctrlOutput.flush();
    }
    catch(Exception e) {
      throw e;
    }
  }


  public void doLs() throws Exception {
    try {
      int n;
      byte[] buff = new byte[1024];

      Socket dataSocket = dataConnection("LIST");

      BufferedInputStream dataInput = new BufferedInputStream(dataSocket.getInputStream());

      while((n = dataInput.read(buff)) > 0) {
        System.out.write(buff, 0, n);
      }
      dataSocket.close();
    }
    catch(Exception e) {
      throw e;
    }
  }


  public void doGet(String fileName, String path) throws Exception {

    try {
      byte[] buff = new byte[1024];
      FileServices f = new FileServices();
      int n;
      Socket dataSocket = null;

      FileOutputStream outfile = new FileOutputStream(fileName);
      if(serverType.equals("windows")) {
        dataSocket = dataConnection("RECV " + fileName);
      }
      else
        if(serverType.equals("unix")) {
        dataSocket = dataConnection("RETR " + fileName);
      }

      BufferedInputStream dataInput = new BufferedInputStream(dataSocket.getInputStream());
      while((n = dataInput.read(buff)) > 0) {
        outfile.write(buff, 0, n);
      }
      dataSocket.close();
      outfile.close();
      f.copy(fileName, "", path);
      f.delete(fileName, "");

    }
    catch(Exception e) {
      throw e;
    }
  }


  public void doPut(String fileName, String path) throws Exception {

    try {
      byte[] buff = new byte[1024];
      FileServices f = new FileServices();
      int n;
      FileInputStream sendfile = null;
      Socket dataSocket = null;

      f.copy(fileName, path, "");

      try {
        sendfile = new FileInputStream(fileName);
      }
      catch(Exception e) {
        throw e;
      }

      if(serverType.equals("windows")) {
        dataSocket = dataConnection("SEND " + fileName);
      }
      else
        if(serverType.equals("unix")) {
        dataSocket = dataConnection("STOR " + fileName);
      }

      OutputStream outstr = dataSocket.getOutputStream();

      while((n = sendfile.read(buff)) > 0) {
        outstr.write(buff, 0, n);
      }
      dataSocket.close();
      sendfile.close();

      f.delete(fileName, "");

    }
    catch(Exception e) {
      throw e;
    }
  }


  public FTPServices(String type) {
    serverType = type;
  }


  public String getServerType(String type) {
    return serverType;
  }


  public void openConnection(String host, String userName, String password) throws IOException, UnknownHostException {
    openConnection(host, CTRLPORT, userName, password);
  }


  public void openConnection(String host, int port, String userName, String password) throws IOException, UnknownHostException {
    try {
      ctrlSocket = new Socket(host, port);
      ctrlOutput = new PrintWriter(ctrlSocket.getOutputStream());
      ctrlInput = new BufferedReader(new InputStreamReader(ctrlSocket.getInputStream()));

      ctrlOutput.println("USER " + userName);
      ctrlOutput.flush();
      ctrlOutput.println("PASS " + password);
      ctrlOutput.flush();
    }
    catch(IOException e) {
      throw e;
    }
  }


  public void setServerType(String type) {
    serverType = type;
  }

}