123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196 |
- # Copyright (C) 2012-2016 Zammad Foundation, http://zammad-foundation.org/
- class Store < ApplicationModel
- load 'store/object.rb'
- load 'store/file.rb'
- store :preferences
- belongs_to :store_object, class_name: 'Store::Object'
- belongs_to :store_file, class_name: 'Store::File'
- validates :filename, presence: true
- =begin
- add an attachment to storage
- result = Store.add(
- object: 'Ticket::Article',
- o_id: 4711,
- data: binary_string,
- preferences: {
- content_type: 'image/png',
- content_id: 234,
- }
- )
- returns
- result = true
- =end
- def self.add(data)
- data = data.stringify_keys
- # lookup store_object.id
- store_object = Store::Object.create_if_not_exists(name: data['object'])
- data['store_object_id'] = store_object.id
- # add to real store
- file = Store::File.add(data['data'])
- data['size'] = data['data'].to_s.bytesize
- data['store_file_id'] = file.id
- # not needed attributes
- data.delete('data')
- data.delete('object')
- # store meta data
- store = Store.create!(data)
- store
- end
- =begin
- get attachment of object
- list = Store.list(
- object: 'Ticket::Article',
- o_id: 4711,
- )
- returns
- result = [store1, store2]
- store1 = {
- size: 94123,
- filename: 'image.png',
- preferences: {
- content_type: 'image/png',
- content_id: 234,
- }
- }
- store1.content # binary_string
- =end
- def self.list(data)
- # search
- store_object_id = Store::Object.lookup(name: data[:object])
- stores = Store.where(store_object_id: store_object_id, o_id: data[:o_id].to_i)
- .order('created_at ASC, id ASC')
- stores
- end
- =begin
- remove attachments of object from storage
- result = Store.remove(
- object: 'Ticket::Article',
- o_id: 4711,
- )
- returns
- result = true
- =end
- def self.remove(data)
- # search
- store_object_id = Store::Object.lookup(name: data[:object])
- stores = Store.where(store_object_id: store_object_id)
- .where(o_id: data[:o_id])
- .order('created_at ASC, id ASC')
- stores.each do |store|
- # check backend for references
- Store.remove_item(store.id)
- end
- true
- end
- =begin
- remove one attachment from storage
- Store.remove_item(store_id)
- =end
- def self.remove_item(store_id)
- store = Store.find(store_id)
- file_id = store.store_file_id
- # check backend for references
- files = Store.where(store_file_id: file_id)
- if files.count > 1 || files.first.id != store.id
- store.destroy!
- return true
- end
- store.destroy!
- Store::File.find(file_id).destroy!
- end
- =begin
- get content of file
- store = Store.find(store_id)
- content_as_string = store.content
- returns
- content_as_string
- =end
- def content
- file = Store::File.find_by(id: store_file_id)
- if !file
- raise "No such file #{store_file_id}!"
- end
- file.content
- end
- =begin
- get content of file
- store = Store.find(store_id)
- location_of_file = store.save_to_file
- returns
- location_of_file
- =end
- def save_to_file(path = nil)
- content
- file = Store::File.find_by(id: store_file_id)
- if !file
- raise "No such file #{store_file_id}!"
- end
- if !path
- path = Rails.root.join('tmp', filename)
- end
- ::File.open(path, 'wb') do |handle|
- handle.write file.content
- end
- path
- end
- def provider
- file = Store::File.find_by(id: store_file_id)
- if !file
- raise "No such file #{store_file_id}!"
- end
- file.provider
- end
- end
|