ubicloud/helpers/web.rb

# frozen_string_literal: true

class Clover < Roda
  PG_STATE_LABEL_COLOR = Hash.new("bg-slate-100 text-slate-800").merge!(
    "running" => "bg-green-100 text-green-800",
    "creating" => "bg-yellow-100 text-yellow-800",
    "deleting" => "bg-red-100 text-red-800"
  ).freeze

  PS_STATE_LABEL_COLOR = Hash.new("bg-yellow-100 text-yellow-80").merge!(
    "available" => "bg-green-100 text-green-800"
  ).freeze

  KUBERNETES_STATE_LABEL_COLOR = Hash.new("bg-slate-100 text-slate-800").merge!(
    "running" => "bg-green-100 text-green-800",
    "creating" => "bg-yellow-100 text-yellow-800",
    "deleting" => "bg-red-100 text-red-800"
  ).freeze

  VM_STATE_LABEL_COLOR = Hash.new("bg-slate-100 text-slate-800").merge!(
    "running" => "bg-green-100 text-green-800",
    "creating" => "bg-yellow-100 text-yellow-800",
    "deleting" => "bg-red-100 text-red-800"
  )
  ["rebooting", "starting", "waiting for capacity", "restarting"].each do
    VM_STATE_LABEL_COLOR[it] = VM_STATE_LABEL_COLOR["creating"]
  end
  VM_STATE_LABEL_COLOR["deleted"] = VM_STATE_LABEL_COLOR["deleting"]
  VM_STATE_LABEL_COLOR.freeze

  BUTTON_COLOR = Hash.new { |h, k| raise "unsupported button type: #{k}" }.merge!(
    "primary" => "bg-orange-600 hover:bg-orange-700 focus-visible:outline-orange-600",
    "safe" => "bg-green-600 hover:bg-green-700 focus-visible:outline-green-600",
    "warning" => "bg-amber-600 hover:bg-amber-700 focus-visible:outline-amber-600",
    "danger" => "bg-rose-600 hover:bg-rose-700 focus-visible:outline-rose-600"
  ).freeze

  PG_HA_DATA = {
    PostgresResource::HaType::NONE => "Inactive",
    PostgresResource::HaType::ASYNC => "Active (1 standby with asynchronous replication)",
    PostgresResource::HaType::SYNC => "Active (2 standbys with synchronous replication)"
  }.freeze

  def csrf_tag(*)
    part("components/form/hidden", name: csrf_field, value: csrf_token(*))
  end

  def raise_web_error(message)
    raise CloverError.new(400, nil, message)
  end

  def handle_validation_failure(template, &block)
    return unless web?
    @validation_failure_template = template
    @validation_failure_block = block
  end

  def redirect_back_with_inputs
    # :nocov:
    if (template = @validation_failure_template)
      # :nocov:
      flash.sweep
      @validation_failure_block&.call

      request.on do
        view(template)
      rescue Roda::RodaPlugins::TypecastParams::Error
        @page_title = "Invalid Parameter Type"
        view(content: "<p>An invalid parameter type was submitted.</p>")
      end
    end

    # :nocov:
    # This code path is deprecated and will be removed after all routes have been updated
    # to use handle_validation_failure.

    if Config.test?
      # Raise error in the tests if we get here. If this error is raised, the route
      # should be fixed to call handle_validation_failure.
      raise "Request failure without handle_validation_failure: #{request.request_method} #{request.path_info}"
    else
      # Emit error messages in other environments. This will allow detection of errors in production
      # for cases where we don't have specs that cover the error. These errors will be
      # monitored and specs will be added for them.
      Clog.emit("web error without handle_validation_failure") do
        {
          missing_handle_validation_failure: {
            request_method: request.request_method,
            path_info: request.path_info,
            referrer: env["HTTP_REFERER"]
          }
        }
      end
    end

    referrer = flash["referrer"] || env["HTTP_REFERER"]
    uri = begin
      Kernel.URI(referrer)
    rescue URI::InvalidURIError, ArgumentError
      nil
    end

    request.redirect "/" unless uri

    flash["old"] = redirect_back_with_inputs_params

    if uri && env["REQUEST_METHOD"] != "GET"
      # Force flash rotation, so flash works correctly for internal redirects
      _roda_after_40__flash(nil)

      rack_response = Clover.call(env.merge("REQUEST_METHOD" => "GET", "PATH_INFO" => uri.path, "rack.input" => StringIO.new("".b), "rack.request.form_input" => nil, "rack.request.form_hash" => nil))
      flash.discard
      flash["referrer"] = referrer
      env.delete("roda.session.serialized")
      rack_response[0] = response.status || 400
      request.halt rack_response
    else
      request.redirect referrer
    end
  end

  def redirect_back_with_inputs_params
    request.params
  end
  # :nocov:

  def redirect_default_project_dashboard
    if (project = current_account.projects_dataset.order(:created_at, :name).first)
      request.redirect "#{project.path}/dashboard"
    else
      request.redirect "/project"
    end
  end

  def omniauth_provider_name(provider)
    omniauth_providers.each { |sym, name| return name if sym.name == provider }
    OidcProvider.name_for_ubid(provider) || provider
  end

  def omniauth_providers
    @omniauth_providers ||= [
      # :nocov:
      Config.omniauth_google_id ? [:google, "Google"] : nil,
      Config.omniauth_github_id ? [:github, "GitHub"] : nil
      # :nocov:
    ].compact
  end

  def sort_aces!(aces)
    @aces.sort! do |a, b|
      # :nocov:
      # Admin tag at the top (one of these branches will be hit, but
      # cannot force which)
      next -1 unless a.last
      next 1 unless b.last
      # :nocov:
      # Label sorting by subject, action, object for remaining ACEs
      a_tags = a[1]
      b_tags = b[1]
      x = nil
      a_tags.each_with_index do |v, i|
        x = ace_label(v) <=> ace_label(b_tags[i])
        break unless x.nil? || x.zero?
      end
      next x unless x.nil? || x.zero?
      # Tie break using ubid
      a[0] <=> b[0]
    end
  end

  def ace_label(obj)
    case obj
    when nil
      "All"
    when ActionType
      obj.name
    when ActionTag
      "#{"Global " unless obj.project_id}Tag: #{obj.name}"
    when ObjectTag, SubjectTag
      "Tag: #{obj.name}"
    when ObjectMetatag
      "ObjectTag: #{obj.name}"
    when ApiKey
      "InferenceApiKey: #{obj.name}"
    else
      "#{obj.class.name}: #{obj.name}"
    end
  end

  def html_attrs(attributes)
    attributes.map do |key, value|
      case key
      when :required, :checked, :readonly
        case value
        when true
          key.name
        when false, nil
          ""
        else
          "#{key.name}=\"#{h(value)}\""
        end
      else
        "#{h(key)}=\"#{h(value)}\""
      end
    end.join(" ")
  end

  def object_tag_membership_label(obj)
    case obj
    when ObjectTag
      "Tag: #{obj.name}"
    when ObjectMetatag
      "ObjectTag: #{obj.name}"
    when ApiKey
      "InferenceApiKey: #{obj.name}"
    else
      "#{obj.class.name}: #{obj.name}"
    end
  end

  def check_ace_subject(subject)
    # Do not allow personal access tokens as subjects
    # Do not allow modifiction or addition of an ace entry with the Admin subject,
    # which is reserved for full access.
    if UBID.uuid_class_match?(subject, ApiKey) ||
        UBID.uuid_class_match?(subject, SubjectTag) && SubjectTag[subject].name == "Admin"
      raise Authorization::Unauthorized
    end
  end

  def config_hash_from_kvs(keys, values)
    hash = keys.map(&:strip).zip(values.map(&:strip)).to_h.compact
    hash.delete_if { |key, value| key.empty? && value.empty? }
    hash
  end

  def humanize_size(bytes)
    return nil if bytes.nil? || bytes.zero?
    units = %w[B KB MB GB]
    exp = (Math.log(bytes) / Math.log(1024)).to_i
    exp = [exp, units.size - 1].min

    return "%d %s" % [bytes.to_f / 1024**exp, units[exp]] if exp == 0
    "%.1f %s" % [bytes.to_f / 1024**exp, units[exp]]
  end

  def humanize_time(time)
    seconds = Time.now - time
    return "just now" if seconds < 60
    return "#{(seconds / 60).to_i} minutes ago" if seconds < 3600
    return "#{(seconds / 3600).to_i} hours ago" if seconds < 86400
    "#{(seconds / 86400).to_i} days ago"
  end

  def million_token_price(resource)
    (BillingRate.from_resource_properties("InferenceTokens", resource, "global")["unit_price"] * 1_000_000).round(2)
  end

  def money(amount)
    "$%0.02f" % amount
  end
end