Editor/CoreAPI/CoreApi.cs

// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0

using System;
using System.Collections.Generic;
using System.IO;
using AmazonGameLift.Runtime;
using AmazonGameLiftPlugin.Core.AccountManagement;
using AmazonGameLiftPlugin.Core.AccountManagement.Models;
using AmazonGameLiftPlugin.Core.ContainerManagement;
using AmazonGameLiftPlugin.Core.BucketManagement;
using AmazonGameLiftPlugin.Core.BucketManagement.Models;
using AmazonGameLiftPlugin.Core.CredentialManagement;
using AmazonGameLiftPlugin.Core.CredentialManagement.Models;
using AmazonGameLiftPlugin.Core.DeploymentManagement;
using AmazonGameLiftPlugin.Core.DeploymentManagement.Models;
using AmazonGameLiftPlugin.Core.GameLiftLocalTesting;
using AmazonGameLiftPlugin.Core.GameLiftLocalTesting.Models;
using AmazonGameLiftPlugin.Core.JavaCheck;
using AmazonGameLiftPlugin.Core.JavaCheck.Models;
using AmazonGameLiftPlugin.Core.SettingsManagement.Models;
using AmazonGameLiftPlugin.Core.Shared;
using AmazonGameLiftPlugin.Core.Shared.FileSystem;
using AmazonGameLiftPlugin.Core.Shared.FileZip;
using AmazonGameLiftPlugin.Core.Shared.ProcessManagement;
using AmazonGameLiftPlugin.Core.Shared.S3Bucket;
using System.Drawing;
using Amazon.ECR.Model;
using AmazonGameLiftPlugin.Core.ContainerManagement.Models;

namespace AmazonGameLift.Editor
{
    public class CoreApi
    {
        private const int MinJavaMajorVersion = 8;
        private static readonly ProcessWrapper s_process = new ProcessWrapper();

        public static CoreApi SharedInstance { get; } = new CoreApi();

        public CoreApi()
        {
            _settingsStore = new Settings<SettingsKeys>(Paths.PluginSettingsFile);
            Bootstrapper.Initialize();
        }        

        #region File system

        private readonly IFileWrapper _fileWrapper = new FileWrapper();

        public virtual bool FolderExists(string path)
        {
            return _fileWrapper.DirectoryExists(path);
        }

        public virtual string GetUniqueTempFilePath()
        {
            return _fileWrapper.GetUniqueTempFilePath();
        }

        public virtual bool FileExists(string path)
        {
            return _fileWrapper.FileExists(path);
        }

        public virtual void FileDelete(string path)
        {
            _fileWrapper.Delete(path);
        }

        public virtual FileReadAllTextResponse FileReadAllText(string path)
        {
            try
            {
                string text = _fileWrapper.ReadAllText(path);
                return Response.Ok(new FileReadAllTextResponse(text));
            }
            catch (Exception ex)
            {
                var response = new FileReadAllTextResponse()
                {
                    ErrorCode = ErrorCode.ReadingFileFailed,
                    ErrorMessage = ex.Message
                };
                return Response.Fail(response);
            }
        }

        public virtual Response FileWriteAllText(string path, string text)
        {
            try
            {
                _fileWrapper.WriteAllText(path, text);
                return Response.Ok(new Response());
            }
            catch (Exception ex)
            {
                var response = new Response()
                {
                    ErrorCode = ErrorCode.ReadingFileFailed,
                    ErrorMessage = ex.Message
                };
                return Response.Fail(response);
            }
        }

        #endregion

        #region Compression

        private readonly IFileZip _fileZip = new FileZip();

        public virtual void Zip(string sourceFolderPath, string targetFilePath)
        {
            _fileZip.Zip(sourceFolderPath, targetFilePath);
        }

        #endregion

        #region Credentials

        private readonly ICredentialsStore _credentialsStore = new CredentialsStore(new FileWrapper());

        public virtual GetProfilesResponse ListCredentialsProfiles()
        {
            var request = new GetProfilesRequest();
            return _credentialsStore.GetProfiles(request);
        }

        public virtual RetriveAwsCredentialsResponse RetrieveAwsCredentials(string profileName)
        {
            var request = new RetriveAwsCredentialsRequest() { ProfileName = profileName };
            return _credentialsStore.RetriveAwsCredentials(request);
        }

        public virtual SaveAwsCredentialsResponse SaveAwsCredentials(string profileName, string accessKey, string secretKey, string region)
        {
            var request = new SaveAwsCredentialsRequest()
            {
                ProfileName = profileName,
                AccessKey = accessKey,
                SecretKey = secretKey,
                Region = region,
            };
            return _credentialsStore.SaveAwsCredentials(request);
        }

        public virtual UpdateAwsCredentialsResponse UpdateAwsCredentials(string profileName, string accessKey, string secretKey)
        {
            var request = new UpdateAwsCredentialsRequest()
            {
                ProfileName = profileName,
                AccessKey = accessKey,
                SecretKey = secretKey
            };
            return _credentialsStore.UpdateAwsCredentials(request);
        }

        public virtual GetCredentialsFileResponse GetCredentialsFile()
        {
            var request = new GetCredentialsFileRequest();
            return _credentialsStore.GetCredentialsFile(request);
        }


        #endregion

        #region Settings

        private readonly Settings<SettingsKeys> _settingsStore;

        public virtual GetSettingResponse GetSetting(SettingsKeys key)
        {
            return _settingsStore.GetSetting(key);
        }

        public virtual PutSettingResponse PutSetting(SettingsKeys key, string value)
        {
            return _settingsStore.PutSetting(key, value);
        }

        public virtual ClearSettingResponse ClearSetting(SettingsKeys key)
        {
            return _settingsStore.ClearSetting(key);
        }

        public virtual Response PutSettingOrClear(SettingsKeys key, string value)
        {
            if (string.IsNullOrEmpty(value))
            {
                return _settingsStore.ClearSetting( key );
            }

            return _settingsStore.PutSetting(key, value);
        }

        #endregion

        #region S3

        public virtual IEnumerable<string> ListAvailableRegions()
        {
            return AwsRegionMapper.AvailableRegions();
        }

        public virtual bool IsValidRegion(string region)
        {
            return AwsRegionMapper.IsValidRegion(region);
        }

        public virtual RetrieveAccountIdByCredentialsResponse RetrieveAccountId(string profileName)
        {
            RetriveAwsCredentialsResponse credentialsResponse = RetrieveAwsCredentials(profileName);
            var request = new RetrieveAccountIdByCredentialsRequest()
            {
                AccessKey = credentialsResponse.AccessKey,
                SecretKey = credentialsResponse.SecretKey
            };
            var accountManager = new AccountManager(new AmazonSecurityTokenServiceClientWrapper());
            return accountManager.RetrieveAccountIdByCredentials(request);
        }

        /// <summary>
        /// Needs AWS credentials set up for <paramref name="profileName"/>.
        /// </summary>
        public virtual GetBucketsResponse ListBuckets(string profileName, string region)
        {
            var bucketStore = new BucketStore(CreateS3Wrapper(profileName, region));
            var request = new GetBucketsRequest()
            {
                Region = region
            };
            return bucketStore.GetBuckets(request);
        }

        /// <summary>
        /// Needs AWS credentials set up for <paramref name="profileName"/>.
        /// </summary>
        public virtual CreateBucketResponse CreateBucket(string profileName, string region, string bucketName)
        {
            var accountIdResponse = RetrieveAccountId(profileName);
            if (!accountIdResponse.Success)
            {
                return Response.Fail(new CreateBucketResponse()
                {
                    ErrorCode = accountIdResponse.ErrorCode,
                    ErrorMessage = accountIdResponse.ErrorMessage
                });
            }
            var bucketStore = new BucketStore(CreateS3Wrapper(profileName, region));
            var request = new CreateBucketRequest()
            {
                AccountId = accountIdResponse.AccountId,
                BucketName = bucketName,
                Region = region
            };
            return bucketStore.CreateBucket(request);
        }

        /// <summary>
        /// Needs AWS credentials set up for <paramref name="profileName"/>.
        /// </summary>
        public virtual PutLifecycleConfigurationResponse PutBucketLifecycleConfiguration(
            string profileName, string region, string bucketName, BucketPolicy expirationPolicy)
        {
            var bucketStore = new BucketStore(CreateS3Wrapper(profileName, region));
            var request = new PutLifecycleConfigurationRequest()
            {
                BucketName = bucketName,
                BucketPolicy = expirationPolicy
            };
            return bucketStore.PutLifecycleConfiguration(request);
        }

        internal virtual IAmazonS3Wrapper CreateS3Wrapper(string profileName, string region)
        {
            RetriveAwsCredentialsResponse response = RetrieveAwsCredentials(profileName);

            if (!response.Success)
            {
                return CreateDefaultS3Wrapper();
            }

            string accessKey = response.AccessKey;
            string secretKey = response.SecretKey;
            return new AmazonS3Wrapper(accessKey, secretKey, region);
        }

        private static IAmazonS3Wrapper CreateDefaultS3Wrapper()
        {
            return new AmazonS3Wrapper(string.Empty, string.Empty, string.Empty);
        }
        #endregion

        #region ECR

        public virtual ListECRImagesResponse ListECRImages(string profileName, string region, string repositoryName)
        {
            var ECRWrapper = CreateECRWrapper(profileName, region);
            return ECRWrapper.ListECRImages(repositoryName);
        }

        public virtual DescribeECRRepositoriesResponse DescribeECRRepositories(string profileName, string region, List<string> RepositoryNames = null)
        {
            var ECRWrapper = CreateECRWrapper(profileName, region);
            return ECRWrapper.DescribeECRRepositories(RepositoryNames);
        }

        public virtual CreateECRRepositoryResponse CreateRepository(string profileName, string region, string repositoryName)
        {
            var ECRWrapper = CreateECRWrapper(profileName, region);
            return ECRWrapper.CreateRepository(repositoryName);
        }

        internal virtual IAmazonECRWrapper CreateECRWrapper(string profileName, string region)
        {
            RetriveAwsCredentialsResponse response = RetrieveAwsCredentials(profileName);

            if (!response.Success)
            {
                return CreateDefaultECRWrapper();
            }

            string accessKey = response.AccessKey;
            string secretKey = response.SecretKey;
            return new AmazonECRWrapper(accessKey, secretKey, region);
        }

        private static IAmazonECRWrapper CreateDefaultECRWrapper()
        {
            return new AmazonECRWrapper(string.Empty, string.Empty, string.Empty);
        }
        #endregion

        #region Deployment

        private static readonly DeploymentFormatter s_deploymentFormatter = new DeploymentFormatter();

        public virtual string GetBuildS3Key()
        {
            return s_deploymentFormatter.GetBuildS3Key();
        }

        public virtual string GetStackName(string gameName)
        {
            return s_deploymentFormatter.GetStackName(gameName);
        }

        public virtual string GetStackNameContainers(string gameName)
        {
            return s_deploymentFormatter.GetStackNameContainers(gameName);
        }

        public virtual string GetServerGamePath(string gameFilePathInBuild, string operatingSystem)
        {
            return s_deploymentFormatter.GetServerGamePath(gameFilePathInBuild, operatingSystem);
        }

        /// <summary>
        /// Needs AWS credentials set up for <paramref name="profileName"/>.
        /// </summary>
        public virtual ValidateCfnTemplateResponse ValidateCfnTemplate(string profileName, string region, string templateFilePath)
        {
            IDeploymentManager deploymentManager = CreateDeploymentManager(profileName, region);
            var request = new ValidateCfnTemplateRequest()
            {
                TemplateFilePath = templateFilePath
            };
            return deploymentManager.ValidateCfnTemplate(request);
        }

        /// <summary>
        /// Needs AWS credentials set up for <paramref name="profileName"/>.
        /// </summary>
        public virtual StackExistsResponse StackExists(string profileName, string region, string stackName)
        {
            IDeploymentManager deploymentManager = CreateDeploymentManager(profileName, region);
            var request = new StackExistsRequest()
            {
                StackName = stackName
            };
            return deploymentManager.StackExists(request);
        }

        /// <summary>
        /// Needs AWS credentials set up for <paramref name="profileName"/>.
        /// </summary>
        public virtual CreateChangeSetResponse CreateChangeSet(string profileName, string region,
            string bucketName, string stackName, string templateFilePath, string parametersFilePath,
            string gameName, string lambdaFolderPath, string buildS3Key = null)
        {
            IDeploymentManager deploymentManager = CreateDeploymentManager(profileName, region);
            var request = new CreateChangeSetRequest()
            {
                StackName = stackName,
                TemplateFilePath = templateFilePath,
                ParametersFilePath = parametersFilePath,
                BootstrapBucketName = bucketName,
                GameName = gameName,
                LambdaSourcePath = lambdaFolderPath,
                BuildS3Key = buildS3Key,
            };
            return deploymentManager.CreateChangeSet(request);
        }

        /// <summary>
        /// Needs AWS credentials set up for <paramref name="profileName"/>.
        /// </summary>
        public virtual DescribeChangeSetResponse DescribeChangeSet(string profileName, string region,
            string stackName, string changeSetName)
        {
            IDeploymentManager deploymentManager = CreateDeploymentManager(profileName, region);
            var request = new DescribeChangeSetRequest()
            {
                StackName = stackName,
                ChangeSetName = changeSetName,
            };
            return deploymentManager.DescribeChangeSet(request);
        }

        /// <summary>
        /// Needs AWS credentials set up for <paramref name="profileName"/>.
        /// </summary>
        public virtual ExecuteChangeSetResponse ExecuteChangeSet(string profileName, string region, string stackName, string changeSetName)
        {
            IDeploymentManager deploymentManager = CreateDeploymentManager(profileName, region);
            var request = new ExecuteChangeSetRequest
            {
                StackName = stackName,
                ChangeSetName = changeSetName
            };
            return deploymentManager.ExecuteChangeSet(request);
        }

        /// <summary>
        /// Needs AWS credentials set up for <paramref name="profileName"/>.
        /// </summary>
        public virtual DeleteChangeSetResponse DeleteChangeSet(string profileName, string region,
            string stackName, string changeSetName)
        {
            IDeploymentManager deploymentManager = CreateDeploymentManager(profileName, region);
            var request = new DeleteChangeSetRequest()
            {
                StackName = stackName,
                ChangeSetName = changeSetName,
            };
            return deploymentManager.DeleteChangeSet(request);
        }

        /// <summary>
        /// Needs AWS credentials set up for <paramref name="profileName"/>.
        /// </summary>
        public virtual DescribeStackResponse DescribeStack(string profileName, string region, string stackName)
        {
            IDeploymentManager deploymentManager = CreateDeploymentManager(profileName, region);
            var request = new DescribeStackRequest
            {
                StackName = stackName,
                OutputKeys = new List<string>
                {
                    StackOutputKeys.ApiGatewayEndpoint,
                    StackOutputKeys.UserPoolClientId,
                }
            };
            return deploymentManager.DescribeStack(request);
        }

        public virtual DescribeStackResourceResponse DescribeStackResource(string profileName, string region, string stackName, string resourceIdentifier)
        {
            IDeploymentManager deploymentManager = CreateDeploymentManager(profileName, region);
            var request = new DescribeStackResourceRequest
            {
                StackName = stackName,
                LogicalResourceId = resourceIdentifier
                
            };
            return deploymentManager.DescribeStackResource(request);
        }

        /// <summary>
        /// Needs AWS credentials set up for <paramref name="profileName"/>.
        /// </summary>
        public virtual DeleteStackResponse DeleteStack(string profileName, string region, string stackName)
        {
            IDeploymentManager deploymentManager = CreateDeploymentManager(profileName, region);
            var request = new DeleteStackRequest
            {
                StackName = stackName
            };
            return deploymentManager.DeleteStack(request);
        }

        /// <summary>
        /// Needs AWS credentials set up for <paramref name="profileName"/>.
        /// </summary>
        public virtual UploadServerBuildResponse UploadServerBuild(string profileName, string region,
            string bucketName, string bucketKey, string filePath)
        {
            IDeploymentManager deploymentManager = CreateDeploymentManager(profileName, region);
            var request = new UploadServerBuildRequest()
            {
                BucketName = bucketName,
                BuildS3Key = bucketKey,
                FilePath = filePath,
            };
            return deploymentManager.UploadServerBuild(request);
        }

        /// <summary>
        /// Needs AWS credentials set up for <paramref name="profileName"/>. <paramref name="clientRequestToken"/> can be null.
        /// </summary>
        public virtual CancelDeploymentResponse CancelDeployment(string profileName, string region,
            string stackName, string clientRequestToken)
        {
            IDeploymentManager deploymentManager = CreateDeploymentManager(profileName, region);
            var request = new CancelDeploymentRequest()
            {
                StackName = stackName,
                ClientRequestToken = clientRequestToken,
            };
            return deploymentManager.CancelDeployment(request);
        }

        internal virtual IDeploymentManager CreateDeploymentManager(string profileName, string region)
        {
            RetriveAwsCredentialsResponse response = RetrieveAwsCredentials(profileName);
            string accessKey = response.Success ? response.AccessKey : string.Empty;
            string secretKey = response.Success ? response.SecretKey : string.Empty;
            var amazonS3Wrapper = new AmazonS3Wrapper(accessKey, secretKey, region);
            IAmazonCloudFormationWrapper cloudFormationWrapper = new AmazonCloudFormationWrapper(accessKey, secretKey, region);
            return new DeploymentManager(cloudFormationWrapper, amazonS3Wrapper, _fileWrapper, new FileZip());
        }

        #endregion

        #region JRE

        private readonly InstalledJavaVersionProvider _installedJavaVersionProvider =
            new InstalledJavaVersionProvider(s_process);

        public virtual bool CheckInstalledJavaVersion()
        {
            var request = new CheckInstalledJavaVersionRequest()
            {
                ExpectedMinimumJavaMajorVersion = MinJavaMajorVersion
            };
            CheckInstalledJavaVersionResponse response = _installedJavaVersionProvider.CheckInstalledJavaVersion(request);
            return response.IsInstalled;
        }

        #endregion

        #region Local Testing

        private readonly GameLiftProcess _gameLiftProcess = new GameLiftProcess(s_process);

        public virtual StartResponse StartGameLiftLocal(string gameLiftLocalFilePath, int port,
                LocalOperatingSystem localOperatingSystem = LocalOperatingSystem.WINDOWS)
        {
            var request = new StartRequest()
            {
                GameLiftLocalFilePath = gameLiftLocalFilePath,
                Port = port,
                LocalOperatingSystem = localOperatingSystem
            };
            return _gameLiftProcess.Start(request);
        }

        public virtual StopResponse StopProcess(int processId,
                LocalOperatingSystem localOperatingSystem = LocalOperatingSystem.WINDOWS)
        {
            var request = new StopRequest()
            {
                ProcessId = processId,
                LocalOperatingSystem = localOperatingSystem
            };
            return _gameLiftProcess.Stop(request);
        }

        public virtual RunLocalServerResponse RunLocalServer(string exeFilePath, string applicationProductName,
                LocalOperatingSystem localOperatingSystem = LocalOperatingSystem.WINDOWS)
        {
            var request = new RunLocalServerRequest()
            {
                FilePath = exeFilePath,
                ShowWindow = true,
                ApplicationProductName = applicationProductName,
                LocalOperatingSystem = localOperatingSystem
            };
            return _gameLiftProcess.RunLocalServer(request);
        }

        #endregion
    }
}