package org.kuali.common.devops.archive.test;
   
   import static com.google.common.base.Optional.absent;
   import static com.google.common.collect.Sets.newHashSet;
   import static java.lang.String.format;
   import static org.apache.commons.io.FileUtils.forceDelete;
   import static org.apache.commons.io.FileUtils.write;
   import static org.apache.commons.lang3.StringUtils.repeat;
   import static org.kuali.common.core.io.Files.getCanonicalFile;
  import static org.kuali.common.util.FormatUtils.getSize;
  import static org.kuali.common.util.FormatUtils.getTime;
  import static org.kuali.common.util.base.Exceptions.illegalState;
  import static org.kuali.common.util.encrypt.Encryption.getDefaultEncryptor;
  import static org.kuali.common.util.log.Loggers.newLogger;
  
  import java.io.File;
  import java.io.IOException;
  import java.util.Random;
  import java.util.Set;
  
  import org.junit.Test;
  import org.kuali.common.aws.model.ImmutableAWSCredentials;
  import org.kuali.common.aws.s3.DefaultS3Service;
  import org.kuali.common.aws.s3.S3Service;
  import org.kuali.common.devops.archive.ArchiveFilesRequest;
  import org.kuali.common.devops.archive.ArchiveFilesResult;
  import org.kuali.common.devops.archive.ArchivedFile;
  import org.kuali.common.devops.archive.FileArchiveService;
  import org.kuali.common.devops.archive.s3.S3FileArchiveService;
  import org.kuali.common.devops.archive.s3.S3FileArchiveServiceContext;
  import org.kuali.common.util.encrypt.Encryptor;
  import org.slf4j.Logger;
  
  import com.amazonaws.auth.AWSCredentials;
  import com.google.common.base.Optional;
  
  public class S3FileArchiveServiceTest {
  
  	private static final Logger logger = newLogger();
  
  	@Test
  	public void test() {
  		try {
  			String hostname = "test.kdo.kuali.org";
  			String bucket = "archive.kuali.org";
  			// Optional<String> qualifier = Optional.of("jcaddel");
  			Optional<String> qualifier = absent();
  			S3Service s3 = DefaultS3Service.build(getFoundation());
  			S3FileArchiveServiceContext context = S3FileArchiveServiceContext.builder().noHostname().withBucket(bucket).withThreads(5).build();
  			FileArchiveService service = S3FileArchiveService.builder().withS3(s3).withContext(context).build();
  			logger.info(format("region: %s", s3.getRegion()));
  			logger.info(format("hostname: %s", hostname));
  			Set<File> files = createFiles();
  			ArchiveFilesRequest request = ArchiveFilesRequest.builder().withFiles(files).withHostname(hostname).withQualifier(qualifier).build();
  			ArchiveFilesResult result = service.archive(request);
  			long elapsed = result.getTiming().getElapsed();
  			long bytes = 0;
  			for (ArchivedFile archive : result.getArchives()) {
  				bytes += archive.getSize();
  			}
  			// for (ArchivedFile archive : result.getArchives()) {
  			// logger.info(format(" source url -> %s", archive.getSourceUrl()));
  			// logger.info(format("archive url -> %s", archive.getArchiveUrl()));
  			// }
  			logger.info(format("amount: %s  elapsed: %s", getSize(bytes), getTime(elapsed)));
  		} catch (Throwable e) {
  			e.printStackTrace();
  		}
  	}
  
  	private Set<File> createFiles() {
  		Random random = new Random();
  		Set<File> files = newHashSet();
  		for (int i = 0; i < 25; i++) {
  			int repeat = random.nextInt(5000) + 25000;
  			String content = repeat("hello world " + i + "\n", repeat);
  			String path = "./target/s3/archive/file" + i + ".txt";
  			files.add(deleteCreate(path, content));
  		}
  		return files;
  	}
  
  	private File deleteCreate(String path, String data) {
  		try {
  			File file = getCanonicalFile(path);
  			if (file.exists()) {
  				forceDelete(file);
  			}
  			// logger.info(format("creating -> %s", file));
  			write(file, data);
  			return file;
  		} catch (IOException e) {
  			throw illegalState(e);
  		}
  	}
  
  	private static AWSCredentials getFoundation() {
  		Encryptor enc = getDefaultEncryptor();
  		String accessKey = enc.decrypt("U2FsdGVkX19A2e6dN/ipVfb/9n0DROCPIrLK6H8PvvPmt0h6cBqccGaJW0NSoX3S");
 		String secretKey = enc.decrypt("U2FsdGVkX19Y9SZ5GAU82/X5Z0xZdeQf7DFuVDW07R9lfyHK4VaOj5R7pviRBKmIyn7jrVT2lv8Edeu7098k1A==");
 		return new ImmutableAWSCredentials(accessKey, secretKey);
 	}
 }