import com.google.gson.Gson;
import javax.crypto.Cipher;
import javax.crypto.spec.SecretKeySpec;
import java.util.Base64;

public class EncryptionExample {
  
  private static final String AES_KEY = "1234567890123456";
  
  public static void main(String[] args) {
    // 암호화할 데이터 객체 생성
    MyData data = new MyData("John", "Doe", "john.doe@example.com");
    
    // Gson을 사용하여 객체를 JSON 문자열로 직렬화
    Gson gson = new Gson();
    String json = gson.toJson(data);
    
    // JSON 문자열을 AES 알고리즘을 사용해 암호화
    String encryptedJson = encrypt(json, AES_KEY);
    
    System.out.println("암호화된 JSON 데이터: " + encryptedJson);
    
    // 필요한 경우, 암호화된 JSON 데이터를 복호화하여 Java 객체로 역직렬화
    String decryptedJson = decrypt(encryptedJson, AES_KEY);
    MyData decryptedData = gson.fromJson(decryptedJson, MyData.class);
    
    System.out.println("복호화된 데이터: " + decryptedData);
  }
  
  private static String encrypt(String input, String key) {
    try {
      Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5Padding");
      SecretKeySpec secretKey = new SecretKeySpec(key.getBytes(), "AES");
      cipher.init(Cipher.ENCRYPT_MODE, secretKey);
      byte[] encryptedBytes = cipher.doFinal(input.getBytes());
      return Base64.getEncoder().encodeToString(encryptedBytes);
    } catch (Exception ex) {
      ex.printStackTrace();
    }
    return null;
  }
  
  private static String decrypt(String input, String key) {
    try {
      Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5Padding");
      SecretKeySpec secretKey = new SecretKeySpec(key.getBytes(), "AES");
      cipher.init(Cipher.DECRYPT_MODE, secretKey);
      byte[] decryptedBytes = cipher.doFinal(Base64.getDecoder().decode(input));
      return new String(decryptedBytes);
    } catch (Exception ex) {
      ex.printStackTrace();
    }
    return null;
  }
  
  private static class MyData {
    private String firstName;
    private String lastName;
    private String email;
    
    public MyData(String firstName, String lastName, String email) {
      this.firstName = firstName;
      this.lastName = lastName;
      this.email = email;
    }
    
    // Getters and setters
  }
}