gallery3/modules/local_import/controllers/local_import.php

<?php defined("SYSPATH") or die("No direct script access.");
/**
 * Gallery - a web based photo album viewer and editor
 * Copyright (C) 2000-2008 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 Local_Import_Controller extends Controller {
  public function index($id) {
    $paths = unserialize(module::get_var("local_import", "authorized_paths"));

    $item = ORM::factory("item", $id);
    access::can("local_import", $item);

    $view = new View("local_import_tree_dialog.html");
    $view->action = url::site("local_import/add/$id");
    $view->hidden = array("csrf" => access::csrf_token(), "base_url" => url::base(true));
    $view->parents = $item->parents();
    $view->album_title = $item->title;

    $tree = new View("local_import_tree.html");
    $tree->data = array();
    $tree->uid = "tree_$id";
    foreach (array_keys($paths) as $path) {
      $tree->data[$path] = array("path" => $path, "is_dir" => true);
    }
    $view->tree = $tree->__toString();
    print $view;
  }

  public function children() {
    access::verify_csrf();

    $path = $this->input->post("path");
    $path =  implode("/", $this->input->post("path"));
    if (!is_readable($path)) {
      kohana::show_404();
    }

    $tree = new View("local_import_tree.html");
    $tree->data = $this->_get_children($path);
    $tree->uid = "tree_" . md5($path);
    print $tree;
  }

  function add($id) {
    access::verify_csrf();

    $parent = ORM::factory("item", $id);
    access::can("local_import", $item);
    if (!$parent->is_album() && !$parent->loaded ) {
      throw new Exception("@todo BAD_ALBUM");
    }

    $path = $this->input->post("path");

    $source_path = $path[0];
    for ($i = 1; $i < count($path); $i++) {  // skip the first path
      $source_path .= "/$path[$i]";
      $pathinfo = pathinfo($source_path);
      set_time_limit(30);
      if (is_dir($source_path)) {
        $album = ORM::factory("item")
          ->where("name", $path[$i])
          ->where("parent_id", $parent->id)
          ->find();
        if (!$album->loaded) {
          $album = album::create($parent, $path[$i], $path[$i], null, user::active()->id);
        }
        $parent = $album;
      } else if (in_array($pathinfo["extension"], array("flv", "mp4"))) {
        $movie = movie::create($parent, $source_path, basename($source_path),
                               basename($source_path), null, user::active()->id);
      } else {
        $photo = photo::create($parent, $source_path, basename($source_path),
                        basename($source_path), null, user::active()->id);
      }
    }
  }

  private function _get_children($path) {
    $file_list = array();
    $files = scandir($path);
    foreach ($files as $file) {
      if ($file[0] != ".") {
        $full_path = "$path/$file";
        if (is_dir($full_path)) {
          $file_list[$file] =
            array("path" => $full_path, "is_dir" => true);
        } else {
          $extension = strtolower(substr(strrchr($file, '.'), 1));
          // Make sure the file is readable
          if (is_readable($full_path) &&
              in_array($extension, array("gif", "jpeg", "jpg", "png", "flv", "mp4"))) {
            $file_list[$file] = array("path" => $full_path, "is_dir" => false);
          }
        }
      }
    }
    return $file_list;
  }
}