#include <daemons.h>
#include <lib.h>

inherit LIB_COMMAND;

mapping alpha1 = ([
  1 : "a",
  2 : "b",
  3 : "c",
  4 : "d",
  5 : "e",
  6 : "f",
  7 : "g",
  8 : "h",
  9 : "i",
  10 : "j",
  11 : "k",
  12 : "l",
  13 : "m",
  14 : "n",
  15 : "o",
  16 : "p",
  17 : "q",
  18 : "r",
  19 : "s",
  20 : "t",
  21 : "u",
  22 : "v",
  23 : "w",
  24 : "x",
  25 : "y",
  26 : "z",
]);

mapping alpha2 = ([]);

string doRotate(int code, string str){
    string *str_array = explode(str,"");
    string ret = "";

    foreach(string element in str_array){
        if(alpha2[element]){
            int operand;
            operand = alpha2[element]+code;
            if(operand > 26){
                operand -= 26;
            }
            if(operand < 1){
                operand = 26 - abs(operand);
            }
            ret += alpha1[operand];
        }
        else ret += element;
    }
    return ret;
}    

int cmd(string str){
    int code;
    string tmp, tmp2;
    string ret = "";
    string *str_array = ({});

    if(!str){
        write("Please see \"help rot\".");
        return 1;
    }

    str = lower_case(str);

    if(!alpha2 || !sizeof(alpha2)){
        alpha2 = ([]);
        foreach(mixed key, mixed val in alpha1) alpha2[val] = key;
    }

    if(sscanf(str,"%d %s",code, tmp)){
        str_array = explode(str,"");
        ret = doRotate(code, str);
    }

    else if(sscanf(str,"%s %s",tmp, tmp2)){
        if(tmp = "-q"){
            string tmp3, tmp4;
            if(sscanf(tmp2,"%s %s",tmp3, tmp4) == 2 &&
              sizeof(tmp3) == 1 && sizeof(tmp4) == 1){
                ret = alpha2[tmp3] - alpha2[tmp4];
            }
            else {
                for(int i=0;i<26;i++){
                    write(i+": "+doRotate(i, tmp2));
                }
            }
        }
    }

    write("Output: "+ret);
    return 1;
}



string GetHelp(string str) {
    return "Syntax: rot -q <FIRSTLETTER> <SECONDLETTER>\n"
    "        rot <NUMBER> <STRING>\n"
    "Provides arbitrary alphabetical place rotation "
    "of letters. For example, \n"
    "rot 13 hello\n"
    "returns uryyb, and\n"
    "rot 13 uryyb\n"
    "returns hello.";
}