#include <cs50.h>
#include <stdio.h>
#include <string.h>

int FindCharInKey(char _char, string _key);

int main(int argc, string argv[])
{

    if(argc != 2)
    {
        printf("A key must be submitted, containing 26 characters.\n");
        return 1;
    }

    string key = argv[1];

    if(strlen(key) == 26)
    {
        string ciphertext = "";
        string input = get_string("Plaintext:");
        for(int i = 0;i<strlen(input);i++)
        {
            if((int)input[i] <= 122 && (int)input[i] >= 97)
            {
                printf("check1\n");
                ciphertext[i] = key[FindCharInKey(input[i],key)];

            }
            else if((int)input[i] >= 65 && (int)input[i] <= 90)
            {
                printf("check2\n");
                ciphertext[i] = key[FindCharInKey(input[i],key)];

            }

        }
        printf("ciphertext:\n%s",ciphertext);
        return 0;

    }
    printf("Key must contain 26 characters. Current length: %lu\n", strlen(key));
    return 1;

}

int FindCharInKey(char _char, string _key)
{
    for(int i =0; i<strlen(_key);i++)
    {
        if(_char == (char)_key[i])
        {
            return i;

        }

    }
    return -1;

}