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

#include "RemovePathsCommand.h"

#include <new>

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

#include "Container.h"
#include "PathSourceShape.h"
#include "Shape.h"
#include "VectorPath.h"


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


using std::nothrow;


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

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


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


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


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

        // remove paths 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->Paths()->RemoveItem(fItems[i]);
                }
        }
        fItemsRemoved = true;

        return ret;
}


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

        // add paths to 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->Paths()->AddItem(fItems[i]);
                }
        }

        fItemsRemoved = false;

        return ret;
}


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