module Danbooru
  module Extensions
    module String
      def to_escaped_for_sql_like
        string = self.gsub(/%|_|\*|\\\*|\\\\|\\/) do |str|
          case str
          when '%'    then '\%'
          when '_'    then '\_'
          when '*'    then '%'
          when '\*'   then '*'
          when '\\\\' then '\\\\'
          when '\\'   then '\\\\'
          end
        end

        string
      end

      # escape \ and * characters so that they're treated literally in LIKE searches.
      def escape_wildcards
        gsub(/\\/, '\\\\').gsub(/\*/, '\*')
      end

      def to_escaped_for_tsquery_split
        scan(/\S+/).map {|x| x.to_escaped_for_tsquery}.join(" & ")
      end

      def to_escaped_for_tsquery
        "'#{gsub(/\0/, '').gsub(/'/, '\0\0').gsub(/\\/, '\0\0\0\0')}'"
      end

      def truthy?
        self.match?(/\A(true|t|yes|y|on|1)\z/i)
      end

      def falsy?
        self.match?(/\A(false|f|no|n|off|0)\z/i)
      end

      def ilike?(pattern)
        pattern = Regexp.escape(pattern).gsub(/\\\*/, ".*")
        match?(/\A#{pattern}\z/i)
      end

      def normalize_whitespace
        # Normalize various horizontal space characters to ASCII space.
        text = gsub(/\p{Zs}|\t/, " ")

        # Strip various zero width space characters. Zero width joiner (200D)
        # is allowed because it's used in emoji.
        text = text.gsub(/[\u180E\u200B\u200C\u2060\uFEFF]/, "")

        # Normalize various line ending characters to CRLF.
        text = text.gsub(/\r?\n|\r|\v|\f|\u0085|\u2028|\u2029/, "\r\n")

        text
      end
    end
  end
end

class String
  include Danbooru::Extensions::String
end

module MimeNegotationExtension
  # Ignore all file extensions except for .html, .js, .json, and .xml when
  # parsing the file extension from the URL. Needed for wiki pages (e.g.
  # /wiki_pages/rnd.jpg).
  private def format_from_path_extension
    mime = super

    if mime&.symbol.in?(%i[html js json xml])
      mime
    else
      nil
    end
  end
end

ActionDispatch::Http::MimeNegotiation.prepend(MimeNegotationExtension)

# Make Symbol#to_s return a frozen string. This reduces allocations, but may be
# incompatible with some libraries.
#
# https://bugs.ruby-lang.org/issues/16150
# https://github.com/Shopify/symbol-fstring
Symbol.alias_method(:to_s, :name)