#include "./attachmentinfo.h"

#include <tagparser/abstractcontainer.h>
#include <tagparser/abstractattachment.h>

#include <c++utilities/conversion/conversionexception.h>
#include <c++utilities/conversion/stringconversion.h>

#include <cstring>
#include <iostream>

using namespace std;
using namespace ConversionUtilities;
using namespace Media;

namespace Cli {

void AttachmentInfo::parseDenotation(const char *denotation)
{
    if(!strncmp(denotation, "id=", 3)) {
        try {
            id = stringToNumber<uint64, string>(denotation + 3);
            hasId = true;
        } catch(const ConversionException &) {
            cerr << "The specified attachment ID \"" << (denotation + 3) << "\" is invalid.";
        }
    } else if(!strncmp(denotation, "path=", 5)) {
        path = denotation + 5;
    } else if(!strncmp(denotation, "name=", 5)) {
        name = denotation + 5;
    } else if(!strncmp(denotation, "mime=", 5)) {
        mime = denotation + 5;
    } else if(!strncmp(denotation, "desc=", 5)) {
        desc = denotation + 5;
    } else {
        cerr << "The attachment specification \"" << denotation << "\" is invalid and will be ignored.";
    }
}

void AttachmentInfo::apply(AbstractContainer *container)
{
    static const string context("applying specified attachments");
    AbstractAttachment *attachment = nullptr;
    bool attachmentFound = false;
    switch(action) {
    case AttachmentAction::Add:
        if(!path || !name) {
            cerr << "Argument --update-argument specified but no name/path provided." << endl;
            return;
        }
        apply(container->createAttachment());
        break;
    case AttachmentAction::Update:
        if(hasId) {
            for(size_t i = 0, count = container->attachmentCount(); i < count; ++i) {
                attachment = container->attachment(i);
                if(attachment->id() == id) {
                    apply(attachment);
                    attachmentFound = true;
                }
            }
            if(!attachmentFound) {
                container->addNotification(NotificationType::Critical, "Attachment with the specified ID \"" + numberToString(id) + "\" does not exist and hence can't be updated.", context);
            }
        } else if(name) {
            for(size_t i = 0, count = container->attachmentCount(); i < count; ++i) {
                attachment = container->attachment(i);
                if(attachment->name() == name) {
                    apply(attachment);
                    attachmentFound = true;
                }
            }
            if(!attachmentFound) {
                container->addNotification(NotificationType::Critical, "Attachment with the specified name \"" + string(name) + "\" does not exist and hence can't be updated.", context);
            }
        } else {
            cerr << "Argument --update-argument specified but no ID/name provided." << endl;
        }
        break;
    case AttachmentAction::Remove:
        if(hasId) {
            for(size_t i = 0, count = container->attachmentCount(); i < count; ++i) {
                attachment = container->attachment(i);
                if(attachment->id() == id) {
                    attachment->setIgnored(true);
                    attachmentFound = true;
                }
            }
            if(!attachmentFound) {
                container->addNotification(NotificationType::Critical, "Attachment with the specified ID \"" + numberToString(id) + "\" does not exist and hence can't be removed.", context);
            }
        } else if(name) {
            for(size_t i = 0, count = container->attachmentCount(); i < count; ++i) {
                attachment = container->attachment(i);
                if(attachment->name() == name) {
                    attachment->setIgnored(true);
                    attachmentFound = true;
                }
            }
            if(!attachmentFound) {
                container->addNotification(NotificationType::Critical, "Attachment with the specified name \"" + string(name) + "\" does not exist and hence can't be removed.", context);
            }
        } else {
            cerr << "Argument --remove-argument specified but no ID/name provided." << endl;
        }
        break;
    }
}

void AttachmentInfo::apply(AbstractAttachment *attachment)
{
    if(hasId) {
        attachment->setId(id);
    }
    if(path) {
        attachment->setFile(path);
    }
    if(name) {
        attachment->setName(name);
    }
    if(mime) {
        attachment->setMimeType(mime);
    }
    if(desc) {
        attachment->setDescription(desc);
    }
}

void AttachmentInfo::reset()
{
    action = AttachmentAction::Add;
    id = 0;
    hasId = false;
    path = name = mime = desc = nullptr;
}

bool AttachmentInfo::next(AbstractContainer *container)
{
    if(!id && !path && !name && !mime && !desc) {
        // skip empty attachment infos
        return false;
    }
    apply(container);
    reset();
    return true;
}

}