zammad/app/models/store/file.rb

# Copyright (C) 2012-2014 Zammad Foundation, http://zammad-foundation.org/

class Store
  class File < ApplicationModel
    include ApplicationLib
    after_destroy :destroy_provider

    # add new file
    def self.add(data)
      sha = Digest::SHA256.hexdigest( data )

      file = Store::File.where( sha: sha ).first
      if file == nil

        # load backend based on config
        adapter_name = Setting.get('storage_provider') || 'DB'
        if !adapter_name
          raise 'Missing storage_provider setting option'
        end
        adapter = self.load_adapter( "Store::Provider::#{ adapter_name }" )
        adapter.add( data, sha )
        file = Store::File.create(
          provider: adapter_name,
          sha: sha,
        )
      end
      file
    end

    # read content
    def content
      adapter = self.class.load_adapter("Store::Provider::#{ self.provider }")
      if self.sha
        c = adapter.get( self.sha )
      else
        # fallback until migration is done
        c = Store::Provider::DB.where( md5: self.md5 ).first.data
      end
      c
    end

    # check data and sha, in case fix it
    def self.verify(fix_it = nil)
      success = true
      Store::File.all.each {|item|
        content = item.content
        sha = Digest::SHA256.hexdigest( content )
        logger.info "CHECK: Store::File.find(#{item.id}) "

        next if sha == item.sha

        success = false
        logger.error "DIFF: sha diff of Store::File.find(#{item.id}) "
        if fix_it
          item.update_attribute( :sha, sha )
        end
      }
      success
    end

    # move file from one to other provider
    # e. g. Store::File.move('File', 'DB')
    # e. g. Store::File.move('DB', 'File')
    def self.move(source, target)
      adapter_source = load_adapter("Store::Provider::#{ source }")
      adapter_target = load_adapter("Store::Provider::#{ target }")

      Store::File.all.each {|item|
        next if item.provider == target
        content = item.content

        # add to new provider
        adapter_target.add( content, item.sha )

        # update meta data
        item.update_attribute( :provider, target )

        # remove from old provider
        adapter_source.delete( item.sha )

        logger.info "Moved file #{item.sha} from #{source} to #{target}"
      }
      true
    end

    private

    def destroy_provider
      adapter = self.class.load_adapter("Store::Provider::#{ self.provider }")
      adapter.delete( self.sha )
    end
  end
end