@aws-amplify/storage

const Storage: StorageClass;

type CommonStorageOptions = Omit<
    StorageOptions,
    | 'credentials'
    | 'region'
    | 'bucket'
    | 'dangerouslyConnectToHttpEndpointForTesting'
>;

type CustomPrefix = {
    [key in StorageAccessLevel]?: string;
};

type PutResult = {
    key: string;
};

type ResumableUploadConfig = {
    resumable: true;
    progressCallback?: (progress: UploadTaskProgressEvent) => any;
    completeCallback?: (event: UploadTaskCompleteEvent) => any;
    errorCallback?: (err: any) => any;
};

type S3CopyDestination = Omit<S3CopyTarget, 'identityId'>;

type S3CopySource = S3CopyTarget;

type S3ProviderCopyConfig = Omit<CommonStorageOptions, 'level'> & {
    provider?: 'AWSS3';
    bucket?: CopyObjectRequest['Bucket'];
    cacheControl?: CopyObjectRequest['CacheControl'];
    contentDisposition?: CopyObjectRequest['ContentDisposition'];
    contentLanguage?: CopyObjectRequest['ContentLanguage'];
    contentType?: CopyObjectRequest['ContentType'];
    expires?: CopyObjectRequest['Expires'];
    tagging?: CopyObjectRequest['Tagging'];
    acl?: CopyObjectRequest['ACL'];
    metadata?: CopyObjectRequest['Metadata'];
    serverSideEncryption?: CopyObjectRequest['ServerSideEncryption'];
    SSECustomerAlgorithm?: CopyObjectRequest['SSECustomerAlgorithm'];
    SSECustomerKey?: CopyObjectRequest['SSECustomerKey'];
    SSECustomerKeyMD5?: CopyObjectRequest['SSECustomerKeyMD5'];
    SSEKMSKeyId?: CopyObjectRequest['SSEKMSKeyId'];
};

type S3ProviderCopyOutput = {
    key: string;
};

type S3ProviderGetConfig = CommonStorageOptions & {
    download?: boolean;
    track?: boolean;
    expires?: number;
    provider?: 'AWSS3';
    identityId?: string;
    progressCallback?: (progress: any) => any;
    cacheControl?: GetObjectRequest['ResponseCacheControl'];
    contentDisposition?: GetObjectRequest['ResponseContentDisposition'];
    contentEncoding?: GetObjectRequest['ResponseContentEncoding'];
    contentLanguage?: GetObjectRequest['ResponseContentLanguage'];
    contentType?: GetObjectRequest['ResponseContentType'];
    SSECustomerAlgorithm?: GetObjectRequest['SSECustomerAlgorithm'];
    SSECustomerKey?: GetObjectRequest['SSECustomerKey'];
    SSECustomerKeyMD5?: GetObjectRequest['SSECustomerKeyMD5'];
};

type S3ProviderGetOuput<T> = T extends {
    download: true;
}
    ? GetObjectCommandOutput
    : string;

type S3ProviderListConfig = CommonStorageOptions & {
    bucket?: string;
    maxKeys?: number;
    provider?: 'AWSS3';
    identityId?: string;
};

type S3ProviderListOutput = S3ProviderListOutputItem[];

type S3ProviderPutConfig = CommonStorageOptions &
    (_S3ProviderPutConfig | (_S3ProviderPutConfig & ResumableUploadConfig));

type S3ProviderPutOutput<T> = T extends {
    resumable: true;
}
    ? UploadTask
    : Promise<PutResult>;

type S3ProviderRemoveConfig = CommonStorageOptions & {
    bucket?: string;
    provider?: 'AWSS3';
};

type S3ProviderRemoveOutput = DeleteObjectCommandOutput;

type StorageAccessLevel = 'public' | 'protected' | 'private';

type StorageCopyConfig<T> = T extends StorageProviderWithCopy
    ? StorageOperationConfig<T, 'copy'>
    : StorageOperationConfigMap<StorageOperationConfig<AWSS3Provider, 'copy'>, T>;

type StorageCopyDestination = Omit<StorageCopyTarget, 'identityId'>;

type StorageCopyOutput<T> = PickProviderOutput<
    Promise<S3ProviderCopyOutput>,
    T,
    'copy'
>;

type StorageCopySource = StorageCopyTarget;

type StorageCopyTarget = {
    key: string;
    level?: string;
    identityId?: string;
};

type StorageGetConfig<T> = T extends StorageProvider
    ? StorageOperationConfig<T, 'get'>
    : StorageOperationConfigMap<StorageOperationConfig<AWSS3Provider, 'get'>, T>;

type StorageGetOutput<T extends StorageProvider | Record<string, any>> =
    PickProviderOutput<Promise<S3ProviderGetOuput<T>>, T, 'get'>;

type StorageListConfig<T> = T extends StorageProvider
    ? StorageOperationConfig<T, 'list'>
    : StorageOperationConfigMap<StorageOperationConfig<AWSS3Provider, 'list'>, T>;

type StorageListOutput<T> = PickProviderOutput<
    Promise<S3ProviderListOutput>,
    T,
    'list'
>;

type StorageOperationConfigMap<Default, T extends Record<string, any>> = T extends {
    provider: string;
}
    ? T extends {
          provider: 'AWSS3';
      }
        ? Default
        : T & {
              provider: string;
          }
    : Default;

type StorageProviderApi = 'copy' | 'get' | 'put' | 'remove' | 'list';

type StoragePutConfig<T> = T extends StorageProvider
    ? StorageOperationConfig<T, 'put'>
    : StorageOperationConfigMap<StorageOperationConfig<AWSS3Provider, 'put'>, T>;

type StoragePutOutput<T> = PickProviderOutput<S3ProviderPutOutput<T>, T, 'put'>;

type StorageRemoveConfig<T> = T extends StorageProvider
    ? StorageOperationConfig<T, 'remove'>
    : StorageOperationConfigMap<StorageOperationConfig<AWSS3Provider, 'remove'>, T>;

type StorageRemoveOutput<T> = PickProviderOutput<
    Promise<S3ProviderRemoveOutput>,
    T,
    'remove'
>;