#include <iostream>
#include <fstream>
#include <string>
#include <vector>
#include <map>

// The most magnificent function in this whole program.
// Prints a RASSE
void print_rasse()
{
    std::cout <<
                 "=====//==================//===\n"
                 "  __<<__________________<<__   \n"
                 " | ____ ____ ____ ____ ____ |  \n"
                 " | |  | |  | |  | |  | |  | |  \n"
                 " |_|__|_|__|_|__|_|__|_|__|_|  \n"
                 ".|                  RASSE   |. \n"
                 ":|__________________________|: \n"
                 "___(o)(o)___(o)(o)___(o)(o)____\n"
                 "-------------------------------" << std::endl;
}

std::vector<std::string> split(const std::string& s, char delimiter = ';'){
    std::vector<std::string> result;
    std::string tmp = s;

    while(tmp.find(delimiter) != std::string::npos)
    {
        std::string new_part = tmp.substr(0, tmp.find(delimiter));
        tmp = tmp.substr(tmp.find(delimiter)+1, tmp.size());
        if(not (new_part.empty()))
        {
            result.push_back(new_part);
        }
    }
    if(not tmp.empty())
    {
        result.push_back(tmp);
    }
    return result;
}

void routing(std::map<std::string, std::vector<std::string>> route, std::string inp)
{
    std::ifstream file(inp);
    std::string row;
    while(getline(file, row)){
        std::vector<std::string> parts = split(row);
        std::string line = parts.at(0);
        std::string station = parts.at(1);
        if (route.count(line)>0){
            route.at(line).push_back(station);
        }
        else{
            std::vector <std::string> state = {station};
            route.insert({line,state});
        }
    }
}

// Short and sweet main.
int main()
{
    std::map<std::string, std::vector<std::string>> route;
    print_rasse();
    std::string inp;
    std::cout << "Give a name for input file: ";
    std::cin >> inp;
    std::ifstream file(inp);
    if ( not file ) {
        std::cout << "Error: File could not be read." << std::endl;
        return EXIT_FAILURE;
    }else{
        std::string row;
        while(getline(file, row)){
            if (row.find(";")==std::string::npos){
                std::cout << "Error: Invalid format in file." << std::endl;
                return EXIT_FAILURE;
            }
            else{
                routing(route, inp);
            }
        }
    }
    return EXIT_SUCCESS;
}