<?php

namespace LEClient;

/**
 * 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
 * @version    1.2.2
 * @link       https://github.com/yourivw/LEClient
 * @since      Class available since Release 1.0.0
 */
if ( ! class_exists( 'LEAccount' ) ) {

	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 new \RuntimeException( 'Account not found or deactivated.' );
			}
			$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 ) {
			$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 new \RuntimeException( 'Account data cannot be found.' );
			}
		}

		/**
		 * 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;
			}
		}
	}
}

?>
