/*
 * Copyright 2019-2026, Andrew Lindesay <apl@lindesay.co.nz>.
 * All rights reserved. Distributed under the terms of the MIT License.
 */
#include "Captcha.h"

#include <DataIO.h>

#include "Logger.h"

// These are keys that are used to store this object's data into a BMessage
// instance.

static const char* const kKeyToken = "token";
static const char* const kKeyPngImageData = "png_image_data";


Captcha::Captcha(BMessage* from)
        :
        fToken(""),
        fPngImageData(NULL)
{
        if (from->FindString(kKeyToken, &fToken) != B_OK)
                HDERROR("expected key [%s] in the message data when creating a captcha", kKeyToken);

        const void* data;
        ssize_t len;

        if (from->FindData(kKeyPngImageData, B_ANY_TYPE, &data, &len) != B_OK)
                HDERROR("expected key [%s] in the message data", kKeyPngImageData);
        else
                SetPngImageData(data, len);
}


Captcha::Captcha()
        :
        fToken(""),
        fPngImageData(NULL)
{
}


Captcha::~Captcha()
{
        delete fPngImageData;
}


const BString&
Captcha::Token() const
{
        return fToken;
}


BPositionIO*
Captcha::PngImageData() const
{
        return fPngImageData;
}


void
Captcha::SetToken(const BString& value)
{
        fToken = value;
}


void
Captcha::SetPngImageData(const void* data, size_t len)
{
        if (fPngImageData != NULL)
                delete fPngImageData;
        fPngImageData = NULL;
        if (data != NULL) {
                fPngImageData = new BMallocIO();
                fPngImageData->Write(data, len);
        }
}


status_t
Captcha::Archive(BMessage* into, bool deep) const
{
        status_t result = B_OK;
        if (result == B_OK && into == NULL)
                result = B_ERROR;
        if (result == B_OK)
                result = into->AddString(kKeyToken, fToken);
        if (result == B_OK && fPngImageData != NULL) {
                result = into->AddData(kKeyPngImageData, B_ANY_TYPE, fPngImageData->Buffer(),
                        fPngImageData->BufferLength());
        }
        return result;