require_dependency 'carto/storage' require_dependency 'carto/assets/assets_service' require 'singleton' module Carto class KuvizAssetsService < AssetsService include Singleton EXTENSION = ".html".freeze DEFAULT_MAX_SIZE_IN_BYTES = 1048576 def upload(visualization, resource) super(visualization.id, resource) end def fetch_file(resource) temp_file = Tempfile.new(["kuviz_asset_#{Time.now.utc.to_i}", EXTENSION]) begin read = IO.copy_stream(resource, temp_file, max_size_in_bytes + 1) if read > max_size_in_bytes message = "resource is too big (> #{max_size_in_bytes} bytes)" raise UnprocesableEntityError.new(message) end ensure temp_file.close end temp_file end def read_source_data(asset) if asset.storage_info[:type] == 'local' File.open(asset.storage_info[:identifier]).read else URI.parse(asset.public_url).open.read end end DEFAULT_LOCATION = 'kuviz_assets'.freeze def location @location ||= Cartodb.get_config(:assets, 'kuviz', 'bucket') || Cartodb.get_config(:assets, 'kuviz', 'location') || DEFAULT_LOCATION end def max_size_in_bytes return @max_size_in_bytes if @max_size_in_bytes configured = Cartodb.get_config(:assets, 'kuviz', 'max_size_in_bytes') @max_size_in_bytes = configured || DEFAULT_MAX_SIZE_IN_BYTES end end end