/*
 * MinIO Java SDK for Amazon S3 Compatible Cloud Storage,
 * (C) 2015-2021 MinIO, Inc.
 *
 * 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.
 */

import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
import io.minio.admin.*;
import java.util.Map;
import org.junit.Assert;

@SuppressFBWarnings(
 value = "REC",
 justification = "Allow catching super class Exception since it's tests")
public class TestMinioAdminClient {

 private final MinioAdminClient adminClient;
 private final boolean mintEnv;

 private static String userAccessKey = FunctionalTest.getRandomName();
 private static String userSecretKey = FunctionalTest.getRandomName();
 private static String policyName = FunctionalTest.getRandomName();

 public TestMinioAdminClient(MinioAdminClient adminClient, boolean mintEnv) {
 this.adminClient = adminClient;
 this.mintEnv = mintEnv;
 }

 public void addUser() throws Exception {
 String methodName = "addUser()";
 if (!mintEnv) {
 System.out.println(methodName);
 }
 long startTime = System.currentTimeMillis();
 try {
 adminClient.addUser(userAccessKey, UserInfo.Status.ENABLED, userSecretKey, null, null);
 } catch (Exception e) {
 FunctionalTest.handleException(methodName, null, startTime, e);
 }
 }

 public void addCannedPolicy() throws Exception {
 String methodName = "addCannedPolicy()";
 if (!mintEnv) {
 System.out.println(methodName);
 }
 long startTime = System.currentTimeMillis();
 try {
 String policyJson =
 "{'Version': '2012-10-17','Statement': [{'Action': ['s3:GetObject'],'Effect':"
 + " 'Allow','Resource': ['arn:aws:s3:::my-bucketname/*'],'Sid': ''}]}";
 adminClient.addCannedPolicy(policyName, policyJson.replaceAll("'", "\""));
 } catch (Exception e) {
 FunctionalTest.handleException(methodName, null, startTime, e);
 }
 }

 public void listCannedPolicies() throws Exception {
 String methodName = "listCannedPolicies()";
 if (!mintEnv) {
 System.out.println(methodName);
 }

 long startTime = System.currentTimeMillis();
 try {
 Map<String, String> policies = adminClient.listCannedPolicies();
 String policy = policies.get(policyName);
 Assert.assertTrue(policy != null);
 } catch (Exception e) {
 FunctionalTest.handleException(methodName, null, startTime, e);
 }
 }

 public void removeCannedPolicy() throws Exception {
 String methodName = "removeCannedPolicy()";
 if (!mintEnv) {
 System.out.println(methodName);
 }
 long startTime = System.currentTimeMillis();
 try {
 adminClient.removeCannedPolicy(policyName);
 } catch (Exception e) {
 FunctionalTest.handleException(methodName, null, startTime, e);
 }
 }

 public void setPolicy() throws Exception {
 String methodName = "setPolicy()";
 if (!mintEnv) {
 System.out.println(methodName);
 }

 long startTime = System.currentTimeMillis();
 try {
 adminClient.setPolicy(userAccessKey, false, policyName);
 } catch (Exception e) {
 FunctionalTest.handleException(methodName, null, startTime, e);
 }
 }

 public void getUserInfo() throws Exception {
 String methodName = "getUserInfo()";
 if (!mintEnv) {
 System.out.println(methodName);
 }

 long startTime = System.currentTimeMillis();
 try {
 UserInfo userInfo = adminClient.getUserInfo(userAccessKey);
 Assert.assertEquals(userInfo.status(), UserInfo.Status.ENABLED);
 Assert.assertEquals(userInfo.policyName(), policyName);
 } catch (Exception e) {
 FunctionalTest.handleException(methodName, null, startTime, e);
 }
 }

 public void listUsers() throws Exception {
 String methodName = "listUsers()";
 if (!mintEnv) {
 System.out.println(methodName);
 }

 long startTime = System.currentTimeMillis();
 try {
 Map<String, UserInfo> users = adminClient.listUsers();
 Assert.assertTrue(users.containsKey(userAccessKey));
 Assert.assertEquals(users.get(userAccessKey).status(), UserInfo.Status.ENABLED);
 Assert.assertEquals(users.get(userAccessKey).policyName(), policyName);
 } catch (Exception e) {
 FunctionalTest.handleException(methodName, null, startTime, e);
 }
 }

 public void deleteUser() throws Exception {
 String methodName = "deleteUser()";
 if (!mintEnv) {
 System.out.println(methodName);
 }

 long startTime = System.currentTimeMillis();
 try {
 adminClient.deleteUser(userAccessKey);
 } catch (Exception e) {
 FunctionalTest.handleException(methodName, null, startTime, e);
 }
 }

 public void runAdminTests() throws Exception {
 addUser();
 addCannedPolicy();
 setPolicy();
 getUserInfo();
 listUsers();
 listCannedPolicies();
 deleteUser();
 removeCannedPolicy();
 }
}