de.zalando.spring.cloud.config.aws.kms.test.EncryptionCLI.java Source code

Introduction

Here is the source code for de.zalando.spring.cloud.config.aws.kms.test.EncryptionCLI.java

Source

/**
 * Copyright (C) 2015 Zalando SE (https://tech.zalando.com)
 *
 * 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.
 */
package de.zalando.spring.cloud.config.aws.kms.test;

import static com.amazonaws.regions.Regions.fromName;

import static com.google.common.base.Preconditions.checkArgument;

import org.springframework.boot.CommandLineRunner;

import org.springframework.stereotype.Component;

import com.amazonaws.services.kms.AWSKMSClient;

import de.zalando.spring.cloud.config.aws.kms.KmsTextEncryptor;

@Component
public class EncryptionCLI implements CommandLineRunner {

    @Override
    public void run(final String... args) {
        try {
            checkArgument(args.length >= 3, "Too few arguments.");

            final String text = args[1];
            final AWSKMSClient kms = new AWSKMSClient();
            kms.setRegion(fromName(args[2]));

            switch (args[0]) {

            case "encrypt":
                checkArgument(args.length == 4, "Too few arguments.");
                System.out.println(new KmsTextEncryptor(kms, args[3]).encrypt(text));
                break;

            case "decrypt":
                System.out.println(new KmsTextEncryptor(kms, null).decrypt(text));
                break;

            default:

                break;
            }
        } catch (final IllegalArgumentException e) {
            System.out.println(e.getMessage() + " Usage:\n" //
                    + "./run.sh encrypt 'plaintext' eu-west-1 ${kmsKeyId}\n" //
                    + "./run.sh decrypt 'base64cipherText' eu-west-1");
        }
    }
}