#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 get_conversation_result = pam->get_conversation(pamh, pam_conversation);
    if (get_conversation_result != 0) {
        return;
    }

    struct pam_message prompt;
    std::string message("Dual control token: ");
    prompt.msg = const_cast<char *>(message.c_str());
    prompt.msg_style = PAM_PROMPT_ECHO_OFF;
    std::vector<const struct pam_message *> prompts(1);
    prompts[0] = &prompt;
    std::vector<struct pam_response *> answers(1);
    int conversation_result = pam_conversation->conv(prompts, answers);
    if (conversation_result) {
        return;
    }

    if (answers[0]->resp_retcode) {
        return;
    }
    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_;
}