/*
 * Copyright 1998-1999 Be, Inc. All Rights Reserved.
 * Copyright 2003-2019 Haiku, Inc. All rights reserved.
 * Distributed under the terms of the MIT License.
 */


#include "SftpClient.h"

#include <unistd.h>

#include <String.h>


SftpClient::SftpClient()
        :
        SpawningUploadClient()
{
}


SftpClient::~SftpClient()
{
        if (OutputPipe() >= 0)
                SendCommand("quit\n");
}


bool
SftpClient::ChangeDir(const string& dir)
{
        int len;
        BString cmd("cd");
        BString reply;

        cmd << " " << dir.c_str() << "\n";
        SendCommand(cmd.String());

        len = ReadReply(&reply);
        if (len < 0) {
                fprintf(stderr, _GetReadText(), len);
                return false;
        }

        fprintf(stderr, _GetReplyText(), reply.String());
        if (reply.FindFirst("sftp>") < 0)
                return false;

        return true;
}


bool
SftpClient::ListDirContents(string& listing)
{
        int len;
        BString reply;

        SendCommand("ls\n");
        do {
                len = ReadReply(&reply);
                if (len < 0) {
                        fprintf(stderr, _GetReadText(), len);
                        return false;
                }
                fprintf(stderr, _GetReplyText(), reply.String());
                printf("%s", reply.String());
                if (reply.FindFirst("sftp>") == 0)
                        return true;
        } while (true);

        return false;
}


bool
SftpClient::PrintWorkingDir(string& dir)
{
        SendCommand("pwd\n");
        return false;
}


bool
SftpClient::Connect(const string& server, const string& login,
        const string& passwd)
{
        BString cmd("sftp ");
        BString host(server.c_str());
        BString port;

        if (host.FindFirst(':'))
                host.MoveInto(port, host.FindFirst(':'), host.Length());
        port.RemoveAll(":");

        if (port.Length())
                cmd << "-oPort=" << port << " ";

        cmd << login.c_str();
        cmd << "@" << host.String();
        printf("COMMAND: '%s'\n", cmd.String());
        SetCommandLine(cmd.String());

        if (!SpawningUploadClient::Connect(server, login, passwd))
                return false;

        BString reply;
        ssize_t len;

        len = ReadReply(&reply);
        if (len < 0) {
                fprintf(stderr, _GetLongReadText(), len);
                return false;
        }
        fprintf(stderr, _GetReplyText(), reply.String());
        if (reply.FindFirst("Connecting to ") != 0)
                return false;

        len = ReadReply(&reply);
        if (len < 0) {
                fprintf(stderr, _GetLongReadText(), len);
                return false;
        }

        fprintf(stderr, _GetReplyText(), reply.String());
        if (reply.FindFirst(/*[pP]*/"assword:") < 0)
                return false;

        write(OutputPipe(), passwd.c_str(), strlen(passwd.c_str()));
        write(OutputPipe(), "\n", 1);

        len = ReadReply(&reply);
        if (len < 0) {
                fprintf(stderr, _GetLongReadText(), len);
                return false;
        }

        fprintf(stderr, _GetReplyText(), reply.String());
        if (reply != "\n")
                return false;

        len = ReadReply(&reply);
        if (len < 0) {
                fprintf(stderr, _GetLongReadText(), len);
                return false;
        }

        fprintf(stderr, _GetReplyText(), reply.String());
        if (reply.FindFirst("sftp>") < 0)
                return false;

        return true;
}


bool
SftpClient::PutFile(const string& local, const string& remote, ftp_mode mode)
{
        int len;
        //XXX: handle mode ?
        BString cmd("put");
        cmd << " " << local.c_str() << " " << remote.c_str() << "\n";
        SendCommand(cmd.String());

        BString reply;

        len = ReadReply(&reply);
        if (len < 0) {
                fprintf(stderr, _GetReadText(), len);
                return false;
        }

        fprintf(stderr, _GetReplyText(), reply.String());
        if (reply.FindFirst("Uploading") < 0)
                return false;

        len = ReadReply(&reply);
        if (len < 0) {
                fprintf(stderr, _GetReadText(), len);
                return false;
        }

        fprintf(stderr, _GetReplyText(), reply.String());
        if (reply.FindFirst("sftp>") < 0)
                return false;

        return true;
}


bool
SftpClient::GetFile(const string& remote, const string& local, ftp_mode mode)
{
        //XXX
        return false;
}


// Note: this only works for local remote moves, cross filesystem moves
// will not work
bool
SftpClient::MoveFile(const string& oldPath, const string& newPath)
{
        int len;

        // sftpd can't rename to an existing file...
        BString cmd("rm");
        cmd << " " << newPath.c_str() << "\n";
        fprintf(stderr, "CMD: '%s'\n", cmd.String());
        SendCommand(cmd.String());
        BString reply;

        len = ReadReply(&reply);
        if (len < 0) {
                fprintf(stderr, _GetReadText(), len);
                return false;
        }

        fprintf(stderr, _GetReplyText(), reply.String());
        // we don't care if it worked or not.
        //if (reply.FindFirst("Removing") != 0 && reply.FindFirst("Couldn't") )
        //      return false;

        len = ReadReply(&reply);
        if (len < 0) {
                fprintf(stderr, _GetReadText(), len);
                return false;
        }

        fprintf(stderr, _GetReplyText(), reply.String());
        if (reply.FindFirst("sftp>") < 0)
                return false;

        cmd = "rename";
        cmd << " " << oldPath.c_str() << " " << newPath.c_str() << "\n";
        SendCommand(cmd.String());

        len = ReadReply(&reply);
        if (len < 0) {
                fprintf(stderr, _GetReadText(), len);
                return false;
        }

        fprintf(stderr, _GetReplyText(), reply.String());
        if (reply.FindFirst("sftp>") < 0)
                return false;

        return true;
}


bool
SftpClient::Chmod(const string& path, const string& mod)
{
        int len;
        //XXX: handle mode ?
        BString cmd("chmod");
        cmd << " " << mod.c_str() << " " << path.c_str() << "\n";
        SendCommand(cmd.String());

        BString reply;

        len = ReadReply(&reply);
        if (len < 0) {
                fprintf(stderr, _GetReadText(), len);
                return false;
        }

        fprintf(stderr, _GetReplyText(), reply.String());
        if (reply.FindFirst("Changing") < 0)
                return false;

        len = ReadReply(&reply);
        if (len < 0) {
                fprintf(stderr, _GetReadText(), len);
                return false;
        }

        fprintf(stderr, _GetReplyText(), reply.String());
        if (reply.FindFirst("sftp>") < 0)
                return false;

        return true;
}


void
SftpClient::SetPassive(bool on)
{
}


const char*
SftpClient::_GetLongReadText() const
{
        return B_TRANSLATE("read: %ld\n");
}


const char*
SftpClient::_GetReadText() const
{
        return B_TRANSLATE("read: %d\n");
}


const char*
SftpClient::_GetReplyText() const
{
        return B_TRANSLATE("reply: '%s'\n");
}