/*
 * Copyright 2006, 2023, Haiku. All rights reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *              Stephan Aßmus <superstippi@gmx.de>
 *              Zardshard
 */

#include "RemoveStylesCommand.h"

#include <new>

#include <Catalog.h>
#include <Locale.h>
#include <StringFormat.h>

#include "PathSourceShape.h"
#include "Style.h"


#undef B_TRANSLATION_CONTEXT
#define B_TRANSLATION_CONTEXT "Icon-O-Matic-RemoveStylesCmd"


using std::nothrow;


RemoveStylesCommand::RemoveStylesCommand(Container<Style>* container,
                int32* const indices, int32 count)
        : RemoveCommand<Style>(container, indices, count),
          fShapes(indices && count > 0 ? new (nothrow) BList[count] : NULL)
{
        if (!fShapes)
                return;

        // get the shapes associated with each style
        for (int32 i = 0; i < fCount; i++) {
                if (fItems[i]) {
                        int32 listenerCount = fItems[i]->CountObservers();
                        for (int32 j = 0; j < listenerCount; j++) {
                                PathSourceShape* shape
                                        = dynamic_cast<PathSourceShape*>(fItems[i]->ObserverAtFast(j));
                                if (shape != NULL)
                                        fShapes[i].AddItem((void*)shape);
                        }
                }
        }
}


RemoveStylesCommand::~RemoveStylesCommand()
{
        delete[] fShapes;
}


status_t
RemoveStylesCommand::InitCheck()
{
        return fShapes ? B_OK : B_NO_INIT;
}


status_t
RemoveStylesCommand::Perform()
{
        // remove styles from the container
        status_t ret = RemoveCommand<Style>::Perform();
        if (ret != B_OK)
                return ret;
        fItemsRemoved = false; // We're not done yet!

        // remove styles from shapes that reference them
        for (int32 i = 0; i < fCount; i++) {
                if (!fItems[i])
                        continue;
                int32 shapeCount = fShapes[i].CountItems();
                for (int32 j = 0; j < shapeCount; j++) {
                        PathSourceShape* shape = (PathSourceShape*)fShapes[i].ItemAtFast(j);
                        shape->SetStyle(fContainer->ItemAt(0));
                }
        }

        fItemsRemoved = true;

        return B_OK;
}


status_t
RemoveStylesCommand::Undo()
{
        // add styles to the container
        status_t ret = RemoveCommand<Style>::Undo();
        if (ret != B_OK)
                return ret;
        fItemsRemoved = true; // We're not done yet!

        // add styles to the shapes which previously referenced them
        for (int32 i = 0; i < fCount; i++) {
                if (!fItems[i])
                        continue;
                int32 shapeCount = fShapes[i].CountItems();
                for (int32 j = 0; j < shapeCount; j++) {
                        PathSourceShape* shape = (PathSourceShape*)fShapes[i].ItemAtFast(j);
                        shape->SetStyle(fItems[i]);
                }
        }

        fItemsRemoved = false;

        return ret;
}


void
RemoveStylesCommand::GetName(BString& name)
{
        static BStringFormat format(B_TRANSLATE("Remove {0, plural, "
                "one{style} other{styles}}"));
        format.Format(name, fCount);
}