# frozen_string_literal: true

#
# Copyright 2013 whiteleaf. All rights reserved.
#

module Command
  class Web < CommandBase
    def self.oneline_help
      "WEBアプリケーション用サーバを起動します"
    end

    def initialize
      super("[options...]")
      @opt.separator <<-EOS

  ・WEBアプリケーション用サーバを起動します
  ・小説の管理及び設定をブラウザで行うことができます
  ・--port を指定しない場合、ポートは初回起動時にランダムで設定します
    (以降同じ設定を引き継ぎます)
  ・サーバ起動後にブラウザを立ち上げます
  ・サーバの停止はコンソールで Ctrl+C を入力します

  Examples:
    narou web   # サーバ起動(ポートはランダム。ポート設定保存)
    narou web -p 4567   # ポート4567で起動(保存はされない)

    # 先に決めておく
    narou s server-port=8000
    narou web   # ポート8000で起動

  Options:
      EOS
      @opt.on("-p", "--port PORT", Integer, "起動するポートを指定") { |port|
        @options["port"] = port
      }
      @opt.on("-n", "--no-browser", "起動時にブラウザは開かない") {
        @options["no-browser"] = true
      }
    end

    def confirm_of_first
      setting = Inventory.load("server_setting", :global)
      is_first = !setting["already-server-boot"]
      if is_first
        puts <<-EOS
初めてサーバを起動します。ファイアウォールのアクセス許可を尋ねられた場合、許可をして下さい。
また、起動したサーバを止めるにはコンソール上で Ctrl+C を入力するか、ブラウザ上で「設定(歯車マーク)→サーバをシャットダウン」を実行して下さい。

        EOS
        if @options["no-browser"]
          puts "(何かキーを押して下さい)"
        else
          puts "(何かキーを押して下さい。サーバ起動後ブラウザが立ち上がります)"
        end
        $stdin.getch
        setting["already-server-boot"] = true
        setting.save
      end
      is_first
    end

    def create_push_server(params)
      host, port = params[:host], params[:port]
      push_server = Narou::PushServer.instance
      accepted_domains = (host == "0.0.0.0" ? "*" : host)
      if accepted_domains != "*"
        global_setting = Inventory.load("global_setting", :global)
        addtional_accepted_domains = global_setting["server-ws-add-accepted-domains"]
        if addtional_accepted_domains
          accepted_domains = [
            accepted_domains,
            addtional_accepted_domains.split(",").map(&:strip)
          ].flatten
        end
      end
      push_server.accepted_domains = accepted_domains
      push_server.port = port + 1
      push_server.host = host
      push_server
    end

    def execute(argv)
      if argv.delete("--boot")
        @rebooted = !!argv.delete("--reboot")
        super
        boot
      else
        argv << "--backtrace" if $display_backtrace
        argv << "--no-color" if $disable_color
        argv << "--boot"
        argv_copy = argv.dup
        kill_threads
        begin
          loop do
            if $development
              system(RbConfig.ruby, "-x", $0, "web", *argv)
            else
              system("narou", "web", *argv)
            end
            break unless $?.exitstatus == Narou::EXIT_REQUEST_REBOOT
            argv = argv_copy.dup
            argv.push("--no-browser", "--reboot")
          end
        rescue Interrupt => e
          # 中断されてコンソールへの入力が可能になってから、WEBrick が終了するまで
          # タイムラグがあって表示がごちゃまぜになるので、終わるのを少し待つ
          sleep 1
        end
      end
    end

    def kill_threads
      Narou::Worker.stop
    end

    def boot
      require_relative "../web/all"
      confirm_of_first
      params = Narou::AppServer.create_address(@options["port"])
      push_server = create_push_server(params)
      Narou.web = true
      Thread.abort_on_exception = true

      address = "http://#{params[:host]}:#{params[:port]}/"
      puts address
      puts "サーバを止めるには Ctrl+C を入力"
      puts

      push_server.run
      open_browser_when_server_boot(address)
      send_rebooted_event_when_connection_recover(push_server)

      $stdout = Narou::StreamingLogger.new(push_server)
      $stdout2 = if Inventory.load["concurrency"]
                   Narou::StreamingLogger.new(push_server, $stdout2, target_console: "stdout2")
                 else
                   $stdout
                 end
      ProgressBar.push_server = push_server
      Narou::Worker.push_server = push_server
      Narou::AppServer.push_server = push_server
      Narou::WebWorker.run
      Narou::AppServer.run!
      push_server.quit
      Narou::WebWorker.stop
      Narou::Worker.stop
      if Narou::AppServer.request_reboot?
        exit Narou::EXIT_REQUEST_REBOOT
      end
    rescue Errno::EADDRINUSE => e
      Helper.open_browser(address) unless @options["no-browser"]
      STDOUT.puts <<-EOS
#{e}
ポートが使われています。サーバがすでに立ち上がっているかどうか確認して下さい。
他のアプリケーションが使っているポートだった場合、ポートを変更して下さい。

ポートの変更方法
  $ narou s server-port=5678
      EOS
      exit Narou::EXIT_ERROR_CODE
    end

    def open_browser_when_server_boot(address)
      return if @options["no-browser"]
      Thread.new do
        sleep 0.2 until Narou::AppServer.running?
        Helper.open_browser(address)
      end
    end

    def send_rebooted_event_when_connection_recover(push_server)
      return unless @rebooted
      Thread.new do |th|
        timeout = Time.now + 20
        # WebSocketのコネクションが回復するまで待つ
        until push_server.connections.count != 0
          sleep 0.2
          th.kill if Time.now > timeout
        end
        puts "<yellow>再起動が完了しました。</yellow>".termcolor
        push_server.send_all(:"server.rebooted")
      end
    end

  end
end