mirror of
https://github.com/discourse/discourse.git
synced 2024-12-13 19:13:45 +08:00
30990006a9
This reduces chances of errors where consumers of strings mutate inputs and reduces memory usage of the app. Test suite passes now, but there may be some stuff left, so we will run a few sites on a branch prior to merging
82 lines
1.8 KiB
Ruby
82 lines
1.8 KiB
Ruby
# 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" then value.to_i
|
|
when "TrueClass", "FalseClass" then value == "true"
|
|
when "JSON" then map_json(::JSON.parse(value))
|
|
else value
|
|
end
|
|
end
|
|
end
|