package br.com.rsa;

import java.security.*;
import java.io.*;
import java.util.*;
import javax.crypto.*;

public class Geracao {

    public static void main(String[] args) throws IOException {

        //Gerando um arquivo que será encriptado e descriptografado.
        Scanner entrada1 = new Scanner(System.in);

        System.out.println("Digite qualquer coisa: ");
        String entrada = entrada1.nextLine();

        System.out.println("Arquivo criado.");

        FileOutputStream saida = new FileOutputStream("arquivo.txt");
        PrintStream volta_saida = System.out;
        PrintStream imprimir = new PrintStream(saida);
        System.setOut(imprimir);
        System.out.println(entrada);
        saida.close();
        System.setOut(volta_saida);

        //Gerando as chaves publica e privada.
        try {
        KeyPairGenerator chave = KeyPairGenerator.getInstance("RSA");
        chave.initialize(1024);

        KeyPair chaves = chave.generateKeyPair();

        PrivateKey privada = chaves.getPrivate();
        PublicKey publica = chaves.getPublic();

        Base64.Encoder cripto = Base64.getEncoder();

        FileOutputStream saida_chaves = new FileOutputStream("chaves.txt");
        PrintStream volta_saida_c = System.out;
        PrintStream imprimir_chaves = new PrintStream(saida_chaves);
        System.setOut(imprimir_chaves);
        System.out.println("Chave privada: " + cripto.encodeToString(privada.getEncoded()));
        System.out.println("");
        System.out.println("Chave publica: " + cripto.encodeToString(publica.getEncoded()));
        System.out.println("");
        saida.close();
        System.setOut(volta_saida_c);

            //Salvando as chaves publica e privada.
            try (FileOutputStream s_prv = new FileOutputStream("privada" + ".key")){

                s_prv.write(chaves.getPrivate().getEncoded());

            }

            try (FileOutputStream s_pub = new FileOutputStream("publica" + ".key")){

                s_pub.write(chaves.getPublic().getEncoded());
            }

            Criptografar(chaves, null);
            Descriptografar(chaves, null);
        }

        //Qualquer erro dentro da geração das chaves
        catch (Exception e){

            System.out.println(e);
        }
    }

    //TODO - Comentario
     static private void processFile(Cipher cifra, InputStream entrada_arq_c, OutputStream saida_arq_c){
           try {
            byte[] ibuf = new byte[1024];
            int len;
            while ((len = entrada_arq_c.read(ibuf)) != -1) {
                byte[] obuf = cifra.update(ibuf, 0, len);
                if ( obuf != null ) saida_arq_c.write(obuf);
            }
            byte[] obuf = cifra.doFinal();
            if ( obuf != null ) saida_arq_c.write(obuf);
           }
          catch(Exception e) {
              e.printStackTrace();
              System.out.println("Problema no manuseio do arquivo.");
          }
    }

    //Metodo para criptografar.
     static private void Criptografar(KeyPair chaves, Cipher ci){
        try {
            PublicKey publica = chaves.getPublic();
            Cipher cifra = Cipher.getInstance("RSA");
            cifra.init(Cipher.ENCRYPT_MODE, publica);

            FileInputStream entrada_arq_c = new FileInputStream("arquivo.txt");
            FileOutputStream saida_arq_c = new FileOutputStream("criptografado.txt");
            processFile(cifra, entrada_arq_c, saida_arq_c);

        }
        catch(Exception e){

            System.out.println("Erro ao criptografar.");
        }
    }

    //Metodo para descriptografar.
     static private void Descriptografar(KeyPair chaves, Cipher ci){

        try {
            PrivateKey privada = chaves.getPrivate();
            Cipher cifra = Cipher.getInstance("RSA");
            cifra.init(Cipher.DECRYPT_MODE, privada);

            FileInputStream entrada_arq_c = new FileInputStream("criptografado.txt");
            FileOutputStream saida_arq_c = new FileOutputStream("descriptografado.txt");
            processFile(cifra, entrada_arq_c, saida_arq_c);
        }
        catch(Exception e){

            System.out.println("Erro ao descriptografar.");
        }

    }
}