vignere cipher cpp

// program to implement ceasar cipher encryption

#include <iostream>
#include <vector>
#include <algorithm>
#include <fstream>
#include <sstream>

using namespace std;

// method that recives string as input and returns encrypted string as output
string encrypt(string original_msg, string key){

    // allchars is our keyspace
    string allChars = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
    int all_chars_length = allChars.length();

    // converting our string to uppercase
    transform(original_msg.begin(), original_msg.end(), original_msg.begin(), ::toupper);

    // removing all numbers, punctuations
    original_msg.erase(remove_if(original_msg.begin(), original_msg.end(), [](char c) { return (!isalpha(c) && c!='\n' && c!='\r'); } ), original_msg.end());

    // converting key to uppercase
    transform(key.begin(), key.end(), key.begin(), ::toupper);
    cout<<original_msg<<endl;
    cout<<key<<endl;
    int index = 0;
    string encrypted_msg = "";

    // iterating over the original message character by chracter and encrypting it and appending it to the new string
    for(int i=0; i< original_msg.length(); i++){
        char letter = original_msg[i];
        if(isalpha(letter)) {
            int position = (allChars.find(letter) + allChars.find(key[index])) % 26;
            char e_letter = allChars[position];
            encrypted_msg += e_letter;
            index++;
            // reset index if it is greater that key length
            if(index >= key.length()){
                index %= key.length();
            }
        // case when \n or \r is encountered
        }else{
            encrypted_msg += letter;
            index = 0;
        }
    }

    cout<<encrypted_msg;
    return encrypted_msg;
}


// method that encrypts file
static void encrypt_file(string key){
    ifstream input_file;
    string file_name, data;
    cout<<"Enter File name : ";
    cin>>file_name;

    // reading the file
    try {
        input_file.open(file_name);
        stringstream ss;
        ss << input_file.rdbuf();
        data = ss.str();
        cout<<"File read successufully!!"<<endl;
    } catch(exception e){
        cout<<e.what();
    }

    // encrypt the string read
    string encrypted_data = encrypt(data, key);
    cout<<"File data encrypted successfully!!"<<endl;
    ofstream output_file;
    // save the encrypted string in the file
    try {
        output_file.open("C:\\Users\\MOHIT\\Desktop\\ss.txt");
        output_file << encrypted_data;
        output_file.close();
        cout<<"encrypted file stored on Desktop!!"<<endl;
    }catch(exception ee){
        cout<<ee.what();
    }


}

int main()
{
    cout<<"Program to implement Vignere Cipher"<<endl;
    cout<<"Enter the Encryption key : ";
    string key;
    cin>>key;
    encrypt_file(key);
}

/*
Output for encrypting a file


Enter File name :F:/python_3.4/sample.txt
Enter File name : F:/python_3.4/sample.txt
File read successufully!!
File data encrypted successfully!!
encrypted file stored on Desktop!!


 */