package ma.ma01;

import drjava.util.StringUtil;
import net.luaos.tb.tb07.PasswordUtil;
import net.schmizz.sshj.SSHClient;
import net.schmizz.sshj.xfer.InMemorySourceFile;
import net.schmizz.sshj.xfer.scp.SCPFileTransfer;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class TinyBrainSCPUploader {
  public static void uploadIndexPHP(String hostKey, String url, String text) throws IOException {
    Matcher matcher = Pattern.compile("http://tinybrain.de:8080/(.*/)").matcher(url);
    if (!matcher.matches())
      throw new RuntimeException("URL not matched: " + url);

    String remoteDir = "/root/www-8080/" + matcher.group(1);
    String remoteFilePath = remoteDir + "index.php";
    String host = "tinybrain.de", user = "root";

    System.out.println("Uploading to: " + remoteFilePath);
    System.out.print("Password for " + user + "@" + host + ": ");
    char[] pw = PasswordUtil.readPasswordFromConsole();

    final SSHClient client = new SSHClient();
    client.loadKnownHosts();
    if (!StringUtil.isEmpty(hostKey))
      client.addHostKeyVerifier(hostKey);
    client.connect(host);
    try {
      client.authPassword(user, pw);
      client.newSFTPClient().mkdirs(remoteDir);
      uploadBySCP(client, remoteFilePath, text);
      System.out.println("File copied!");
    } finally {
      client.disconnect();
    }
  }

  private static void uploadBySCP(SSHClient client, String remoteFilePath, String text) throws IOException {
    uploadBySCP(client, remoteFilePath, text.getBytes("UTF-8"));
  }

  private static void uploadBySCP(SSHClient client, String remoteFilePath, final byte[] data) throws IOException {
    SCPFileTransfer fileTransfer = client.newSCPFileTransfer();
    fileTransfer.upload(new InMemorySourceFile() {
      public String getName() {
        return "index.php";
      }

      public long getLength() {
        return data.length;
      }

      public InputStream getInputStream() throws IOException {
        return new ByteArrayInputStream(data);
      }
    }, remoteFilePath);
  }
}