#include <security/pam_modules.h>
#include <algorithm>
#include <vector>
#include <string>

#include "conversation.h"
#include "pam.h"

pam_token_conversation::pam_token_conversation(pam_handle_t *pamh, const pam_p pam) {
    pam_conversation_p pam_conversation;
    int conversation_result = pam->get_conversation(pamh, pam_conversation);
    if (conversation_result != 0) {
        return;
    }

    const std::vector<const struct pam_message *> prompts(1);
    std::vector<struct pam_response *> answers(1);
    pam_conversation->conv(prompts, answers);
    std::string answer(answers[0]->resp);
    std::string::iterator delim = std::find(answer.begin(), answer.end(), ':');
    if (delim == answer.end()) {
        return;
    }
    std::string user_name(answer.begin(), delim);
    std::string token(delim + 1, answer.end());
    user_ = user_name;
    token_ = token;
}

std::string pam_token_conversation::token() {
    return token_;
}

std::string pam_token_conversation::user_name() {
    return user_;
}