// Aseprite
// Copyright (C) 2018-present  Igara Studio S.A.
// Copyright (C) 2015-2018  David Capello
//
// This program is distributed under the terms of
// the End-User License Agreement for Aseprite.

#ifdef HAVE_CONFIG_H
  #include "config.h"
#endif

#include "app/app.h"
#include "app/cmd/add_layer.h"
#include "app/cmd/add_slice.h"
#include "app/cmd/add_tile.h"
#include "app/cmd/add_tileset.h"
#include "app/cmd/assign_color_profile.h"
#include "app/cmd/clear_cel.h"
#include "app/cmd/convert_color_profile.h"
#include "app/cmd/copy_region.h"
#include "app/cmd/flatten_layers.h"
#include "app/cmd/remove_layer.h"
#include "app/cmd/remove_slice.h"
#include "app/cmd/remove_tag.h"
#include "app/cmd/remove_tile.h"
#include "app/cmd/remove_tileset.h"
#include "app/cmd/set_grid_bounds.h"
#include "app/cmd/set_layer_tileset.h"
#include "app/cmd/set_mask.h"
#include "app/cmd/set_pixel_ratio.h"
#include "app/cmd/set_sprite_size.h"
#include "app/cmd/set_sprite_tile_management_plugin.h"
#include "app/cmd/set_transparent_color.h"
#include "app/color_spaces.h"
#include "app/commands/commands.h"
#include "app/commands/params.h"
#include "app/console.h"
#include "app/context.h"
#include "app/doc.h"
#include "app/doc_access.h"
#include "app/doc_api.h"
#include "app/doc_range.h"
#include "app/doc_undo.h"
#include "app/doc_undo_observer.h"
#include "app/file/palette_file.h"
#include "app/script/docobj.h"
#include "app/script/engine.h"
#include "app/script/luacpp.h"
#include "app/script/security.h"
#include "app/script/userdata.h"
#include "app/site.h"
#include "app/transaction.h"
#include "app/tx.h"
#include "app/ui/doc_view.h"
#include "base/convert_to.h"
#include "base/fs.h"
#include "base/gcd.h"
#include "doc/layer.h"
#include "doc/layer_tilemap.h"
#include "doc/mask.h"
#include "doc/palette.h"
#include "doc/slice.h"
#include "doc/sprite.h"
#include "doc/tag.h"
#include "doc/tileset.h"
#include "doc/tilesets.h"
#include "undo/undo_state.h"

#include <algorithm>

namespace app { namespace script {

namespace {

int Sprite_new(lua_State* L)
{
  std::unique_ptr<Doc> doc;

  // Duplicate a sprite
  if (auto otherSpr = may_get_docobj<doc::Sprite>(L, 1)) {
    Doc* otherDoc = static_cast<Doc*>(otherSpr->document());
    doc.reset(otherDoc->duplicate(DuplicateExactCopy));
  }
  else {
    doc::ImageSpec spec(doc::ColorMode::RGB, 1, 1, 0);
    if (auto spec2 = may_get_obj<doc::ImageSpec>(L, 1)) {
      spec = *spec2;
    }
    else {
      if (lua_istable(L, 1)) {
        // Sprite{ fromFile }
        int type = lua_getfield(L, 1, "fromFile");
        if (type != LUA_TNIL) {
          if (const char* fromFile = lua_tostring(L, -1)) {
            std::string fn = fromFile;
            lua_pop(L, 1);

            bool oneFrame = (lua_is_key_true(L, -1, "oneFrame"));

            return load_sprite_from_file(L,
                                         fn.c_str(),
                                         (oneFrame ? LoadSpriteFromFileParam::OneFrameAsSprite :
                                                     LoadSpriteFromFileParam::FullAniAsSprite));
          }
        }
        lua_pop(L, 1);

        // In case that there is no "fromFile" field
        if (type == LUA_TNIL) {
          // Sprite{ width, height, colorMode }
          lua_getfield(L, 1, "width");
          lua_getfield(L, 1, "height");
          spec.setWidth(lua_tointeger(L, -2));
          spec.setHeight(lua_tointeger(L, -1));
          lua_pop(L, 2);

          type = lua_getfield(L, 1, "colorMode");
          if (type != LUA_TNIL)
            spec.setColorMode((doc::ColorMode)lua_tointeger(L, -1));
          lua_pop(L, 1);
        }
      }
      else {
        const int w = lua_tointeger(L, 1);
        const int h = lua_tointeger(L, 2);
        const int colorMode = (lua_isnone(L, 3) ? IMAGE_RGB : lua_tointeger(L, 3));
        spec.setWidth(w);
        spec.setHeight(h);
        spec.setColorMode((doc::ColorMode)colorMode);
      }
      spec.setColorSpace(get_working_rgb_space_from_preferences());
    }

    if (spec.width() < 1)
      return luaL_error(L, "invalid width value = %d in Sprite()", spec.width());
    if (spec.height() < 1)
      return luaL_error(L, "invalid height value = %d in Sprite()", spec.height());

    std::unique_ptr<Sprite> sprite(Sprite::MakeStdSprite(spec, 256));
    doc.reset(new Doc(sprite.get()));
    sprite.release();
  }

  app::Context* ctx = App::instance()->context();
  doc->setContext(ctx);

  push_docobj(L, doc->sprite());
  doc.release();
  return 1;
}

int Sprite_eq(lua_State* L)
{
  const auto a = may_get_docobj<Sprite>(L, 1);
  const auto b = may_get_docobj<Sprite>(L, 2);
  lua_pushboolean(L, (!a && !b) || (a && b && a->id() == b->id()));
  return 1;
}

int Sprite_resize(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  gfx::Size size = convert_args_into_size(L, 2);

  // Fix invalid sizes
  size.w = std::max(1, size.w);
  size.h = std::max(1, size.h);

  Command* resizeCommand = Commands::instance()->byId(CommandId::SpriteSize());

  // TODO use SpriteSizeParams directly instead of converting back and
  //      forth between strings.
  Params params;
  params.set("ui", "false");
  params.set("width", base::convert_to<std::string>(size.w).c_str());
  params.set("height", base::convert_to<std::string>(size.h).c_str());

  app::Context* appCtx = App::instance()->context();
  auto oldDoc = appCtx->activeDocument();
  appCtx->setActiveDocument(static_cast<Doc*>(sprite->document()));
  appCtx->executeCommand(resizeCommand, params);
  appCtx->setActiveDocument(oldDoc);
  return 0;
}

int Sprite_crop(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  Doc* doc = static_cast<Doc*>(sprite->document());
  gfx::Rect bounds;

  // Use mask bounds
  if (lua_isnone(L, 2)) {
    if (doc->isMaskVisible())
      bounds = doc->mask()->bounds();
    else
      bounds = sprite->bounds();
  }
  else {
    bounds = convert_args_into_rect(L, 2);
  }

  if (!bounds.isEmpty()) {
    Tx tx(doc);
    DocApi(doc, tx).cropSprite(sprite, bounds);
    tx.commit();
  }
  return 0;
}

int Sprite_saveAs_base(lua_State* L, std::string& absFn)
{
  bool result = false;
  auto sprite = get_docobj<Sprite>(L, 1);
  const char* fn = luaL_checkstring(L, 2);
  if (fn && sprite) {
    Doc* doc = static_cast<Doc*>(sprite->document());
    app::Context* appCtx = App::instance()->context();
    appCtx->setActiveDocument(doc);

    absFn = base::get_absolute_path(fn);
    if (!ask_access(L, absFn.c_str(), FileAccessMode::Write, ResourceType::File))
      return luaL_error(L, "script doesn't have access to write file %s", absFn.c_str());

    Command* saveCommand = Commands::instance()->byId(CommandId::SaveFileCopyAs());

    Params params;
    params.set("filename", absFn.c_str());
    params.set("useUI", "false");
    appCtx->executeCommand(saveCommand, params);

    result = true;
  }
  lua_pushboolean(L, result);
  return 1;
}

int Sprite_saveAs(lua_State* L)
{
  std::string fn;
  int res = Sprite_saveAs_base(L, fn);
  if (!fn.empty()) {
    auto sprite = get_docobj<Sprite>(L, 1);
    if (sprite) {
      Doc* doc = static_cast<Doc*>(sprite->document());
      doc->setFilename(fn);
      doc->markAsSaved();
    }
  }
  return res;
}

int Sprite_saveCopyAs(lua_State* L)
{
  std::string fn;
  return Sprite_saveAs_base(L, fn);
}

int Sprite_close(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  Doc* doc = static_cast<Doc*>(sprite->document());
  try {
    DocDestroyer destroyer(static_cast<app::Context*>(doc->context()), doc, 500);
    destroyer.destroyDocument();
    return 0;
  }
  catch (const LockedDocException& ex) {
    return luaL_error(L, "cannot lock document to close it\n%s", ex.what());
  }
}

int Sprite_loadPalette(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  const char* fn = luaL_checkstring(L, 2);
  if (fn && sprite) {
    std::string absFn = base::get_absolute_path(fn);
    if (!ask_access(L, absFn.c_str(), FileAccessMode::Read, ResourceType::File))
      return luaL_error(L, "script doesn't have access to open file %s", absFn.c_str());

    Doc* doc = static_cast<Doc*>(sprite->document());
    std::unique_ptr<doc::Palette> palette(load_palette(absFn.c_str()));
    if (palette) {
      Tx tx(doc);
      // TODO Merge this with the code in LoadPaletteCommand
      doc->getApi(tx).setPalette(sprite, 0, palette.get());
      tx.commit();
    }
  }
  return 0;
}

int Sprite_setPalette(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  auto pal = get_palette_from_arg(L, 2);
  if (sprite && pal) {
    Doc* doc = static_cast<Doc*>(sprite->document());

    Tx tx(doc);
    doc->getApi(tx).setPalette(sprite, 0, pal);
    tx.commit();
  }
  return 0;
}

int Sprite_assignColorSpace(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  auto cs = get_obj<gfx::ColorSpace>(L, 2);
  Tx tx(sprite);
  tx(new cmd::AssignColorProfile(sprite, base::make_ref<gfx::ColorSpace>(*cs)));
  tx.commit();
  return 1;
}

int Sprite_convertColorSpace(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  auto cs = get_obj<gfx::ColorSpace>(L, 2);
  Tx tx(sprite);
  tx(new cmd::ConvertColorProfile(sprite, base::make_ref<gfx::ColorSpace>(*cs)));
  tx.commit();
  return 1;
}

int Sprite_flatten(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);

  DocRange range;
  for (auto layer : sprite->root()->layers())
    range.selectLayer(layer);

  Tx tx(sprite);
  cmd::FlattenLayers::Options options;
  options.newBlendMethod = true;
  tx(new cmd::FlattenLayers(sprite, range.selectedLayers(), options));
  tx.commit();
  return 0;
}

int Sprite_newLayer(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  doc::Layer* newLayer = new doc::LayerImage(sprite);

  Tx tx(sprite);
  tx(new cmd::AddLayer(sprite->root(), newLayer, sprite->root()->lastLayer()));
  tx.commit();

  push_docobj(L, newLayer);
  return 1;
}

int Sprite_newGroup(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  doc::Layer* newGroup = new doc::LayerGroup(sprite);

  Tx tx(sprite);
  tx(new cmd::AddLayer(sprite->root(), newGroup, sprite->root()->lastLayer()));
  tx.commit();

  push_docobj(L, newGroup);
  return 1;
}

int Sprite_deleteLayer(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  auto layer = may_get_docobj<Layer>(L, 2);
  if (!layer && lua_isstring(L, 2)) {
    const char* layerName = lua_tostring(L, 2);
    if (layerName) {
      for (Layer* child : sprite->allLayers()) {
        if (child->name() == layerName) {
          layer = child;
          break;
        }
      }
    }
  }
  if (layer) {
    if (sprite != layer->sprite())
      return luaL_error(L, "the layer doesn't belong to the sprite");
    Tx tx(sprite);
    tx(new cmd::RemoveLayer(layer));
    tx.commit();
    return 0;
  }
  else {
    return luaL_error(L, "layer not found");
  }
}

int Sprite_newFrame(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  doc::frame_t frame = sprite->lastFrame() + 1;
  doc::frame_t copyThis = frame;
  if (lua_gettop(L) >= 2) {
    frame = get_frame_number_from_arg(L, 2);
    if (frame < 0)
      return luaL_error(L, "frame index out of bounds %d", frame + 1);
    copyThis = frame + 1; // addFrame() copies the previous frame of the given one.
  }

  Doc* doc = static_cast<Doc*>(sprite->document());

  Tx tx(doc);
  doc->getApi(tx).addFrame(sprite, copyThis);
  tx.commit();

  push_sprite_frame(L, sprite, frame);
  return 1;
}

int Sprite_newEmptyFrame(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  doc::frame_t frame = sprite->lastFrame() + 1;
  if (lua_gettop(L) >= 2) {
    frame = get_frame_number_from_arg(L, 2);
    if (frame < 0)
      return luaL_error(L, "frame index out of bounds %d", frame + 1);
  }

  Doc* doc = static_cast<Doc*>(sprite->document());

  Tx tx(doc);
  DocApi(doc, tx).addEmptyFrame(sprite, frame);
  tx.commit();

  push_sprite_frame(L, sprite, frame);
  return 1;
}

int Sprite_deleteFrame(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  doc::frame_t frame = get_frame_number_from_arg(L, 2);
  if (frame < 0 || frame > sprite->lastFrame())
    return luaL_error(L, "frame index out of bounds %d", frame + 1);

  Doc* doc = static_cast<Doc*>(sprite->document());

  Tx tx(doc);
  doc->getApi(tx).removeFrame(sprite, frame);
  tx.commit();
  return 0;
}

int Sprite_newCel(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  auto layerBase = get_docobj<Layer>(L, 2);
  if (!layerBase->isImage())
    return luaL_error(L, "unexpected kind of layer in Sprite:newCel()");

  frame_t frame = get_frame_number_from_arg(L, 3);
  if (frame < 0 || frame > sprite->lastFrame())
    return luaL_error(L, "frame index out of bounds %d", frame + 1);

  Doc* doc = static_cast<Doc*>(sprite->document());
  LayerImage* layer = static_cast<LayerImage*>(layerBase);
  ImageRef image(nullptr);

  Image* srcImage = may_get_image_from_arg(L, 4);
  gfx::Point pos = convert_args_into_point(L, 5);
  Cel* cel = nullptr;

  // For background layers we just draw the image in the existent cel
  if (layer->isBackground()) {
    cel = layer->cel(frame);
    ASSERT(cel);

    Tx tx(doc);
    DocApi api = doc->getApi(tx);
    api.clearCel(layer, frame);
    if (srcImage) {
      tx(new cmd::CopyRegion(cel->image(), srcImage, gfx::Region(srcImage->bounds()), pos, false));
    }
    tx.commit();
  }
  // For transparent layers we just draw the image in the existent cel
  else {
    if (srcImage)
      image.reset(Image::createCopy(srcImage));
    else
      image.reset(Image::create(sprite->spec()));

    cel = new Cel(frame, image);
    cel->setPosition(pos);

    Tx tx(doc);
    DocApi api = doc->getApi(tx);
    if (layer->cel(frame))
      api.clearCel(layer, frame);
    api.addCel(layer, cel);
    tx.commit();
  }

  push_docobj(L, cel);
  return 1;
}

int Sprite_deleteCel(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  auto cel = may_get_docobj<doc::Cel>(L, 2);
  if (!cel) {
    if (auto layer = may_get_docobj<doc::Layer>(L, 2)) {
      if (sprite != layer->sprite())
        return luaL_error(L, "the layer doesn't belong to the sprite");
      doc::frame_t frame = get_frame_number_from_arg(L, 3);
      if (layer->isImage())
        cel = static_cast<doc::LayerImage*>(layer)->cel(frame);
    }
  }

  if (cel) {
    Tx tx(sprite);
    tx(new cmd::ClearCel(cel));
    tx.commit();
    return 0;
  }
  else {
    return luaL_error(L, "cel not found");
  }
}

int Sprite_newTag(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  auto from = get_frame_number_from_arg(L, 2);
  auto to = get_frame_number_from_arg(L, 3);
  auto tag = new doc::Tag(from, to);

  Tx tx(sprite);
  tx(new cmd::AddTag(sprite, tag));
  tx.commit();

  push_docobj(L, tag);
  return 1;
}

int Sprite_deleteTag(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  auto tag = may_get_docobj<Tag>(L, 2);
  if (!tag && lua_isstring(L, 2)) {
    const char* tagName = lua_tostring(L, 2);
    if (tagName)
      tag = sprite->tags().getByName(tagName);
  }
  if (tag) {
    if (sprite != tag->owner()->sprite())
      return luaL_error(L, "the tag doesn't belong to the sprite");
    Tx tx(sprite);
    tx(new cmd::RemoveTag(sprite, tag));
    tx.commit();
    return 0;
  }
  else {
    return luaL_error(L, "tag not found");
  }
}

int Sprite_newSlice(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  auto slice = new doc::Slice();

  gfx::Rect bounds = convert_args_into_rect(L, 2);
  if (!bounds.isEmpty())
    slice->insert(0, doc::SliceKey(bounds));

  Tx tx(sprite);
  tx(new cmd::AddSlice(sprite, slice));
  tx.commit();

  push_docobj(L, slice);
  return 1;
}

int Sprite_deleteSlice(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  doc::Slice* slice = may_get_docobj<Slice>(L, 2);
  if (!slice && lua_isstring(L, 2)) {
    const char* sliceName = lua_tostring(L, 2);
    if (sliceName)
      slice = sprite->slices().getByName(sliceName);
  }
  if (slice) {
    if (sprite != slice->owner()->sprite())
      return luaL_error(L, "the slice doesn't belong to the sprite");
    Tx tx(sprite);
    tx(new cmd::RemoveSlice(sprite, slice));
    tx.commit();
    return 0;
  }
  else {
    return luaL_error(L, "slice not found");
  }
}

int Sprite_newTileset(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  Tileset* tileset = nullptr;

  // Make this tileset a clone of the given tileset
  if (auto reference = may_get_docobj<Tileset>(L, 2)) {
    if (reference->sprite() != sprite) {
      return luaL_error(L, "cannot duplicate a tileset that belongs to another sprite");
    }
    tileset = Tileset::MakeCopyCopyingImages(reference);
  }
  else {
    Grid grid(sprite->gridBounds().size()); // Use sprite grid bounds by default
    int ntiles = 1;
    if (!lua_isnone(L, 2)) {
      if (auto g = may_get_obj<Grid>(L, 2)) {
        grid = *g;
      }
      // Convert Rectangle into a Grid
      else if (lua_istable(L, 2) || may_get_obj<gfx::Rect>(L, 2)) {
        gfx::Rect rect = convert_args_into_rect(L, 2);
        grid = Grid(rect.size());
        grid.origin(rect.origin());
      }
      else {
        return luaL_error(L, "grid or table expected");
      }

      int type = lua_type(L, 3);
      if (type != LUA_TNONE) {
        if (type != LUA_TNUMBER) {
          return luaL_error(L, "ntiles field must be a number");
        }
        else if ((ntiles = lua_tointeger(L, 3)) <= 0) {
          return luaL_error(L, "ntiles field must be an integer greater than 0");
        }
      }
    }

    // This a limitation in our code and doesn't make too much sense
    // to specify a different origin by default (because the origin is
    // specified on the tilemap cel).
    if (grid.origin() != gfx::Point(0, 0)) {
      return luaL_error(L, "a tileset with origin different than 0,0 cannot be created");
    }

    tileset = new Tileset(sprite, grid, ntiles);
  }

  Tx tx(sprite);
  tx(new cmd::AddTileset(sprite, tileset));
  tx.commit();

  push_docobj(L, tileset);
  return 1;
}

int Sprite_deleteTileset(lua_State* L)
{
  int tsi = -1;
  auto sprite = get_docobj<Sprite>(L, 1);
  doc::Tileset* tileset = may_get_docobj<Tileset>(L, 2);
  if (!tileset && lua_isinteger(L, 2)) {
    tsi = lua_tointeger(L, 2);
    tileset = sprite->tilesets()->get(tsi);
  }
  else if (tileset) {
    tsi = sprite->tilesets()->getIndex(tileset);
  }
  if (tileset && tsi >= 0) {
    if (sprite != tileset->sprite())
      return luaL_error(L, "the tileset doesn't belong to the sprite");
    Tx tx(sprite);

    // Set the tileset from all layers that are using it
    for (auto layer : sprite->allLayers()) {
      if (layer->isTilemap()) {
        auto tilemap = static_cast<doc::LayerTilemap*>(layer);
        if (tilemap->tilesetIndex() == tsi) {
          // TODO improve this in some way, we're setting tileset
          //      index 0, but probably we should support a tilemap
          //      without tileset (as a temporal state)
          tx(new cmd::SetLayerTileset(tilemap, 0));
        }
      }
    }

    tx(new cmd::RemoveTileset(sprite, tsi));
    tx.commit();
    return 0;
  }
  else {
    return luaL_error(L, "tileset not found");
  }
}

int Sprite_newTile(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  auto ts = get_docobj<Tileset>(L, 2);
  if (!ts)
    return luaL_error(L, "empty argument not allowed and must be a Tileset object");
  if (ts->sprite() != sprite)
    return luaL_error(L, "the tileset belongs to another sprite");
  tile_index ti = ts->size();
  if (lua_isinteger(L, 3)) {
    ti = tile_index(lua_tointeger(L, 3));
    if (ti < 1)
      return luaL_error(L, "index must be equal to or greater than 1");
  }
  ts->insert(ti, ts->makeEmptyTile());
  Tx tx(sprite);
  tx(new cmd::AddTile(ts, ti));
  tx.commit();
  push_tile(L, ts, ti);
  return 1;
}

int Sprite_deleteTile(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  tile_index ti = 0;
  auto ts = may_get_docobj<Tileset>(L, 2);
  if (ts)
    ti = lua_tointeger(L, 3);
  else
    ts = get_tile_index_from_arg(L, 2, ti);
  if (!ts)
    return luaL_error(L, "Sprite:deleteTile() needs a Tileset or Tile as first argument");
  if (ts->sprite() != sprite)
    return luaL_error(L, "the tileset belongs to another sprite");
  if (ti == 0)
    return luaL_error(L, "tile index = 0 cannot be removed");
  if (ti < 0 || ti >= ts->size())
    return luaL_error(L, "index out of bounds");
  Tx tx(sprite);
  tx(new cmd::RemoveTile(ts, ti));
  tx.commit();
  push_tile(L, ts, ti);
  return 1;
}

int Sprite_get_events(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  push_sprite_events(L, sprite);
  return 1;
}

int Sprite_get_id(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  lua_pushinteger(L, sprite->id());
  return 1;
}

int Sprite_get_filename(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  lua_pushstring(L, sprite->document()->filename().c_str());
  return 1;
}

int Sprite_get_isModified(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  Doc* doc = static_cast<Doc*>(sprite->document());
  lua_pushboolean(L, doc->isModified());
  return 1;
}

int Sprite_get_hasAssociatedFile(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  Doc* doc = static_cast<Doc*>(sprite->document());
  lua_pushboolean(L, doc->isAssociatedToFile());
  return 1;
}

int Sprite_get_width(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  lua_pushinteger(L, sprite->width());
  return 1;
}

int Sprite_get_height(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  lua_pushinteger(L, sprite->height());
  return 1;
}

int Sprite_get_colorMode(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  lua_pushinteger(L, sprite->pixelFormat());
  return 1;
}

int Sprite_get_colorSpace(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  auto cs = sprite->colorSpace();
  if (cs)
    push_color_space(L, *cs);
  else
    lua_pushnil(L);
  return 1;
}

int Sprite_get_spec(lua_State* L)
{
  const auto sprite = get_docobj<Sprite>(L, 1);
  push_obj(L, sprite->spec());
  return 1;
}

int Sprite_get_selection(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  push_sprite_selection(L, sprite);
  return 1;
}

int Sprite_get_frames(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  push_sprite_frames(L, sprite);
  return 1;
}

int Sprite_get_palettes(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  push_sprite_palettes(L, sprite);
  return 1;
}

int Sprite_get_layers(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  push_sprite_layers(L, sprite);
  return 1;
}

int Sprite_get_cels(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  push_cels(L, sprite);
  return 1;
}

int Sprite_get_tags(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  push_sprite_tags(L, sprite);
  return 1;
}

int Sprite_get_slices(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  push_sprite_slices(L, sprite);
  return 1;
}

int Sprite_get_tilesets(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  push_tilesets(L, sprite->tilesets());
  return 1;
}

int Sprite_get_backgroundLayer(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  doc::Layer* layer = sprite->backgroundLayer();
  if (layer)
    push_docobj(L, layer);
  else
    lua_pushnil(L);
  return 1;
}

int Sprite_get_transparentColor(lua_State* L)
{
  const auto sprite = get_docobj<Sprite>(L, 1);
  lua_pushinteger(L, sprite->transparentColor());
  return 1;
}

int Sprite_set_transparentColor(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  const int index = lua_tointeger(L, 2);
  Tx tx(sprite);
  tx(new cmd::SetTransparentColor(sprite, index));
  tx.commit();
  return 0;
}

int Sprite_set_filename(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  const char* fn = lua_tostring(L, 2);
  sprite->document()->setFilename(fn ? std::string(fn) : std::string());
  return 0;
}

int Sprite_set_width(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  const int width = lua_tointeger(L, 2);
  Tx tx(sprite);
  tx(new cmd::SetSpriteSize(sprite, width, sprite->height()));
  tx.commit();
  return 0;
}

int Sprite_set_height(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  const int height = lua_tointeger(L, 2);
  Tx tx(sprite);
  tx(new cmd::SetSpriteSize(sprite, sprite->width(), height));
  tx.commit();
  return 0;
}

int Sprite_set_selection(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  const auto mask = get_mask_from_arg(L, 2);
  Doc* doc = static_cast<Doc*>(sprite->document());
  Tx tx(sprite);
  tx(new cmd::SetMask(doc, mask));
  tx.commit();
  return 0;
}

int Sprite_get_bounds(lua_State* L)
{
  const auto sprite = get_docobj<Sprite>(L, 1);
  push_obj<gfx::Rect>(L, sprite->bounds());
  return 1;
}

int Sprite_get_gridBounds(lua_State* L)
{
  const auto sprite = get_docobj<Sprite>(L, 1);
  push_obj<gfx::Rect>(L, sprite->gridBounds());
  return 1;
}

int Sprite_set_gridBounds(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  const gfx::Rect bounds = convert_args_into_rect(L, 2);
  Tx tx(sprite);
  tx(new cmd::SetGridBounds(sprite, bounds));
  tx.commit();
  return 0;
}

int Sprite_get_pixelRatio(lua_State* L)
{
  const auto sprite = get_docobj<Sprite>(L, 1);
  push_obj<gfx::Size>(L, sprite->pixelRatio());
  return 1;
}

int Sprite_set_pixelRatio(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  gfx::Size pixelRatio = convert_args_into_size(L, 2);
  if (pixelRatio.w < 1 || pixelRatio.h < 1)
    return luaL_error(L, "invalid pixel ratio = %d:%d", pixelRatio.w, pixelRatio.h);

  double gcd = base::gcd(double(pixelRatio.w), double(pixelRatio.h));
  pixelRatio.w = int(double(pixelRatio.w) / gcd);
  pixelRatio.h = int(double(pixelRatio.h) / gcd);
  Tx tx(sprite);
  tx(new cmd::SetPixelRatio(sprite, pixelRatio));
  tx.commit();
  return 0;
}

int Sprite_get_tileManagementPlugin(lua_State* L)
{
  const auto sprite = get_docobj<Sprite>(L, 1);
  if (sprite->hasTileManagementPlugin())
    lua_pushstring(L, sprite->tileManagementPlugin().c_str());
  else
    lua_pushnil(L);
  return 1;
}

int Sprite_set_tileManagementPlugin(lua_State* L)
{
  auto sprite = get_docobj<Sprite>(L, 1);
  std::string value;
  if (const char* p = lua_tostring(L, 2))
    value = p;

  if (sprite->tileManagementPlugin() != value) {
    Tx tx(sprite);
    tx(new cmd::SetSpriteTileManagementPlugin(sprite, value));
    tx.commit();
  }
  return 0;
}

int Sprite_get_useLayerUuids(lua_State* L)
{
  auto* sprite = get_docobj<Sprite>(L, 1);
  lua_pushboolean(L, sprite->useLayerUuids());
  return 1;
}

int Sprite_set_useLayerUuids(lua_State* L)
{
  auto* sprite = get_docobj<Sprite>(L, 1);
  if (lua_isboolean(L, 2)) {
    const bool value = lua_toboolean(L, 2);
    sprite->useLayerUuids(value);
  }
  return 0;
}

int Sprite_get_undoHistory(lua_State* L)
{
  const auto* sprite = get_docobj<Sprite>(L, 1);
  const auto* doc = static_cast<Doc*>(sprite->document());
  const auto* history = doc->undoHistory();

  if (!history) {
    lua_pushnil(L);
    return 1;
  }

  const undo::UndoState* currentState = history->currentState();
  const undo::UndoState* s = history->firstState();
  const bool canRedo = history->canRedo();
  bool pastCurrent = !currentState && canRedo;

  int undoSteps = 0;
  int redoSteps = 0;
  while (s) {
    if (pastCurrent && canRedo)
      redoSteps++;
    else if (currentState || !canRedo)
      undoSteps++;

    if (s == currentState || !currentState)
      pastCurrent = true;

    s = s->next();
  }

  lua_newtable(L);
  setfield_integer(L, "undoSteps", undoSteps);
  setfield_integer(L, "redoSteps", redoSteps);
  return 1;
}

int Sprite_get_isValid(lua_State* L)
{
  lua_pushboolean(L, may_get_docobj<Sprite>(L, 1) ? true : false);
  return 1;
}

const luaL_Reg Sprite_methods[] = {
  { "__eq",              Sprite_eq                },
  { "resize",            Sprite_resize            },
  { "crop",              Sprite_crop              },
  { "saveAs",            Sprite_saveAs            },
  { "saveCopyAs",        Sprite_saveCopyAs        },
  { "close",             Sprite_close             },
  { "loadPalette",       Sprite_loadPalette       },
  { "setPalette",        Sprite_setPalette        },
  { "assignColorSpace",  Sprite_assignColorSpace  },
  { "convertColorSpace", Sprite_convertColorSpace },
  { "flatten",           Sprite_flatten           },
  // Layers
  { "newLayer",          Sprite_newLayer          },
  { "newGroup",          Sprite_newGroup          },
  { "deleteLayer",       Sprite_deleteLayer       },
  // Frames
  { "newFrame",          Sprite_newFrame          },
  { "newEmptyFrame",     Sprite_newEmptyFrame     },
  { "deleteFrame",       Sprite_deleteFrame       },
  // Cel
  { "newCel",            Sprite_newCel            },
  { "deleteCel",         Sprite_deleteCel         },
  // Tag
  { "newTag",            Sprite_newTag            },
  { "deleteTag",         Sprite_deleteTag         },
  // Slices
  { "newSlice",          Sprite_newSlice          },
  { "deleteSlice",       Sprite_deleteSlice       },
  // Tilesets & Tiles
  { "newTileset",        Sprite_newTileset        },
  { "deleteTileset",     Sprite_deleteTileset     },
  { "newTile",           Sprite_newTile           },
  { "deleteTile",        Sprite_deleteTile        },
  { nullptr,             nullptr                  }
};

const Property Sprite_properties[] = {
  { "id",                   Sprite_get_id,                   nullptr                         },
  { "filename",             Sprite_get_filename,             Sprite_set_filename             },
  { "isModified",           Sprite_get_isModified,           nullptr                         },
  { "hasAssociatedFile",    Sprite_get_hasAssociatedFile,    nullptr                         },
  { "width",                Sprite_get_width,                Sprite_set_width                },
  { "height",               Sprite_get_height,               Sprite_set_height               },
  { "colorMode",            Sprite_get_colorMode,            nullptr                         },
  { "colorSpace",           Sprite_get_colorSpace,           Sprite_assignColorSpace         },
  { "spec",                 Sprite_get_spec,                 nullptr                         },
  { "selection",            Sprite_get_selection,            Sprite_set_selection            },
  { "frames",               Sprite_get_frames,               nullptr                         },
  { "palettes",             Sprite_get_palettes,             nullptr                         },
  { "layers",               Sprite_get_layers,               nullptr                         },
  { "cels",                 Sprite_get_cels,                 nullptr                         },
  { "tags",                 Sprite_get_tags,                 nullptr                         },
  { "slices",               Sprite_get_slices,               nullptr                         },
  { "tilesets",             Sprite_get_tilesets,             nullptr                         },
  { "backgroundLayer",      Sprite_get_backgroundLayer,      nullptr                         },
  { "transparentColor",     Sprite_get_transparentColor,     Sprite_set_transparentColor     },
  { "bounds",               Sprite_get_bounds,               nullptr                         },
  { "gridBounds",           Sprite_get_gridBounds,           Sprite_set_gridBounds           },
  { "color",                UserData_get_color<Sprite>,      UserData_set_color<Sprite>      },
  { "data",                 UserData_get_text<Sprite>,       UserData_set_text<Sprite>       },
  { "properties",           UserData_get_properties<Sprite>, UserData_set_properties<Sprite> },
  { "pixelRatio",           Sprite_get_pixelRatio,           Sprite_set_pixelRatio           },
  { "events",               Sprite_get_events,               nullptr                         },
  { "tileManagementPlugin", Sprite_get_tileManagementPlugin, Sprite_set_tileManagementPlugin },
  { "useLayerUuids",        Sprite_get_useLayerUuids,        Sprite_set_useLayerUuids        },
  { "undoHistory",          Sprite_get_undoHistory,          nullptr                         },
  { "isValid",              Sprite_get_isValid,              nullptr                         },
  { nullptr,                nullptr,                         nullptr                         }
};

} // anonymous namespace

DEF_MTNAME(doc::Sprite);

void register_sprite_class(lua_State* L)
{
  using doc::Sprite;
  REG_CLASS(L, Sprite);
  REG_CLASS_NEW(L, Sprite);
  REG_CLASS_PROPERTIES(L, Sprite);
}

}} // namespace app::script
