#include <iostream>
#include <algorithm>
#include <cctype>
#include <string>
#include "vij.hpp"
using namespace std;
string get(string m,string sa,string k){
if(m[0]=='r') return readvij(sa,k);
if(m[0]=='m') return vij(sa,k);
return "Selecting mode error.";
}
int main(){
string sa,k,m;
cout<<"Write mode to input,make(m) or read(r):";
cin>>m;
cin.ignore();
cout<<m<<endl;
cout<<"Write sentence to input:";
getline(cin,sa);
cout<<sa<<endl;
cout<<"Write key to input:";
cin>>k;
cout<<k<<endl;
cout<<endl<<"->";
cout<<get(m,sa,k)<<endl;
}
#ifndef VIJ_HPP_INCLUDED
#define VIJ_HPP_INCLUDED
#include <iostream>
#include <algorithm>
#include <cctype>
#include <string>
using namespace std;
string vij(string sen,string key){
string ang(sen);
int s=0,k=0,sl=sen.length(),kl=key.length();
for(int i=0;i<sl;i++)sen[i]=tolower(sen[i]);
for(int i=0;i<kl;i++)key[i]=tolower(key[i]);
while(s<sl){
if(sen[s]>='a' && sen[s]<='z'){
ang[s]='A'+(sen[s]+key[k]-2*'a')%26;
s++;
k++;
}else{
ang[s]=sen[s];
s++;
}
if(k>=kl)k=0;
}
return ang;
}
string readvij(string ang,string key){
string sen(ang);
int a=0,k=0,al=ang.length(),kl=key.length();
for(int i=0;i<al;i++)ang[i]=tolower(ang[i]);
for(int i=0;i<kl;i++)key[i]=tolower(key[i]);
while(a<al){
if(sen[a]>='a' && ang[a]<='z'){
sen[a]='A'+((ang[a]-'a')-(key[k]-'a'))%26;
a++;
k++;
}else{
sen[a]=ang[a];
a++;
}
if(k>=kl)k=0;
}
return sen;
}
#endif // VIJ_HPP_INCLUDED
