require 'fusefs'
require 'rubygems'
require 'couchrest'
 
DEBUG = false #true
 
class CouchdbDir
  def initialize()
  	@db = CouchRest.database!("http://127.0.0.1:5984/couchfs")
		create_view()
		create_root()
  end
 
	def create_view
      @view = {}
			@view['path'] = {'map' => "function(doc) {
            if (doc['path']) {
						if(doc['path'] == \"/\")
						{
              emit(doc['path'], null);
						}
						else
						{
              emit(doc['path'].replace(/\\/$/, \"\"), null);
            }
				}
          }"
			}
 
			@view['base_path'] = {'map' => "function(doc) {
            if (doc['base_path']) {
						if(doc['base_path'] == \"/\")
						{
              emit(doc['base_path'], null);
						}
						else
						{
              emit(doc['base_path'].replace(/\\/$/, \"\"), null);
            }
				}
          }"
			}
      @db.save({
        "_id" => "_design/doc",
        :views => @view
      })
	end
 
	#create the root dir
	def create_root
		if @db.view("doc/path", :key => "/")['rows'].nitems == 0
	    ret = {"path" => "/", "name" => "/", "files" => []}
			@db.save(ret)
		end
	end
 
  def contents(path)
		p "call:contents?(#{path})" if DEBUG
		files = @db.view("doc/base_path", :key => path)
		p files if DEBUG
		ret = []
		files['rows'].each do |fil|
			ret << @db.get(fil['id'])['name']
		end
		return ret
  end
  def directory?(path)
		dir = @db.view("doc/path", :key => path)['rows']
		if dir.nitems > 0
			return !@db.get(dir[0]['id']).key?("size")
		else
			return false
		end
  end
 
  def file?(path)
		dir = @db.view("doc/path", :key => path)['rows']
		if dir.nitems > 0
			return @db.get(dir[0]['id']).key?("size")
		else
			return false
		end
  end
 
  def touch(path)
    write_to(path, "")
  end
  def read_file(path)
    ret = @db.view("doc/path", :key => path)
		if ret['rows'].nitems > 0
			file = @db.get(ret['rows'][0]['id'])
			if file.key?("size")
				return @db.fetch_attachment(file["_id"], file["name"])
			end
		end
		return false
  end
  def size(path)
		ret = @db.view("doc/path", :key => path)
		if ret['rows'].nitems > 0
			file = @db.get(ret['rows'][0]['id'])
			if file.key?("size")
				return file["size"]
			else
				#directory
				return 0
			end
		end
		return 0
  end
 
  # File writing
  def can_write?(path)
		return @db.view("doc/path", :key => path)['rows'].nitems == 0
  end
	def write_to(path, body)
		name = File.basename(path)
		path_only = path.sub(/#{name}$/, "")
    file = {"path" => path, "base_path" => path_only, "_attachments" => { name => {"data" => body}}, "name" => name, "size" => body.length, "type" => "f"}
		@db.save(file)
	end
 
  # Delete a file
  def can_delete?(path)
		rows = @db.view("doc/path", :key => path)['rows']
		if rows.nitems > 0
			file = @db.get(rows[0]['id'])
			if file.key?("size")
				return true
			end
		end
		false
  end
  def delete(path)
		row = @db.view("doc/path", :key => path)['rows']
		if row.nitems > 0
			file = @db.get(row[0]['id'])
			@db.delete(file)
		end
  end
 
  def can_mkdir?(path)
		return @db.view("doc/path", :key => path)['rows'].nitems == 0
  end
  def mkdir(path)
		name = File.basename(path)
		path_only = path.sub(/#{name}$/, "")
    dir = {"path" => path, "base_path" => path_only, "name" => name}
		@db.save(dir)
	end
 
  # rmdir
  def can_rmdir?(path)
		if path != "/"
			if @db.view("doc/base_path", :key => path)['rows'].nitems == 0
				row = @db.view("doc/path", :key => path)['rows']
				if row.nitems > 0
					file = @db.get(row[0]['id'])
					if !file.key?('size')
						return true
					end
				end
			end
		end
		return false
  end
  def rmdir(path)
    rows = @db.view("doc/path", :key => path)['rows']
		if rows.nitems > 0
			dir = @db.get(rows[0]['id'])
			if !dir.key?("size")
				@db.delete(dir)
			end
		end
  end
 
	def delete_db
		@db.delete!
	end
end