<?php

namespace WPLEClient;

use WPLEClient\Exceptions\LEAccountException;

/**
 * LetsEncrypt Account class, containing the functions and data associated with a LetsEncrypt account.
 *
 * PHP version 5.2.0
 *
 * MIT License
 *
 * Copyright (c) 2018 Youri van Weegberg
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 *
 * @author     Youri van Weegberg <youri@yourivw.nl>
 * @copyright  2018 Youri van Weegberg
 * @license    https://opensource.org/licenses/mit-license.php  MIT License
 * @link       https://github.com/yourivw/LEClient
 * @since      Class available since Release 1.0.0
 */
class LEAccount
{
    private $connector;
    private $accountKeys;

    public $id;
    public $key;
    public $contact;
    public $agreement;
    public $initialIp;
    public $createdAt;
    public $status;

    private $log;

    /**
     * Initiates the LetsEncrypt Account class.
     *
     * @param LEConnector	$connector 		The LetsEncrypt Connector instance to use for HTTP requests.
     * @param int 			$log 			The level of logging. Defaults to no logging. LOG_OFF, LOG_STATUS, LOG_DEBUG accepted.
     * @param array 		$email	 		The array of strings containing e-mail addresses. Only used when creating a new account.
     * @param array 		$accountKeys 	Array containing location of account keys files.
     */
    public function __construct($connector, $log, $email, $accountKeys)
    {
        $this->connector = $connector;
        $this->accountKeys = $accountKeys;
        $this->log = $log;

        if (!file_exists($this->accountKeys['private_key']) or !file_exists($this->accountKeys['public_key'])) {
            if ($this->log instanceof \Psr\Log\LoggerInterface) {
                $this->log->info('No account found, attempting to create account.');
            } else if ($this->log >= LECLient::LOG_STATUS) LEFunctions::log('No account found, attempting to create account.', 'function LEAccount __construct');

            LEFunctions::RSAgenerateKeys(null, $this->accountKeys['private_key'], $this->accountKeys['public_key']);
            $this->connector->accountURL = $this->createLEAccount($email);
        } else {
            $this->connector->accountURL = $this->getLEAccount();
        }
        if ($this->connector->accountURL == false) throw LEAccountException::AccountNotFoundException();
        $this->getLEAccountData();
    }

    /**
     * Creates a new LetsEncrypt account.
     *
     * @param array 	$email 	The array of strings containing e-mail addresses.
     *
     * @return object	Returns the new account URL when the account was successfully created, false if not.
     */
    private function createLEAccount($email)
    {
        if (!is_array($email)) $email = array($email);

        $contact = array_map(function ($addr) {
            return empty($addr) ? '' : (strpos($addr, 'mailto') === false ? 'mailto:' . $addr : $addr);
        }, $email);

        $sign = $this->connector->signRequestJWK(array('contact' => $contact, 'termsOfServiceAgreed' => true), $this->connector->newAccount);
        $post = $this->connector->post($this->connector->newAccount, $sign);
        if ($post['status'] === 201) {
            if (preg_match('~Location: (\S+)~i', $post['header'], $matches)) return trim($matches[1]);
        }
        return false;
    }

    /**
     * Gets the LetsEncrypt account URL associated with the stored account keys.
     *
     * @return object	Returns the account URL if it is found, or false when none is found.
     */
    private function getLEAccount()
    {
        $sign = $this->connector->signRequestJWK(array('onlyReturnExisting' => true), $this->connector->newAccount);
        $post = $this->connector->post($this->connector->newAccount, $sign);

        if ($post['status'] === 200) {
            if (preg_match('~Location: (\S+)~i', $post['header'], $matches)) return trim($matches[1]);
        }
        return false;
    }

    /**
     * Gets the LetsEncrypt account data from the account URL.
     */
    private function getLEAccountData()
    {
        $sign = $this->connector->signRequestKid(array('' => ''), $this->connector->accountURL, $this->connector->accountURL);
        $post = $this->connector->post($this->connector->accountURL, $sign);
        if ($post['status'] === 200) {
            $this->id = isset($post['body']['id']) ? $post['body']['id'] : '';
            $this->key = $post['body']['key'];
            $this->contact = $post['body']['contact'];
            $this->agreement = isset($post['body']['agreement']) ? $post['body']['agreement'] : '';
            $this->initialIp = $post['body']['initialIp'];
            $this->createdAt = $post['body']['createdAt'];
            $this->status = $post['body']['status'];
        } else {
            throw LEAccountException::AccountNotFoundException();
        }
    }

    /**
     * Updates account data. Now just supporting new contact information.
     *
     * @param array 	$email	The array of strings containing e-mail adresses.
     *
     * @return boolean	Returns true if the update is successful, false if not.
     */
    public function updateAccount($email)
    {
        $contact = array_map(function ($addr) {
            return empty($addr) ? '' : (strpos($addr, 'mailto') === false ? 'mailto:' . $addr : $addr);
        }, $email);

        $sign = $this->connector->signRequestKid(array('contact' => $contact), $this->connector->accountURL, $this->connector->accountURL);
        $post = $this->connector->post($this->connector->accountURL, $sign);
        if ($post['status'] === 200) {
            $this->id = isset($post['body']['id']) ? $post['body']['id'] : '';
            $this->key = $post['body']['key'];
            $this->contact = $post['body']['contact'];
            $this->agreement = isset($post['body']['agreement']) ? $post['body']['agreement'] : '';
            $this->initialIp = $post['body']['initialIp'];
            $this->createdAt = $post['body']['createdAt'];
            $this->status = $post['body']['status'];
            if ($this->log instanceof \Psr\Log\LoggerInterface) {
                $this->log->info('Account data updated.');
            } else if ($this->log >= LEClient::LOG_STATUS) LEFunctions::log('Account data updated.', 'function updateAccount');
            return true;
        } else {
            return false;
        }
    }

    /**
     * Creates new RSA account keys and updates the keys with LetsEncrypt.
     *
     * @return boolean	Returns true if the update is successful, false if not.
     */
    public function changeAccountKeys()
    {
        LEFunctions::RSAgenerateKeys(null, $this->accountKeys['private_key'] . '.new', $this->accountKeys['public_key'] . '.new');
        $oldPrivateKey = openssl_pkey_get_private(file_get_contents($this->accountKeys['private_key']));
        $oldDetails = openssl_pkey_get_details($oldPrivateKey);
        $innerPayload = array('account' => $this->connector->accountURL, 'oldKey' => array(
            "kty" => "RSA",
            "n" => LEFunctions::Base64UrlSafeEncode($oldDetails["rsa"]["n"]),
            "e" => LEFunctions::Base64UrlSafeEncode($oldDetails["rsa"]["e"])
        ));
        $outerPayload = $this->connector->signRequestJWK($innerPayload, $this->connector->keyChange, $this->accountKeys['private_key'] . '.new');
        $sign = $this->connector->signRequestKid($outerPayload, $this->connector->accountURL, $this->connector->keyChange);
        $post = $this->connector->post($this->connector->keyChange, $sign);
        if ($post['status'] === 200) {
            unlink($this->accountKeys['private_key']);
            unlink($this->accountKeys['public_key']);
            rename($this->accountKeys['private_key'] . '.new', $this->accountKeys['private_key']);
            rename($this->accountKeys['public_key'] . '.new', $this->accountKeys['public_key']);

            $this->getLEAccountData();

            if ($this->log instanceof \Psr\Log\LoggerInterface) {
                $this->log->info('Account keys changed.');
            } elseif ($this->log >= LEClient::LOG_STATUS) LEFunctions::log('Account keys changed.', 'function changeAccountKey');
            return true;
        } else {
            return false;
        }
    }

    /**
     * Deactivates the LetsEncrypt account.
     *
     * @return boolean	Returns true if the deactivation is successful, false if not.
     */
    public function deactivateAccount()
    {
        $sign = $this->connector->signRequestKid(array('status' => 'deactivated'), $this->connector->accountURL, $this->connector->accountURL);
        $post = $this->connector->post($this->connector->accountURL, $sign);
        if ($post['status'] === 200) {
            $this->connector->accountDeactivated = true;
            if ($this->log instanceof \Psr\Log\LoggerInterface) {
                $this->log->info('Account deactivated.');
            } elseif ($this->log >= LEClient::LOG_STATUS) LEFunctions::log('Account deactivated.', 'function deactivateAccount');

            return true;
        } else {
            return false;
        }
    }
}
