# -*- coding: utf-8 -*-
#
# @file 
# @brief
# @author ongaeshi
# @date   2011/02/20

require 'yaml'
require 'pathname'
require 'milkode/common/dbdir'

module Milkode
  class CdstkYaml
    class YAMLAlreadyExist < RuntimeError
    end
    
    class YAMLNotExist < RuntimeError
    end

    def self.create(path = ".")
      yf = yaml_file(path)
      raise YAMLAlreadyExist.new if FileTest.exist? yf
      obj = CdstkYaml.new(yf, {'contents' => [], 'version' => 0.1})
      obj.save
      return obj
    end

    def self.load(path = ".")
      yf = yaml_file(path)
      raise YAMLNotExist.new unless FileTest.exist? yf
      open(yf) do |f|
        return CdstkYaml.new(yf, YAML.load(f.read()))
      end
    end

    def initialize(yaml_file, data)
      @yaml_file = yaml_file
      @data = data
      normalize
    end

    def normalize
      if (Util::platform_win?)
        contents.each do |v|
          v['directory'] = Util::normalize_filename v['directory']
        end
      end
    end

    def add(dirs)
      contents.concat(dirs.map{|v|{'directory' => v}})
      contents.uniq!
    end

    def remove(query)
      r = query.select_any?(contents)
      r.each {|v| contents.delete v}
    end

    def save
      open(@yaml_file, "w") { |f| YAML.dump(@data, f) }
    end

    def contents
      @data['contents']
    end

    def package_num
      @data['contents'].size
    end

    def directorys
      contents.map{|v|v['directory']}
    end

    def version
      @data['version']
    end

    def list(query = nil)
      query ? query.select_all?(contents) : contents
    end

    def exist?(shortname)
      @data['contents'].find {|v| File.basename(v['directory']) == shortname }   
    end

    def cant_add_directory?(dir)
      @data['contents'].find {|v|
        v['directory'] != File.expand_path(dir) &&
        File.basename(v['directory']) == File.basename(dir)
      }
    end

    def cleanup
      contents.delete_if do |v|
        if (!File.exist? v['directory'])
          yield v if block_given?
          true
        else
          false
        end
      end
    end

    def package_root(dir)
      nd = Util::normalize_filename dir
      @data['contents'].find do |v|
        v if nd =~ /^#{v['directory']}/
      end
    end

    def package_root_dir(dir)
      package = package_root(dir)
      (package) ? package['directory'] : nil
    end

    def self.yaml_file(path)
      Dbdir.yaml_path(path)
    end

    class Query
      def initialize(keywords)
        @keywords = keywords
      end

      def select_any?(contents)
        contents.find_all do |v|
          @keywords.any? {|s| File.basename(v['directory']).include? s }
        end
      end

      def select_all?(contents)
        contents.find_all do |v|
          @keywords.all? {|s| File.basename(v['directory']).include? s }
        end
      end
    end
  end
end