*  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *      http://www.apache.org/licenses/LICENSE-2.0
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *  See the NOTICE file distributed with this work for additional
 *  information regarding copyright ownership.

package org.seaborne.delta.server.s3;

import static org.seaborne.delta.server.s3.S3Const.pBucketName;
import static org.seaborne.delta.server.s3.S3Const.pPrefix;
import static org.seaborne.delta.server.s3.S3Const.pRegion;

import com.amazonaws.services.s3.AmazonS3;
import com.amazonaws.util.StringUtils;

import org.apache.curator.framework.CuratorFramework;
import org.apache.jena.atlas.logging.FmtLog;
import org.seaborne.delta.Delta;
import org.seaborne.delta.server.Provider;
import org.seaborne.delta.server.local.DPS;
import org.seaborne.delta.server.local.LocalServerConfig;
import org.seaborne.delta.server.local.PatchStore;
import org.seaborne.delta.server.local.patchstores.zk.PatchStoreProviderZk;

public class PatchStoreProviderZkS3 extends PatchStoreProviderZk {

    private static String DEFAULT_PREFIX = "patches/";

    static class DetailsS3 {
        final String prefix;
        final String bucketName;  //DNS name,lower case.
        final AmazonS3 client;

        public DetailsS3(String bucketName, String prefix, AmazonS3 client) {
            this.prefix = prefix;
            this.bucketName = bucketName;
            this.client = client;

    public PatchStoreProviderZkS3() {}

    public PatchStore create(LocalServerConfig config) {
        // Early create.
        DetailsS3 s3 = accessS3(config);
        if ( ! s3.client.doesBucketExistV2(s3.bucketName) ) {
            FmtLog.info(Delta.DELTA_LOG, "Creating bucket %s", s3.bucketName);
        // The usual PatchStoreZk for the index, but remembering DetailsS3
        CuratorFramework client = curator(config);
        return new PatchStoreZkS3(client, this, s3);

    private static DetailsS3 accessS3(LocalServerConfig configuration) {
        // Access and checking.
        String bucketName = configuration.getProperty(pBucketName);
        if ( StringUtils.isNullOrEmpty(bucketName) )
            throw new IllegalArgumentException("Missing required property: "+pBucketName);

        String region = configuration.getProperty(pRegion);
        if ( StringUtils.isNullOrEmpty(region) )
            throw new IllegalArgumentException("Missing required property: "+pRegion);

        String prefixStr = configuration.getProperty(pPrefix);
        if ( StringUtils.isNullOrEmpty(prefixStr) )
            prefixStr = DEFAULT_PREFIX;
        String prefix = (prefixStr!=null) ? prefixStr : DEFAULT_PREFIX;
        AmazonS3 client = S3.buildS3(configuration);
        return new DetailsS3(bucketName, prefix, client);
        //return access.computeIfAbsent(bucketName, n->new DetailsS3(bucketName, prefix, client));

    public Provider getType() { return Provider.ZKS3; }

    /** Short name used in server configuration files to set the default provider via "log_type" */
    public String getShortName() {
        return DPS.pspZkS3;