discourse/app/models/plugin_store.rb

# frozen_string_literal: true

# API to wrap up plugin store rows
class PluginStore
  attr_reader :plugin_name

  def initialize(plugin_name)
    @plugin_name = plugin_name
  end

  def get(key)
    self.class.get(plugin_name, key)
  end

  def get_all(keys)
    self.class.get_all(plugin_name, keys)
  end

  def set(key, value)
    self.class.set(plugin_name, key, value)
  end

  def remove(key)
    self.class.remove(plugin_name, key)
  end

  def self.get(plugin_name, key)
    if row = PluginStoreRow.find_by(plugin_name: plugin_name, key: key)
      cast_value(row.type_name, row.value)
    end
  end

  def self.get_all(plugin_name, keys)
    rows = PluginStoreRow.where("plugin_name = ? AND key IN (?)", plugin_name, keys).to_a

    Hash[rows.map { |row| [row.key, cast_value(row.type_name, row.value)] }]
  end

  def self.set(plugin_name, key, value)
    hash = { plugin_name: plugin_name, key: key }
    row = PluginStoreRow.find_by(hash) || PluginStoreRow.new(hash)

    row.type_name = determine_type(value)
    # nil are stored as nil
    row.value =
      if row.type_name == "JSON"
        value.to_json
      elsif value
        value.to_s
      end

    row.save
  end

  def self.remove(plugin_name, key)
    PluginStoreRow.where(plugin_name: plugin_name, key: key).destroy_all
  end

  def self.determine_type(value)
    value.is_a?(Hash) || value.is_a?(Array) ? "JSON" : value.class.to_s
  end

  def self.map_json(item)
    if item.is_a? Hash
      ActiveSupport::HashWithIndifferentAccess.new item
    elsif item.is_a? Array
      item.map { |subitem| map_json subitem }
    else
      item
    end
  end

  def self.cast_value(type, value)
    case type
    when "Integer", "Fixnum"
      value.to_i
    when "TrueClass", "FalseClass"
      value == "true"
    when "JSON"
      map_json(::JSON.parse(value))
    else
      value
    end
  end
end