invidious/src/invidious/routes/search.cr

{% skip_file if flag?(:api_only) %}

module Invidious::Routes::Search
  def self.opensearch(env)
    locale = env.get("preferences").as(Preferences).locale
    env.response.content_type = "application/opensearchdescription+xml"

    XML.build(indent: "  ", encoding: "UTF-8") do |xml|
      xml.element("OpenSearchDescription", xmlns: "http://a9.com/-/spec/opensearch/1.1/") do
        xml.element("ShortName") { xml.text "Invidious" }
        xml.element("LongName") { xml.text "Invidious Search" }
        xml.element("Description") { xml.text "Search for videos, channels, and playlists on Invidious" }
        xml.element("InputEncoding") { xml.text "UTF-8" }
        xml.element("Image", width: 48, height: 48, type: "image/x-icon") { xml.text "#{HOST_URL}/favicon.ico" }
        xml.element("Url", type: "text/html", method: "get", template: "#{HOST_URL}/search?q={searchTerms}")
      end
    end
  end

  def self.results(env)
    locale = env.get("preferences").as(Preferences).locale

    query = env.params.query["search_query"]?
    query ||= env.params.query["q"]?

    page = env.params.query["page"]?

    if query && !query.empty?
      if page && !page.empty?
        env.redirect "/search?q=" + URI.encode_www_form(query) + "&page=" + page
      else
        env.redirect "/search?q=" + URI.encode_www_form(query)
      end
    else
      env.redirect "/search"
    end
  end

  def self.search(env)
    preferences = env.get("preferences").as(Preferences)
    locale = preferences.locale

    search_disabled = !CONFIG.page_enabled?("search")

    region = env.params.query["region"]? || preferences.region

    query = Invidious::Search::Query.new(env.params.query, :regular, region)

    # empty query → show homepage
    if query.empty?
      env.set "search", ""
      return templated "search_homepage", navbar_search: false
    end

    # When YouTube search is disabled, force subscription-only mode
    if search_disabled
      user = env.get?("user")
      if !user
        return error_template(403, translate(locale, "search_subscriptions_login_required"))
      end

      user = user.as(User)

      begin
        items = Invidious::Search::Processors.subscriptions_and_playlists(query, user)
      rescue ex
        return error_template 500, ex
      end

      redirect_url = Invidious::Frontend::Misc.redirect_url(env)

      page_nav_html = Frontend::Pagination.nav_numeric(locale,
        base_url: "/search?#{query.to_http_params}",
        current_page: query.page,
        show_next: (items.size >= 20)
      )

      env.set "search", query.text
      env.set "subscription_only_search", true

      return templated "search"
    end

    # non‐empty query → process it
    user = env.get?("user")
    if query.url?
      return env.redirect UrlSanitizer.process(query.text).to_s
    end

    begin
      items = user ? query.process(user.as(User)) : query.process
    rescue ex : ChannelSearchException
      return error_template 404, "Unable to find channel with id "#{HTML.escape(ex.channel)}"…"
    rescue ex
      return error_template 500, ex
    end

    redirect_url = Invidious::Frontend::Misc.redirect_url(env)

    # Pagination
    page_nav_html = Frontend::Pagination.nav_numeric(locale,
      base_url: "/search?#{query.to_http_params}",
      current_page: query.page,
      show_next: (items.size >= 20)
    )

    # If it's a channel search, prefix the box; otherwise just show the text
    if query.type == Invidious::Search::Query::Type::Channel
      env.set "search", "channel:#{query.channel} #{query.text}"
    else
      env.set "search", query.text
    end

    templated "search"
  end

  def self.hashtag(env : HTTP::Server::Context)
    locale = env.get("preferences").as(Preferences).locale

    hashtag = env.params.url["hashtag"]?
    if hashtag.nil? || hashtag.empty?
      return error_template(400, "Invalid request")
    end

    page = env.params.query["page"]?
    if page.nil?
      page = 1
    else
      page = Math.max(1, page.to_i)
      env.params.query.delete_all("page")
    end

    begin
      items = Invidious::Hashtag.fetch(hashtag, page)
    rescue ex
      return error_template(500, ex)
    end

    # Pagination
    hashtag_encoded = URI.encode_www_form(hashtag, space_to_plus: false)
    page_nav_html = Frontend::Pagination.nav_numeric(locale,
      base_url: "/hashtag/#{hashtag_encoded}",
      current_page: page,
      show_next: (items.size >= 60)
    )

    templated "hashtag"
  end
end