# -*- coding: utf-8 -*-
#
# Copyright 2013 whiteleaf. All rights reserved.
#

require_relative "../database"
require_relative "../downloader"
require_relative "../inventory"

module Command
  class Tag < CommandBase
    COLORS = %w(green yellow blue magenta cyan red white)
    # 禁止文字
    BAN_CHAR = /[:;"'><$@&^\\\|%'\/`]/
    # 禁止ワード
    BAN_WORD = %w(hotentry)

    def self.oneline_help
      "各小説にタグを設定及び閲覧が出来ます"
    end

    def initialize
      super("<option> <tagname> <target> [<target2> ...]\n" \
            "<tagname> [<tagname2> ...]")
      @opt.separator <<-EOS

  ・小説にタグを設定します。設定個数の上限はありません
  ・タグ名にはスペース以外の文字が使えます(大文字小文字区別)
  ・タグには自動で色がつきます。自分で指定する場合は--colorを指定して下さい
  ・一部特殊なタグがあります。設定することでlistコマンドに反映されます
      - end: 小説が完結状態
      - 404: 掲載サイトから削除された状態
  ・設定したタグは他のコマンドで指定することで、小説ID指定の代わりにすることができます(一部タグを使えないコマンドも存在します。またaliasでつけた別名よりもタグのほうが優先されます)

  Examples:
    narou tag --add fav 0 2     # ID:0と2の小説にfavタグを設定(追加)
    narou t -a fav 0 2          # もしくはこの様に書けます
    narou t -a "fav later" 0 2  # 一度に複数のタグを指定出来ます
    narou t -a fav -c red 0     # favというタグを赤色で設定する
    narou tag --delete fav 2    # ID:2の小説のfavタグを外す
    narou t -d fav 2

    narou tag end               # endタグ(完結)の付いている小説の一覧を表示
    narou tag fav later         # fav,laterタグ両方付いている小説を表示
    narou list -t "fav later" -rl   # listコマンドでもタグで検索出来ます
    narou tag                   # 何も指定しない場合、存在するタグ一覧を表示

    # 他のコマンドでタグを使う
    narou freeze --on end       # end(完結タグ)が付いた小説を一括で凍結
    narou update tag:404        # 404タグを明示する
                                # ID:404 が存在しなければ明示する必要はない

  Options:
      EOS
      @opt.on("-a", "--add TAGS", String, "タグを追加する") { |tags|
        @options["mode"] = :add
        @options["tags"] = tags.split.tap { |array|
          array.each do |tag|
            if tag =~ BAN_CHAR
              error "#{tag} に使用禁止記号が含まれています"
              exit Narou::EXIT_ERROR_CODE
            elsif BAN_WORD.include?(tag)
              error "#{tag} は使用禁止ワードです"
              exit Narou::EXIT_ERROR_CODE
            end
          end
        }
      }
      @opt.on("-d", "--delete TAGS", String, "タグを外す") { |tags|
        @options["mode"] = :delete
        @options["tags"] = tags.split
      }
      @opt.on("-c", "--color COL", String,
              "タグの色を自分で指定する\n" \
              "#{' '*25}COL=#{get_color_list}") { |color|
        color.downcase!
        unless COLORS.include?(color)
          error "#{color}という色は存在しません。色指定は無視されます"
          color = nil
        end
        @options["color"] = color
      }
      @opt.on("--clear", "指定した小説のタグをすべて外す") {
        @options["mode"] = :clear
      }
    end

    #
    # タグ一覧取得
    #
    # ids: 特定の小説のタグだけ取得するためのIDの配列を指定
    # @return: { "タグの名前" => タグが付けられた小説の数, ... }
    #
    def self.get_tag_list(ids = nil)
      database = Database.instance
      tag_list = Hash.new(0)
      database.each_value do |data|
        if ids.kind_of?(Array)
          next unless ids.include?(data["id"])
        end
        tags = data["tags"] || []
        tags.each do |tag|
          tag_list[tag] += 1
        end
      end
      tag_list.default = nil
      tag_list
    end

    def get_color_list
      COLORS.map { |color|
        "<bold><#{color}>#{color}</#{color}></bold>"
      }.join(",").termcolor
    end

    def execute(argv)
      # タグの色がすでに決まっていた場合は変更しない隠しオプション
      no_overwrite_color = !!argv.delete("--no-overwrite-color")
      super
      @options["no-overwrite-color"] = no_overwrite_color   # super で @options.clear されるのでここで代入
      color_changed = change_colors
      @options["mode"] ||= :list
      if argv.empty?
        if @options["mode"] == :list
          display_taglist
          return
        end
        if color_changed
          puts "タグの色を変更しました"
          display_taglist
          return
        else
          error "対象の小説を指定して下さい"
          exit Narou::EXIT_ERROR_CODE
        end
      else
        if @options["mode"] == :list
          search_novel_by_tag(argv)
        else
          edit_tags(argv)
        end
      end
    end

    def display_taglist
      database = Database.instance
      tag_list = Tag.get_tag_list
      puts "タグ一覧"
      puts tag_list.map { |tag, count|
        color = Tag.get_color(tag)
        "<bold><#{color}>#{TermColorLight.escape(tag)}(#{count})</#{color}></bold>"
      }.join(" ").termcolor
    end

    def search_novel_by_tag(argv)
      List.execute!("--tag", argv.join(" "))
    end

    def change_colors
      changed = false
      if @options["color"] && @options["tags"]
        @options["tags"].each do |tag|
          set_color(tag, @options["color"])
        end
        changed = true
      end
      changed
    end

    def edit_tags(argv)
      database = Database.instance
      tagname_to_ids(argv)
      argv.each do |target|
        data = Downloader.get_data_by_target(target)
        unless data
          error "#{target} は存在しません"
          next
        end
        tags = data["tags"] || []
        title = data["title"]
        case @options["mode"]
        when :add
          tags |= @options["tags"]
          puts "#{title} にタグを設定しました"
        when :delete
          tags -= @options["tags"]
          puts "#{title} からタグを外しました"
        when :clear
          tags.clear
          puts "#{title} のタグをすべて外しました"
        end
        if tags.size > 0
          print "現在のタグは "
          print tags.map { |tagname|
            color = Tag.get_color(tagname)
            "<bold><#{color}>#{TermColorLight.escape(tagname)}</#{color}></bold>"
          }.join(" ").termcolor
          puts " です"
        end
        database[data["id"]]["tags"] = tags
      end
      database.save_database
    end

    def self.get_color(tagname)
      tag_colors = Inventory.load("tag_colors")
      color = tag_colors[tagname]
      return color if color
      last_color = tag_colors.values.last || COLORS.last
      index = (COLORS.index(last_color) + 1) % COLORS.size
      color = COLORS[index]
      tag_colors[tagname] = color
      tag_colors.save
      color
    end

    def set_color(tagname, color)
      tag_colors = Inventory.load("tag_colors")
      if @options["no-overwrite-color"]
        return if tag_colors.include?(tagname)
      end
      tag_colors[tagname] = color
      tag_colors.save
    end
  end
end