w3dhub_linux_launcher/lib/pages/server_browser.rb

class W3DHub
  class Pages
    class ServerBrowser < Page
      def setup
        @server_locked_icons = {}
        @refresh_server_list = false
        refresh_server = false

        @selected_server ||= nil
        @selected_server_container ||= nil
        @selected_color = 0xaa_666655

        @filters = Store.settings[:server_list_filters] || {}
        @filter_region = Store.settings[:server_list_region] || "Any" # "Any", "North America", "Europe"

        Store.applications.games.each { |game| @filters[game.id.to_sym] = true if @filters[game.id.to_sym].nil? }

        @ping_icons = {}
        generate_ping_icons

        body.clear do
          stack(width: 1.0, height: 1.0, padding: 8) do
            background 0xaa_252525

            stack(width: 1.0, height: 22) do
              para "<b>#{I18n.t(:"server_browser.filters")}</b>", font: BOLD_FONT
            end

            flow(width: 1.0, height: 36) do
              flow(width: 128, height: 1.0) do
                # para I18n.t(:"server_browser.region"), width: 0.5
                list_box items: ["Any", "North America", "Europe", "Asia"], choose: Store.settings[:server_list_region], width: 1.0, height: 1.0, padding_top: 4, padding_bottom: 4 do |value|
                  @filter_region = value
                  Store.settings[:server_list_region] = @filter_region
                  Store.settings.save_settings

                  populate_server_list
                end
              end

              flow(fill: true, height: 1.0) do
                @filters.each do |app_id, enabled|
                  app = Store.applications.games.find { |a| a.id == app_id.to_s }
                  next unless app

                  image_path = File.exist?("#{CACHE_PATH}/#{app.id}.png") ? "#{CACHE_PATH}/#{app.id}.png" : "#{GAME_ROOT_PATH}/media/icons/default_icon.png"

                  image image_path, tip: "#{app.name}", height: 1.0,
                        border_thickness_bottom: 1, border_color_bottom: 0x00_000000,
                        color: enabled ? 0xff_ffffff : 0xff_444444, hover: { border_color_bottom: 0xff_aaaaaa }, margin_left: 16 do |img|
                    @filters[app_id] = !@filters[app_id]
                    Store.settings[:server_list_filters] = @filters
                    Store.settings.save_settings

                    if @filters[app_id]
                      img.style.color = 0xff_ffffff
                      img.style.default[:color] = 0xff_ffffff
                    else
                      img.style.color = 0xff_444444
                      img.style.default[:color] = 0xff_444444
                    end

                    populate_server_list
                  end
                end
              end

              flow(min_width: 372, width: 0.38, max_width: 512, height: 1.0) do |container|
                button "Direct Connect", height: 1.0, padding_top: 4, padding_bottom: 4 do
                  push_state(W3DHub::States::DirectConnectDialog)
                end

                flow(fill: true)

                para "#{I18n.t(:"server_browser.nickname")}:"
                @nickname_label = para "#{Store.settings[:server_list_username]}"
                image "#{GAME_ROOT_PATH}/media/ui_icons/wrench.png", height: 16, hover: { color: 0xaa_ffffff }, tip: I18n.t(:"server_browser.set_nickname") do
                  # Prompt for player name
                  W3DHub.prompt_for_nickname(
                    accept_callback: proc do |entry|
                      @nickname_label.value = entry
                      Store.settings[:server_list_username] = entry
                      Store.settings.save_settings

                      container.recalculate
                      container.recalculate
                      container.recalculate
                    end
                  )
                end
              end
            end

            flow(width: 1.0, fill: true, margin_top: 16) do
              stack(fill: true, height: 1.0) do
                # Icon
                # Hostname
                # Current Map
                # Players
                # Ping
                flow(width: 1.0, height: 24) do
                  stack(width: 56, padding: 4) do
                  end

                  stack(width: 0.45, height: 1.0) do
                    para "<b>#{I18n.t(:"server_browser.hostname")}</b>", text_wrap: :none, width: 1.0, font: BOLD_FONT
                  end

                  flow(fill: true, height: 1.0) do
                    para "<b>#{I18n.t(:"server_browser.current_map")}</b>", text_wrap: :none, width: 1.0, font: BOLD_FONT
                  end

                  flow(width: 0.11, height: 1.0) do
                    para "<b>#{I18n.t(:"server_browser.players")}</b>", text_wrap: :none, width: 1.0, font: BOLD_FONT
                  end

                  stack(width: 56) do
                    para "<b>#{I18n.t(:"server_browser.ping")}</b>", text_wrap: :none, width: 1.0, font: BOLD_FONT
                  end
                end

                @server_list_container = stack(width: 1.0, fill: true, scroll: true) do
                  para I18n.t(:"server_browser.fetching_server_list")
                end
              end

              @game_server_info_container = stack(min_width: 372, width: 0.38, max_width: 512, height: 1.0) do
                para I18n.t(:"server_browser.no_server_selected"), width: 1.0, text_align: :center
              end
            end
          end
        end

        populate_server_list
        populate_server_info(@selected_server) if @selected_server
      end

      def update
        super

        if @refresh_server_list && Gosu.milliseconds >= @refresh_server_list
          @refresh_server_list = nil

          # populate_server_list
          reorder_server_list

          if @selected_server&.id == @refresh_server&.id
            if @refresh_server
              BackgroundWorker.foreground_job(
                -> { fetch_server_details(@refresh_server) },
                ->(result) {
                  populate_server_info(@refresh_server) if @refresh_server == @selected_server
                  @refresh_server = nil
                }
              )
            end
          end
        end
      end

      def generate_ping_icons
        signal3  = get_image("#{GAME_ROOT_PATH}/media/ui_icons/signal3.png")
        signal2  = get_image("#{GAME_ROOT_PATH}/media/ui_icons/signal2.png")
        signal1  = get_image("#{GAME_ROOT_PATH}/media/ui_icons/signal1.png")
        question = get_image("#{GAME_ROOT_PATH}/media/ui_icons/question.png")

        good = Gosu.render(signal3.width, signal3.height) do
          signal3.draw(0, 0, 0, 1, 1, 0xff_008000)
        end

        fair = Gosu.render(signal3.width, signal3.height) do
          signal3.draw(0, 0, 0, 1, 1, 0xff_444444)
          signal2.draw(0, 0, 0, 1, 1, 0xff_804000)
        end

        poor = Gosu.render(signal3.width, signal3.height) do
          signal3.draw(0, 0, 0, 1, 1, 0xff_444444)
          signal1.draw(0, 0, 0, 1, 1, 0xff_800000)
        end

        bad = Gosu.render(signal3.width, signal3.height) do
          signal3.draw(0, 0, 0, 1, 1, 0xff_444444)
        end

        unknown = Gosu.render(signal3.width, signal3.height) do
          signal3.draw(0, 0, 0, 1, 1, 0xff_222222)
          question.draw(0, 0, 0, 1, 1, 0xff_888888)
        end

        @ping_icons[:good] = good
        @ping_icons[:fair] = fair
        @ping_icons[:poor] = poor
        @ping_icons[:bad]  = bad
        @ping_icons[:unknown] = unknown
      end

      def ping_icon(server)
        case server.ping
        when 0..150
          @ping_icons[:good]
        when 151..200
          @ping_icons[:fair]
        when 201..1_000
          @ping_icons[:poor]
        when 1_001..5_000
          @ping_icons[:bad]
        else
          @ping_icons[:unknown]
        end
      end

      def ping_tip(server)
        server.ping == W3DHub::Api::ServerListServer::NO_OR_BAD_PING ? "Ping failed" : "Ping #{server.ping}ms"
      end

      def find_element_by_tag(container, tag, list = [])
        return unless container

        container.children.each do |child|
          list << child if child.style.tag == tag

          find_element_by_tag(child, tag, list) if child.is_a?(CyberarmEngine::Element::Container)
        end

        return list.first
      end

      def refresh_server_list(server, mode = :update) # :remove, :refresh_all
        if mode == :refresh_all
          populate_server_list
          return
        end

        @refresh_server_list = Gosu.milliseconds + 3_000
        @refresh_server = server if @selected_server&.id == server.id

        server_container = find_element_by_tag(@server_list_container, server.id)

        case mode
        when :update
          if server.status && !server_container
            @server_list_container.append do
              create_server_container(server)
            end
          end
        when :remove
          @server_list_container.remove(server_container) if server_container
          return
        end
        return unless server_container

        game_icon      = find_element_by_tag(server_container, :game_icon)
        server_name    = find_element_by_tag(server_container, :server_name)
        server_channel = find_element_by_tag(server_container, :server_channel)
        server_region  = find_element_by_tag(server_container, :server_region)
        server_map     = find_element_by_tag(server_container, :server_map)
        player_count   = find_element_by_tag(server_container, :player_count)
        server_ping    = find_element_by_tag(server_container, :ping)

        game_icon&.value = game_icon(server)
        server_name&.value = "<b>#{server&.status&.name}</b>"
        server_channel&.value = Store.application_manager.channel_name(server.game, server.channel).to_s
        server_region&.value = server.region
        server_map&.value = server&.status&.map
        player_count&.value = "#{server&.status&.player_count}/#{server&.status&.max_players}"
        server_ping&.value = ping_icon(server)
        server_ping&.parent.parent.tip = ping_tip(server)
      end

      def update_server_ping(server)
        container = find_element_by_tag(@server_list_container, server.id)

        if container
          ping_image = find_element_by_tag(container, :ping)

          if ping_image
            ping_image.value = ping_icon(server)
            ping_image.parent.parent.tip = ping_tip(server)
          end
        end
      end

      def stylize_selected_server(server_container)
        server_container.style.background = @selected_color

        server_container.style.default[:background] = @selected_color
        server_container.style.hover[:background] = @selected_color
        server_container.style.active[:background] = @selected_color
      end

      def reorder_server_list
        @server_list_container.children.sort_by! do |child|
          s = Store.server_list.find { |s| s.id == child.style.tag }

          [s.status.player_count, -s.ping]
        end.reverse!.each_with_index do |child, i|
          next if @selected_server_container && child == @selected_server_container

          child.style.hover[:background] = 0xaa_555566
          child.style.hover[:active] = 0xaa_555588

          child.style.default[:background] = 0xaa_333333 if i.even?
          child.style.default[:background] = 0x00_000000 if i.odd?
        end

        @server_list_container.recalculate
      end

      def populate_server_list
        Store.server_list = Store.server_list.sort_by! { |s| [s.status.player_count, s.id] }.reverse

        @server_list_container.clear do
          Store.server_list.each do |server|
            create_server_container(server)
          end
        end

        reorder_server_list
      end

      def create_server_container(server)
        return unless @filters[server.game.to_sym]
        return unless server.status
        return unless server.region == @filter_region || @filter_region == "Any"
        return unless Store.application_manager.channel_name(server.game, server.channel) # can user access required game and channel for this server?

        server_container = flow(width: 1.0, height: 56, hover: { background: 0xaa_555566 }, active: { background: 0xaa_555588 }, tag: server.id, tip: ping_tip(server)) do
          flow(width: 56, height: 1.0, padding: 4) do
            image game_icon(server), height: 1.0, tag: :game_icon
          end

          stack(width: 0.45, height: 1.0) do
            para server&.status&.name, tag: :server_name, font: BOLD_FONT, text_wrap: :none

            flow(width: 1.0, height: 1.0) do
              para Store.application_manager.channel_name(server.game, server.channel).to_s, width: 172, margin_right: 8, tag: :server_channel
              para server.region, tag: :server_region
            end
          end

          flow(fill: true, height: 1.0) do
            para "#{server&.status&.map}", tag: :server_map
          end

          flow(width: 0.11, height: 1.0) do
            para "#{server&.status&.player_count}/#{server&.status&.max_players}", tag: :player_count
          end

          flow(width: 56, height: 1.0, padding: 4) do
            image ping_icon(server), height: 1.0, tag: :ping
          end
        end

        def server_container.hit_element?(x, y)
          self if hit?(x, y)
        end

        server_container.subscribe(:clicked_left_mouse_button) do
          stylize_selected_server(server_container)

          @selected_server_container = server_container

          @selected_server = server

          reorder_server_list if @selected_server_container

          BackgroundWorker.foreground_job(
            -> { fetch_server_details(server) },
            ->(result) { populate_server_info(server) if server == @selected_server }
          )
        end

        stylize_selected_server(server_container) if server.id == @selected_server&.id
      end

      def populate_server_info(server)
        @game_server_info_container.clear do
          stack(width: 1.0, height: 1.0, padding: 8) do
            stack(width: 1.0, height: 208) do
              flow(width: 1.0, height: 34) do
                flow(fill: true)

                image game_icon(server), height: 1.0
                title server.status.name[0..30], text_wrap: :none

                flow(fill: true)
              end

              flow(width: 1.0, height: 46, margin_top: 16, margin_bottom: 16) do
                game_installed = Store.application_manager.installed?(server.game, server.channel)
                game_updatable = Store.application_manager.updateable?(server.game, server.channel)
                channel = Store.application_manager.channel(server.game, server.channel)
                style = ((channel && channel.user_level.downcase.strip == "public") || server.channel == "release") ? {} : TESTING_BUTTON

                flow(fill: true)
                button "<b>#{I18n.t(:"server_browser.join_server")}</b>", enabled: (game_installed && !game_updatable), **style do
                  # Check for nickname
                  #   prompt for nickname
                  # !abort unless nickname set
                  # Check if password needed
                  #   prompt for password
                  # Launch game
                  if Store.settings[:server_list_username].to_s.length.zero?
                    W3DHub.prompt_for_nickname(
                      accept_callback: proc do |entry|
                        @nickname_label.value = entry
                        Store.settings[:server_list_username] = entry
                        Store.settings.save_settings

                        if server.status.password
                          W3DHub.prompt_for_password(
                            accept_callback: proc do |password|
                              W3DHub.join_server(server, password)
                            end
                          )
                        else
                          W3DHub.join_server(server, nil)
                        end
                      end
                    )
                  else
                    if server.status.password
                      W3DHub.prompt_for_password(
                        accept_callback: proc do |password|
                          W3DHub.join_server(server, password)
                        end
                      )
                    else
                      W3DHub.join_server(server, nil)
                    end
                  end
                end

                if W3DHUB_DEVELOPER
                  list_box(items: (1..12).to_a.map(&:to_s), margin_left: 16, width: 72, tip: "Number of game clients", enabled: (game_installed && !game_updatable), **TESTING_BUTTON)
                  button "Multijoin", tip: "Launch multiple clients with configured username_\#{number}", enabled: (game_installed && !game_updatable), **TESTING_BUTTON
                end

                flow(fill: true)
              end

              # Server Info
              stack(width: 1.0, fill: true, margin_bottom: 16) do
                flow(width: 1.0) do
                  para "<b>#{I18n.t(:"server_browser.game")}</b>", width: 0.12, text_wrap: :none, font: BOLD_FONT
                  para "#{game_name(server.game)} (#{server.channel})", width: 0.71, text_wrap: :none
                end

                flow(width: 1.0) do
                  para "<b>#{I18n.t(:"server_browser.map")}</b>", width: 0.12, text_wrap: :none, font: BOLD_FONT
                  para server.status.map, width: 0.71, text_wrap: :none
                end

                flow(width: 1.0) do
                  para "<b>#{I18n.t(:"server_browser.time")}</b>", width: 0.12, text_wrap: :none, font: BOLD_FONT
                  para formatted_rentime(server.status.started), text_wrap: :none

                  unless server.status.remaining =~ /00:00:00|00.00.00/
                    para "<b>#{I18n.t(:"server_browser.remaining")}</b>", margin_left: 16, margin_right: 8, text_wrap: :none, font: BOLD_FONT
                    para "#{server.status.remaining}", text_wrap: :none
                  end
                end
              end
            end

            game_balance = server_game_balance(server)

            # Game score and balance display
            flow(width: 1.0, height: 52, border_thickness_bottom: 2, border_color_bottom: 0x44_ffffff) do
              stack(fill: true, height: 1.0) do
                para "#{server.status.teams[0].name} (#{server.status.players.select { |pl| pl.team == 0 }.count})", width: 1.0, text_align: :center, font: BOLD_FONT
                para formatted_score(game_balance[:team_0_score].to_i), width: 1.0, text_align: :center
              end

              stack(width: 0.2, height: 1.0) do
                flow(width: 1.0, height: 0.5) do
                  flow(fill: true)
                  image game_balance[:icon], height: 1.0, tip: game_balance[:message], color: game_balance[:color]
                  flow(fill: true)
                end

                para game_balance[:ratio].round(2).to_s, width: 1.0, text_align: :center
              end

              stack(fill: true, height: 1.0) do
                para "#{server.status.teams[1].name} (#{server.status.players.select { |pl| pl.team == 1 }.count})", width: 1.0, text_align: :center, font: BOLD_FONT
                para formatted_score(game_balance[:team_1_score].to_i), width: 1.0, text_align: :center
              end
            end

            # Team roster
            flow(width: 1.0, fill: true, scroll: true) do
              stack(width: 0.5) do
                server.status.players.select { |ply| ply.team == 0 }.sort_by { |ply| ply.score }.reverse.each_with_index do |player, i|
                  flow(width: 1.0, height: 26) do
                    background 0xaa_333333 if i.even?

                    stack(width: 0.6, height: 1.0) do
                      para player.nick, text_wrap: :none
                    end

                    stack(width: 0.4, height: 1.0) do
                      para formatted_score(player.score), width: 1.0, text_align: :right, text_wrap: :none
                    end
                  end
                end
              end

              stack(width: 0.5, border_thickness_left: 2, border_color_left: 0xff_000000) do
                server.status.players.select { |ply| ply.team == 1 }.sort_by { |ply| ply.score }.reverse.each_with_index do |player, i|
                  flow(width: 1.0, height: 26) do
                    background 0xaa_333333 if i.even?

                    stack(width: 0.6, height: 1.0) do
                      para player.nick, text_wrap: :none
                    end

                    stack(width: 0.4, height: 1.0) do
                      para formatted_score(player.score), width: 1.0, text_align: :right, text_wrap: :none
                    end
                  end
                end
              end
            end
          end
        end
      end

      def fetch_server_details(server)
        BackgroundWorker.foreground_job(
          -> { Api.server_details(server.id, 2) },
          ->(server_data) { server.update(server_data) if server_data }
        )
      end

      def game_icon(server)
        image_path = File.exist?("#{CACHE_PATH}/#{server.game.nil? ? 'ren' : server.game}.png") ? "#{CACHE_PATH}/#{server.game.nil? ? 'ren' : server.game}.png" : "#{GAME_ROOT_PATH}/media/icons/default_icon.png"

        if server.status.password
          @server_locked_icons[server.game] ||= Gosu.render(96, 96) do
            i = get_image(image_path)
            lock = get_image("#{GAME_ROOT_PATH}/media/ui_icons/locked.png")
            scale = [96.0 / i.width, 96.0 / i.height].min

            i.draw(0, 0, 0, scale, scale)
            lock.draw(96 - lock.width * 0.5, 96 - lock.height * 0.5, 0, 0.5, 0.5, 0xff_ff8800)
          end
        else
          image_path
        end
      end

      def game_name(game)
        Store.applications.games.detect { |g| g.id == game }&.name
      end

      def server_game_balance(server)
        data = {
          icon: BLACK_IMAGE,
          color: 0xff_ffffff,
          message: "Estimate of game balance based on score"
        }

        # team 0 is left side
        team_0_score = server.status.teams[0].score
        team_0_score = nil if team_0_score.zero?
        team_0_score ||= server.status.players.select { |ply| ply.team == 0 }.map(&:score).sum
        team_0_score = team_0_score.to_f

        # team 1 is right side
        team_1_score = server.status.teams[1].score
        team_1_score = nil if team_1_score.zero?
        team_1_score ||= server.status.players.select { |ply| ply.team == 1 }.map(&:score).sum
        team_1_score = team_1_score.to_f

        ratio = 1.0 / (team_0_score / team_1_score)
        ratio = 1.0 if ratio.to_s == "NaN"

        data[:ratio] = ratio
        data[:team_0_score] = team_0_score
        data[:team_1_score] = team_1_score

        data[:icon] = if server.status.players.size < 20 && server.game != "ren"
                        data[:color] = 0xff_600000
                        data[:message] = "Too few players for a balanced game"
                        "#{GAME_ROOT_PATH}/media/ui_icons/cross.png"
                      elsif team_0_score + team_1_score < 2_500
                        data[:message] = "Score to low to estimate game balance"
                        data[:color] = 0xff_444444
                        "#{GAME_ROOT_PATH}/media/ui_icons/question.png"
                      elsif ratio.between?(0.75, 1.25)
                        data[:message] = "Game seems balanced based on score"
                        data[:color] = 0xff_008000
                        "#{GAME_ROOT_PATH}/media/ui_icons/checkmark.png"
                      elsif ratio < 0.75
                        data[:color] = 0xff_dd8800
                        data[:message] = "#{server.status.teams[0].name} is winning significantly"
                        "#{GAME_ROOT_PATH}/media/ui_icons/arrowRight.png"
                      else
                        data[:color] = 0xff_dd8800
                        data[:message] = "#{server.status.teams[1].name} is winning significantly"
                        "#{GAME_ROOT_PATH}/media/ui_icons/arrowLeft.png"
                      end

        data
      end

      def formatted_score(int)
        int.to_s.reverse.scan(/.{1,3}/).join(",").reverse
      end

      def formatted_rentime(time)
        range = Time.now - Time.parse(time)

        hours   = range / 60.0 / 60.0 / 24.0
        minutes = (range / 60.0) % 59
        seconds = range % 59

        format("%02d:%02d:%02d", hours, minutes, seconds)
      end
    end
  end
end