/**
    * Copyright 2004-2014 The Kuali Foundation
    *
    * Licensed under the Educational Community 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.opensource.org/licenses/ecl2.php
    *
   * 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.
   */
  package org.kuali.common.aws.s3;
  
  import static com.amazonaws.services.s3.Headers.CONTENT_TYPE;
  import static com.google.common.base.Stopwatch.createStarted;
  import static com.google.common.collect.Lists.newArrayList;
  import static com.google.common.collect.Maps.newHashMap;
  import static com.google.common.collect.Ordering.natural;
  import static java.lang.String.format;
  import static java.util.Arrays.asList;
  import static org.kuali.common.core.io.Files.deleteCreateFile;
  import static org.kuali.common.core.io.Files.getCanonicalFile;
  import static org.kuali.common.util.FormatUtils.getTime;
  import static org.kuali.common.util.log.LoggerUtils.logTable;
  import static org.kuali.common.util.log.Loggers.newLogger;
  
  import java.io.File;
  import java.util.List;
  import java.util.Map;
  
  import org.junit.Ignore;
  import org.junit.Test;
  import org.kuali.common.aws.model.ImmutableAWSCredentials;
  import org.kuali.common.aws.s3.model.ObjectMetadata;
  import org.kuali.common.aws.s3.model.ObjectSummary;
  import org.kuali.common.aws.s3.model.PutFileRequest;
  import org.kuali.common.core.base.TimedInterval;
  import org.kuali.common.util.encrypt.Encryption;
  import org.kuali.common.util.encrypt.Encryptor;
  import org.slf4j.Logger;
  
  import com.amazonaws.auth.AWSCredentials;
  import com.google.common.base.Stopwatch;
  
  public class DefaultS3ServiceTest {
  
  	private static final Logger logger = newLogger();
  	private final DefaultS3Service service = DefaultS3Service.build(getCredentials());
  
  	@Test
  	public void testYo() {
  		try {
  			String bucket = "archive.kuali.org";
  			String key = "foo/bar2/";
  			String prefix = "foo";
  			List<ObjectSummary> summaries = service.getCompleteList(bucket, prefix);
  			for (ObjectSummary summary : summaries) {
  				logger.info(summary.getKey());
  			}
  			ObjectMetadata meta = service.getMetadata(bucket, key);
  			show(meta.getRawMetadata());
  			show(meta.getUserMetadata());
  			Map<String, String> user = newHashMap(meta.getUserMetadata());
  			Map<String, Object> aws = newHashMap();
  			aws.put(CONTENT_TYPE, "application/x-directory");
  			ObjectMetadata omd = ObjectMetadata.builder().withUserMetadata(user).withRawMetadata(aws).build();
  			File scratch = deleteCreateFile(getCanonicalFile("./target/s3/foo.txt"), "");
  			String fooDirKey = "foo/bar2/";
  			PutFileRequest request = PutFileRequest.builder().withBucket(bucket).withKey(fooDirKey).withFile(scratch).withMetadata(omd).build();
  			service.putFile(request);
  		} catch (Throwable e) {
  			e.printStackTrace();
  		}
  	}
  
  	@Test
  	public void testS3FuseDir() {
  		try {
  			String bucket = "archive.kuali.org";
  			String key = "foo/bar/";
  			String prefix = "foo/bar";
  			List<ObjectSummary> summaries = service.getCompleteList(bucket, prefix);
  			for (ObjectSummary summary : summaries) {
  				logger.info(summary.getKey());
  			}
  			ObjectMetadata meta = service.getMetadata(bucket, key);
  			show(meta.getRawMetadata());
  			show(meta.getUserMetadata());
  			Map<String, String> user = newHashMap(meta.getUserMetadata());
  			Map<String, Object> aws = newHashMap();
  			aws.put(CONTENT_TYPE, "application/x-directory");
  			ObjectMetadata omd = ObjectMetadata.builder().withUserMetadata(user).withRawMetadata(aws).build();
  			File scratch = deleteCreateFile(getCanonicalFile("./target/s3/foo.txt"), "");
  			String fooDirKey = "foo/bar2/";
  			PutFileRequest request = PutFileRequest.builder().withBucket(bucket).withKey(fooDirKey).withFile(scratch).withMetadata(omd).build();
 			service.putFile(request);
 		} catch (Throwable e) {
 			e.printStackTrace();
 		}
 	}
 
 	@Test
 	@Ignore
 	public void testS3FuseFile() {
 		try {
 			String bucket = "archive.kuali.org";
 			String key = "org/kuali/testci/home/";
 			ObjectMetadata meta = service.getMetadata(bucket, key);
 			// show(meta.getAmazonMetadata());
 			show(meta.getUserMetadata());
 			System.out.format("s3://%s/%s", bucket, key);
 			String fooKey = "org/kuali/testci/home/tomcat7/.jenkins/jobs/kdo-show-environment/builds/2014-08-20_19-33-08/foo.txt";
 			File scratch = deleteCreateFile(getCanonicalFile("./target/s3/foo.txt"), "bar");
 			Map<String, String> map = newHashMap(meta.getUserMetadata());
 			map.put("mtime", scratch.lastModified() + "");
 			ObjectMetadata omd = ObjectMetadata.builder().withUserMetadata(map).build();
 			PutFileRequest request = PutFileRequest.builder().withBucket(bucket).withKey(fooKey).withFile(scratch).withMetadata(omd).build();
 			service.putFile(request);
 		} catch (Throwable e) {
 			e.printStackTrace();
 		}
 	}
 
 	private void show(Map<String, ?> map) {
 		List<String> columns = asList("key", "value");
 		List<Object[]> rows = newArrayList();
 		List<String> keys = natural().immutableSortedCopy(map.keySet());
 		for (String key : keys) {
 			Object[] row = { key, map.get(key) };
 			rows.add(row);
 		}
 		logTable(columns, rows);
 	}
 
 	@Test
 	@Ignore
 	public void testObjectListing() {
 		try {
 			String bucket = "archive.kuali.org";
 			String prefix = "org/kuali/testci/home/tomcat7/.jenkins/jobs";
 			Stopwatch sw = createStarted();
 			List<ObjectSummary> summaries = service.getCompleteList(bucket, prefix);
 			TimedInterval timing = TimedInterval.build(sw);
 			info("summaries -> %s", summaries.size());
 			info("elapsed   -> %s", getTime(timing.getElapsed()));
 		} catch (Throwable e) {
 			e.printStackTrace();
 		}
 	}
 
 	@Test
 	@Ignore
 	public void test() {
 		try {
 			info("hello world");
 			service.writeStringToObject("site.origin.kuali.org", "test/foo.txt", "bar");
 		} catch (Throwable e) {
 			e.printStackTrace();
 		}
 	}
 
 	private static final void info(String msg, Object... args) {
 		logger.info(args == null ? msg : format(msg, args));
 	}
 
 	private static AWSCredentials getCredentials() {
 		Encryptor encryptor = Encryption.getDefaultEncryptor();
 		String accessKey = encryptor.decrypt("U2FsdGVkX18ZhU341cfBkTGC+rNi22YhGUDMtBlB/PNaV94dH4RpLwAPl8UCzDgL");
 		String secretKey = encryptor.decrypt("U2FsdGVkX1/K6kgTQIYZHUd0NQpbTIzel8Q0ypnNmIBKRzIP4zR86HQk6IEd0Xv0Y7XtIpgSkRWuw6qtVIgHiw==");
 		return new ImmutableAWSCredentials(accessKey, secretKey);
 	}
 }