import * as AWS from 'aws-sdk';
import chalk from 'chalk';
import * as crypto from 'crypto';
import {AppContext} from './AppContext';
import {runtimeConfig} from './Config';
import {Rivendell} from './Rivendell';

export class AppUploader {
  constructor(private context: AppContext, private packageData: Buffer) {
  }

  public async upload(): Promise<string> {
    console.log(chalk.gray('Uploading package...'));
    const {vendor, appId, version} = this.context;
    const md5 = crypto.createHash('md5').update(this.packageData).digest('hex');
    const bucket = runtimeConfig().uploadBucket;
    const s3Key = `${vendor}/${appId}@${version}-${md5}.zip`;
    const credentials = await Rivendell.uploadCredentials(vendor!, appId!, version!, md5);

    AWS.config.update({
      region: runtimeConfig().awsRegion,
      credentials: new AWS.Credentials(
        credentials.accessKeyId,
        credentials.secretAccessKey,
        credentials.sessionToken
      )
    });

    await new AWS.S3()
      .putObject( {
        Bucket: bucket,
        Key: s3Key,
        Body: this.packageData
      })
      .promise();

    return `s3://${bucket}/${s3Key}`;
  }
}