e107inc/e107

<?php

class SFTPDeployer extends Deployer
{
    public function start()
    {
        self::println();
        self::println("=== SFTP Deployer – Bring Up ===");
        if (in_array('fs', $this->components))
        {
            $this->start_fs();
        }
    }

    private function getFsParams()
    {
        return $this->params['fs'];
    }

    private function generateSshpassPrefix()
    {
        if (empty($this->getFsParam('privkey_path')) &&
            !empty($this->getFsParam('password')))
        {
            return 'sshpass -p '.escapeshellarg($this->getFsParam('password')).' ';
        }
        return '';
    }

    private function getFsParam($key)
    {
        return $this->getFsParams()[$key];
    }

    private function generateRsyncRemoteShell()
    {
        $prefix = 'ssh -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null -p '.
            escapeshellarg($this->getFsParam('port'));
        if (!empty($this->getFsParam('privkey_path')))
            return $prefix.' -i ' . escapeshellarg($this->getFsParam('privkey_path'));
        else
            return $prefix;
    }

    private static function runCommand($command, &$stdout = null, &$stderr = null)
    {
        $descriptorSpec = [
            1 => ['pipe', 'w'],
            2 => ['pipe', 'w'],
        ];
        $pipes = [];
        self::println("Running this command…:");
        self::println($command);
        $resource = proc_open($command, $descriptorSpec, $pipes, APP_PATH);
        $stdout = stream_get_contents($pipes[1]);
        $stderr = stream_get_contents($pipes[2]);
        self::println("---------- stdout ----------");
        self::println(trim($stdout));
        self::println("---------- stderr ----------");
        self::println(trim($stderr));
        self::println("----------------------------");
        foreach ($pipes as $pipe)
        {
            fclose($pipe);
        }
        return proc_close($resource);
    }

    public function stop()
    {
        self::println("=== SFTP Deployer – Tear Down ===");
    }

    public function unlinkAppFile($relative_path)
    {
        self::println("Deleting file \"$relative_path\" from deployed test location…");
        $fs_params = $this->getFsParams();
        $command = $this->generateSshpassPrefix().
            $this->generateRsyncRemoteShell().
            " ".escapeshellarg("{$fs_params['user']}@{$fs_params['host']}").
            " ".escapeshellarg("rm -v " . escapeshellarg(rtrim($fs_params['path'], '/')."/$relative_path"));
        $retcode = self::runCommand($command);
        if ($retcode === 0)
        {
            self::println("Deleted file \"$relative_path\" from deployed test location");
        }
        else
        {
            self::println("No such file to delete: \"$relative_path\"");
        }
    }

    private function start_fs()
    {
        $fs_params = $this->getFsParams();
        $fs_params['path'] = rtrim($fs_params['path'], '/') . '/';
        $command = $this->generateSshpassPrefix() .
            'rsync -e ' .
            escapeshellarg($this->generateRsyncRemoteShell()) .
            ' --delete -avzHXShs ' .
            escapeshellarg(rtrim(APP_PATH, '/') . '/') . ' ' .
            escapeshellarg("{$fs_params['user']}@{$fs_params['host']}:{$fs_params['path']}");
        $retcode = self::runCommand($command);
        if ($retcode !== 0) {
            throw new Exception("SFTP deployment failed. Run with --debug to see stdout and stderr.");
        }
    }
}