gallery3/modules/organize/controllers/organize.php

<?php defined("SYSPATH") or die("No direct script access.");
/**
 * Gallery - a web based photo album viewer and editor
 * Copyright (C) 2000-2009 Bharat Mediratta
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or (at
 * your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street - Fifth Floor, Boston, MA  02110-1301, USA.
 */
class Organize_Controller extends Controller {
  function dialog($item_id) {
    $item = ORM::factory("item", $item_id);
    $root = $item->id == 1 ? $item : ORM::factory("item", 1);
    access::required("view", $item);
    access::required("edit", $item);

    $v = new View("organize_dialog.html");
    $v->title = $item->title;
    $parents = array();
    foreach ($item->parents() as $parent) {
      $parents[$parent->id] = 1;
    }
    $parents[$item->id] = 1;

    $v->album_tree = self::_tree($root, $parents);
    $v->micro_thumb_grid = self::_get_micro_thumb_grid($item, 0);
    print $v;
  }

  function content($item_id, $offset) {
    $item = ORM::factory("item", $item_id);
    access::required("view", $item);
    access::required("edit", $item);
    print self::_get_micro_thumb_grid($item, $offset);
  }

  function move($target_id) {
    access::verify_csrf();

    $task_def = Task_Definition::factory()
      ->callback("Organize_Controller::move_task_handler")
      ->description(t("Move images"))
      ->name(t("Move Images"));
    $task = task::create($task_def, array("target_id" => $target_id,
                                          "source_ids" => $this->input->post("source_ids")));

    print json_encode(
      array("result" => "started",
            "status" => $task->status,
            "url" => url::site("organize/run/$task->id?csrf=" . access::csrf_token())));
  }

  function rearrange($target_id, $before_or_after) {
    access::verify_csrf();
    $target = ORM::factory("item", $target_id);
    $parent = $target->parent();
    access::required("view", $parent);
    access::required("edit", $parent);

    $task_def = Task_Definition::factory()
      ->callback("Organize_Controller::rearrange_task_handler")
      ->description(t("Rearrange Image"))
      ->name(t("Rearrange Images"));
    $task = task::create(
      $task_def,
      array("target_id" => $target_id,
            "before_or_after" => $before_or_after,
            "parent_id" => $parent->id,
            "source_ids" => $this->input->post("source_ids")));

    print json_encode(
      array("result" => "started",
            "status" => $task->status,
            "url" => url::site("organize/run/$task->id?csrf=" . access::csrf_token())));
  }

  private static function _get_micro_thumb_grid($item, $offset) {
    $v = new View("organize_thumb_grid.html");
    $v->item = $item;
    $v->offset = $offset;
    return $v;
  }

  /**
   * Run the task
   */
  function run($task_id) {
    access::verify_csrf();

    $task = ORM::factory("task", $task_id);
    if (!$task->loaded || $task->owner_id != user::active()->id) {
      access::forbidden();
    }

    $task = task::run($task_id);
    $results = array("done" => $task->done, "status" => $task->status,
                     "percent_complete" => $task->percent_complete);
    foreach (array("tree", "content") as $data) {
      $value = $task->get($data, false);
      if ($value !== false) {
        $results[$data] = $value;
      }
    }
    print json_encode($results);
  }

  private static function _tree($item, $parents) {
    $v = new View("organize_tree.html");
    $v->album = $item;
    $keys = array_keys($parents);
    $v->selected = end($keys) == $item->id;
    $v->can_edit= access::can("edit", $item);
    $v->children = array();
    $v->album_icon = "gBranchEmpty";

    $albums = $item->children(null, 0, array("type" => "album"), array("title" => "ASC"));
    foreach ($albums as $album) {
      if (access::can("view", $album)) {
        $v->children[] = self::_tree($album, $parents);
      }
    }
    if (count($v->children)) {
      $v->album_icon = empty($parents[$item->id]) ? "ui-icon-plus" : "ui-icon-minus";
    }
    return $v;
  }

  static function move_task_handler($task) {
    $start = microtime(true);
    if ($task->percent_complete == 0) {
      batch::start();
    }

    $target = ORM::factory("item", $task->get("target_id"));
    $source_ids = $task->get("source_ids", array());
    $idx = $task->get("current", 0);
    $count = 0;
    for (; $idx < count($source_ids) && microtime(true) - $start < 0.5; $idx++) {
      item::move(ORM::factory("item", $source_ids[$idx]), $target);
      $count++;
    }
    $task->set("current", $idx);
    $task->percent_complete = (int)($idx / count($source_ids) * 100);
    $task->status = t2("Moved one file", "Moved %count files", $count);
    if ($task->percent_complete == 100) {
      batch::stop();
      $task->done = true;
      $task->state = "success";
      $parents = array();
      foreach ($target->parents() as $parent) {
        $parents[$parent->id] = 1;
      }
      $parents[$target->id] = 1;
      // @TODO do we want to set a flag and then generate them in the run method so we don't
      // potentially store large data items in the task?
      $task->set("tree", self::_tree(ORM::factory("item", 1), $parents)->__toString());
      $task->set("content", self::_get_micro_thumb_grid($target, 0)->__toString());
    }
  }

  static function rearrange_task_handler($task) {
    $start = microtime(true);
    $mode = $task->get("mode", "init");

    if ($task->percent_complete == 0) {
      batch::start();
    }
    while (microtime(true) - $start < 1.5) {
      switch ($mode) {
      case "init":
        $album = ORM::factory("item", $task->get("parent_id"));
        if ($album->sort_column != "weight") {
          $mode = "convert-to-weight-order";
        } else {
          $mode = "find-insertion-point";
        }
        break;

      case "convert-to-weight-order":
        $i = 0;
        $album = ORM::factory("item", $task->get("parent_id"));
        foreach ($album->children() as $child) {
          // Do this directly in the database to avoid sending notifications
          Database::Instance()->update("items", array("weight" => $i++), array("id" => $child->id));
        }
        $album->sort_column = "weight";
        $album->sort_order = "ASC";
        $album->save();
        $mode = "find-insertion-point";
        $task->percent_complete = 25;
        break;

      case "find-insertion-point":
        $target = ORM::factory("item", $task->get("target_id"));
        $target_weight = $target->weight;

        if ($task->get("before_or_after") == "after") {
          $target_weight++;
        }
        $task->set("target_weight", $target_weight);
        $task->percent_complete = 40;
        $mode = "make-a-hole";
        break;

      case "make-a-hole":
        $target_weight = $task->get("target_weight");
        $source_ids = $task->get("source_ids");
        $count = count($source_ids);
        $parent_id = $task->get("parent_id");
        Database::Instance()->query(
          "UPDATE {items} " .
          "SET `weight` = `weight` + $count " .
          "WHERE `weight` > $target_weight AND `parent_id` = {$parent_id}");
        $mode = "insert-source-items";
        $task->percent_complete = 80;
        break;

      case "insert-source-items":
        $target_weight = $task->get("target_weight");
        foreach ($source_ids as $source_id) {
          Database::Instance()->update(
            "items", array("weight" => $target_weight++), array("id" => $source_id));
        }
        $mode = "done";
        break;

      case "done":
        $album = ORM::factory("item", $task->get("parent_id"));
        module::event("album_rearrange", $album);
        batch::stop();
        $task->done = true;
        $task->state = "success";
        $task->percent_complete = 100;
        $task->set("content", self::_get_micro_thumb_grid($album, 0)->__toString());
        break;
      }
    }

    $task->set("mode", $mode);
  }
}