/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the MIT License. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

import KeyStoreConstants from './keystores/KeyStoreConstants';
import IdentifierDocument from './IdentifierDocument';
import UserAgentOptions from './UserAgentOptions';
import UserAgentError from './UserAgentError';
import { ProtectionFormat } from './crypto/keyStore/ProtectionFormat';
import SubtleCryptoExtension from './crypto/plugin/SubtleCryptoExtension';
import JwsToken from './crypto/protocols/jose/jws/JwsToken';
import PrivateKey from './crypto/keys/PrivateKey';
import { IJwsSigningOptions, IJweEncryptionOptions } from "./crypto/protocols/jose/IJoseOptions";
import { IdentifierDocumentPublicKey } from './types';
import CryptoHelpers from './crypto/utilities/CryptoHelpers';
import KeyUseFactory, { KeyUse } from './crypto/keys/KeyUseFactory';
import JweToken from './crypto/protocols/jose/jwe/JweToken';
import JoseConstants from './crypto/protocols/jose/JoseConstants';
import PublicKey from './crypto/keys/PublicKey';
import UserAgentConstants from './UserAgentConstants';
import KeyContainer from './crypto/keys/KeyContainer';
import JoseHelpers from './crypto/protocols/jose/JoseHelpers';

/**
 * Class for creating and managing identifiers,
 * retrieving identifier documents.
 */
export default class Identifier {
  /**
   * The string representation of the identier for the persona
   * in the format 'did:{method}:{id}'.
   */
  public id: string;

  /**
   * The identifier document for the identifier
   * if one exists.
   */
  public document: IdentifierDocument | undefined;

  /**
   * User Agent Options
   */
  public options: UserAgentOptions | undefined;

  /**
   * Constructs an instance of the Identifier
   * class using the provided identifier or identifier document.
   * @param identifier either the string representation of an identifier or a identifier document.
   * @param [options] for configuring how to register and resolve identifiers.
   */
  constructor (public identifier: IdentifierDocument | string, options?: UserAgentOptions) {
    // Check whether passed an identifier document
    // or an identifier string
    if (typeof identifier === 'object') {
      this.document = identifier;
      this.id = identifier.id;
    } else {
      this.id = identifier;
    }
    this.options = options;
  }

  /**
   * Creates a new decentralized identifier.
   * @param [options] for configuring how to register and resolve identifiers.
   */
  public static async create (options: UserAgentOptions): Promise<Identifier> {
    const id = <string> options.didPrefix;
    return new Identifier(id, options).createLinkedIdentifier(id, true);
  }

  /**
   * Creates a new decentralized identifier, using the current identifier
   * and the specified target. If the registar flag is true, the newly created
   * identifier will be registered using the
   * @param target entity for which to create the linked identifier
   * @param register flag indicating whether the new identifier should be registered
   * with a ledger.
   */
  public async createLinkedIdentifier (target: string, register: boolean = false): Promise<Identifier> {
    if (this.options && this.options.keyStore) {

      // Create DID key
      const cryptoFactory = this.options.cryptoFactory;
      const signingKeyStorageId = Identifier.keyStorageIdentifier(
        this.id, 
        target, 
        this.options.cryptoOptions.signingAlgorithm, 
        JoseHelpers.createUseViaJwa(this.options.cryptoOptions.signingAlgorithm));
      const signingPublicKey = await this.generateAndSaveKey(
        new SubtleCryptoExtension(cryptoFactory), 
        this.options.cryptoOptions.signingAlgorithm, 
        target, 
        `#${UserAgentConstants.keyTagSigning}1`,
        signingKeyStorageId || this.options.cryptoOptions.signingKeyReference);

        const encryptionKeyStorageId = Identifier.keyStorageIdentifier(
          this.id, 
          target, 
          this.options.cryptoOptions.encryptionAlgorithm, 
          JoseHelpers.createUseViaJwa(this.options.cryptoOptions.encryptionAlgorithm));
        const encryptionPublicKey = await this.generateAndSaveKey(
        new SubtleCryptoExtension(cryptoFactory), 
        this.options.cryptoOptions.encryptionAlgorithm, 
        target,
        `#${UserAgentConstants.keyTagEncryption}1`,
        encryptionKeyStorageId || this.options.cryptoOptions.encryptionKeyReference);

      // Set key format
      // todo switch by leveraging pairwiseKey
      const signingDocumentKey: IdentifierDocumentPublicKey = {
        id: <string>signingPublicKey.kid,
        type: this.getDidDocumentKeyType(),
        publicKeyJwk: signingPublicKey
      };
      const encryptionDocumentKey: IdentifierDocumentPublicKey = {
        id: <string>encryptionPublicKey.kid,
        // we need to add RsaEncryptionKey 2018 as type - todo
        type: 'RsaVerificationKey2018',
        publicKeyJwk: encryptionPublicKey
      };
      let identifier: Identifier;
  
      if (this.options.registrar) {
        // add encryptionDocumentKey to register the encryption key
        const document = await this.createIdentifierDocument(this.id, [signingDocumentKey, encryptionDocumentKey]);
        if (register) {
            // register did document
          identifier = await this.options.registrar.register(document, signingKeyStorageId);
          document.id = identifier.id;
        }
        identifier =  new Identifier(document, this.options);
        // If we create a new identifier save the signing key
        if (target === this.id) {
          this.options.cryptoOptions.signingKeyReference = signingKeyStorageId;
          this.options.cryptoOptions.encryptionKeyReference = encryptionKeyStorageId;
        }
        return identifier;
      } else {
        throw new UserAgentError(`No registrar in options to register DID document`);
      }
    }
    throw new UserAgentError('No keyStore in options');
  }

  /**
   * Generate a key and save it into the store
   * @param generator interface
   * @param algorithm for the key
   * @param target id of peer
   */
  private async generateAndSaveKey(generator: SubtleCryptoExtension, algorithm: string, target: string, kid: string, keyReference: string | undefined): Promise<PublicKey> {
    const alg = CryptoHelpers.jwaToWebCrypto(algorithm);
    const jwk: PrivateKey = await generator.generatePairwiseKey(alg, KeyStoreConstants.masterSeed, this.id, target);
    jwk.kid = kid;
    jwk.use = JoseHelpers.createUseViaJwa(algorithm);
    const pubJwk = jwk.getPublicKey();
    
    pubJwk.kid = jwk.kid;

    const pairwiseKeyStorageId = keyReference || Identifier.keyStorageIdentifier(
      this.id, 
      target, 
      algorithm, 
      JoseHelpers.createUseViaJwa(algorithm));
    await (<UserAgentOptions>this.options).keyStore.save(pairwiseKeyStorageId, new KeyContainer(jwk));
    return pubJwk;
  }

  /**
   * Gets the IdentifierDocument for the identifier
   * instance, throwing if no identifier has been
   * created.
   */
  public async getDocument (): Promise<IdentifierDocument> {
    // If we already have not already
    // retrieved the document use the
    // resolver to get the document
    if (!this.document) {
      if (!this.options || !this.options.resolver) {
        throw new UserAgentError('Resolver not specified in user agent options.');
      }

      // We need to resolve the document
      this.document = <IdentifierDocument> await this.options.resolver.resolve(this);
    }

    return this.document;
  }

  /**
   * Performs a public key lookup using the
   * specified key identifier, returning the
   * key defined in document.
   * @param keyIdentifier the identifier of the public key.
   */
  public async getPublicKey (keyIdentifier?: string): Promise<IdentifierDocumentPublicKey> {
    if (!this.document) {
      await this.getDocument();
    }

    // If we have been provided a key identifier use
    // the identifier to look up a key in the document
    if (this.document && this.document.publicKeys && keyIdentifier) {
      const index = this.document.publicKeys.findIndex((key: any) => key.id === keyIdentifier);
      // trim down the key Identifier to the unique keyID
      const keyIdentifierComponents = keyIdentifier.split('#');
      const keyId = keyIdentifierComponents[keyIdentifierComponents.length - 1];
      const matchingPublicKeys = this.document.publicKeys.filter((PublicKey) => PublicKey.id.endsWith(keyId));
      if (matchingPublicKeys.length === 0) {
        throw new UserAgentError(`No matching public key found for ${keyIdentifier}`);
      }
      return matchingPublicKeys[0];
    } else if (this.document && this.document.publicKeys && this.document.publicKeys.length > 0) {
      // If only one key has been specified in the document
      // return that
      return this.document.publicKeys[0];
    }

    throw new UserAgentError('Document does not contain any public keys');
  }

  /**
   * Generate a storage identifier to store a key
   * @param personaId The identifier for the persona
   * @param target The identifier for the peer. Will be persona for non-pairwise keys
   * @param algorithm Key algorithm
   * @param keyType Key type
   */
  public static keyStorageIdentifier (personaId: string, target: string, algorithm: string, keyType: string): string {
    console.log(`${personaId}-${target}-${algorithm}-${keyType}`);
    return `${personaId}-${target}-${algorithm}-${keyType}`;
  }

  // Create an identifier document. Included the public key.
  private async createIdentifierDocument (id: string, publicKeys: IdentifierDocumentPublicKey[]): Promise <IdentifierDocument> {
    return IdentifierDocument.createAndGenerateId(id, publicKeys, <UserAgentOptions> this.options);
  }

  // Get the did document public key type
  private getDidDocumentKeyType () {
    // Support other key types
    return 'Secp256k1VerificationKey2018';
  }

  /**
   * Sign payload with key specified by keyStorageIdentifier in options.keyStore
   * @param payload object to be signed
   * @param keyReference the identifier for the key used to sign payload.
   */
  public async sign (payload: any, keyReference: string): Promise<string> {
    let body: string;
    if (this.options && this.options.cryptoOptions) {
      if (this.options.keyStore) {
        if (typeof(payload) !== 'string') {
          body = JSON.stringify(payload);
        } else {
          body = payload;
        }
        const signingOptions: IJwsSigningOptions = {
          cryptoFactory: this.options.cryptoFactory
        };
        const jws = new JwsToken(signingOptions);
        const signature = await jws.sign(keyReference, Buffer.from(body), ProtectionFormat.JwsCompactJson);
        return signature.serialize(ProtectionFormat.JwsCompactJson);;
      } else {
        throw new UserAgentError('No KeyStore in Options');
      }
    } else {
      throw new UserAgentError('No Crypto Options in User Agent Options');
    }
  }

  /**
   * Verify the payload with public key from the Identifier Document.
   * @param jws the signed token to be verified.
   */
  public async verify (jws: string): Promise<string> {

    if (!this.document) {
      this.document = await this.getDocument();
    }
    const signingOptions: IJwsSigningOptions = {
      cryptoFactory: (<UserAgentOptions>this.options).cryptoFactory
    };
    const token = JwsToken.deserialize(jws, signingOptions);
    if (await token.verify(this.document.getPublicKeysFromDocument(), signingOptions)) {
      return token.getPayload();
    }

    throw new UserAgentError(`The signature validation for '${this.id}' failed.`);
  }

  /**
   * Encrypt payload using Public Key registered on Identifier Document.
   * @param payload object that will be encrypted.
   * @param encryptionKeys used for the encryption.
   */
  public async encrypt (payload: any): Promise<string> {
    if (!this.options) {
      throw new UserAgentError('Options Undefined');
    }
    // get document if undefined
    if (!this.document) {
      this.document = await this.getDocument();
    }

    const keyStore = this.options.keyStore;
    const cryptoFactory = this.options.cryptoFactory;

    const options: IJweEncryptionOptions = {
      cryptoFactory: cryptoFactory,
      contentEncryptionAlgorithm: JoseConstants.AesGcm256
    };

    // create a jweToken with temp cryptoFactory and algorithm.
    const jweToken = new JweToken(options);

    // get any JWK key marked use as 'enc'
    const publicKey = this.document.getPublicKeysFromDocument().reduce((keyFound: undefined | PublicKey, currentKey: PublicKey): PublicKey | undefined => {
    if (keyFound) {
      return keyFound;
    }
    if (currentKey.use === KeyUse.Encryption) {
      return currentKey;
    }
      return undefined;
    }, undefined);
    if (!publicKey) {
      throw new UserAgentError(`No Public Key found with use equal to 'enc' for ${this.id}`);
    }
    // keyIDs retrieved from the DID Document may not be fully quantified
    if (publicKey.kid && publicKey.kid.indexOf('#') <= 0) {
      publicKey.kid = `${this.id}${publicKey.kid.indexOf('#') === -1 ? '#' : ''}${publicKey.kid}`;
    }

    // encrypt payload using public keys.
    const encryptedToken = await jweToken.encrypt([publicKey], payload, ProtectionFormat.JweCompactJson);

    // return serialized token.
    return encryptedToken.serialize(ProtectionFormat.JweCompactJson);
  }

  /**
   * Decrypt cipher using key referenced in keystore.
   * @param cipher cipher to be decrypted.
   * @param keyReference string that references what key to use from keystore.
   */
  public async decrypt (cipher: Buffer, keyReference: string): Promise<string> {
    
    if (!this.options) {
      throw new UserAgentError('Options Undefined');
    }

    const options = {
      cryptoFactory: this.options.cryptoFactory
    };
    const jweToken = JweToken.deserialize(cipher.toString(), <IJweEncryptionOptions>options);

    // create jweToken, feed in ciphertext, and decrypt.
    const payload = await jweToken.decrypt(keyReference);
    return payload.toString();
  }
}