Moved back to compact class naming to use File.exitst?
This commit is contained in:
parent
135dcceafd
commit
95e27b0994
1 changed files with 71 additions and 74 deletions
|
@ -1,87 +1,84 @@
|
|||
# Copyright (C) 2012-2014 Zammad Foundation, http://zammad-foundation.org/
|
||||
# rubocop:disable ClassAndModuleChildren
|
||||
|
||||
class Store
|
||||
module Provider
|
||||
module File
|
||||
class Store::Provider::File
|
||||
|
||||
def self.add(data, sha)
|
||||
write_to_fs(data, sha)
|
||||
true
|
||||
end
|
||||
def self.add(data, sha)
|
||||
write_to_fs(data, sha)
|
||||
true
|
||||
end
|
||||
|
||||
def self.get(sha)
|
||||
read_from_fs(sha)
|
||||
end
|
||||
def self.get(sha)
|
||||
read_from_fs(sha)
|
||||
end
|
||||
|
||||
def self.delete(sha)
|
||||
unlink_from_fs(sha)
|
||||
end
|
||||
def self.delete(sha)
|
||||
unlink_from_fs(sha)
|
||||
end
|
||||
|
||||
private
|
||||
private
|
||||
|
||||
# generate file location
|
||||
def self.get_locaton(sha)
|
||||
# generate file location
|
||||
def self.get_locaton(sha)
|
||||
|
||||
# generate directory
|
||||
base = Rails.root.to_s + '/storage/fs/'
|
||||
parts = sha.scan(/.{1,4}/)
|
||||
path = parts[ 1 .. 10 ].join('/') + '/'
|
||||
file = parts[ 11 .. parts.count ].join('')
|
||||
location = "#{base}/#{path}"
|
||||
# generate directory
|
||||
base = Rails.root.to_s + '/storage/fs/'
|
||||
parts = sha.scan(/.{1,4}/)
|
||||
path = parts[ 1 .. 10 ].join('/') + '/'
|
||||
file = parts[ 11 .. parts.count ].join('')
|
||||
location = "#{base}/#{path}"
|
||||
|
||||
# create directory if not exists
|
||||
if !File.exist?( location )
|
||||
FileUtils.mkdir_p( location )
|
||||
end
|
||||
location += file
|
||||
end
|
||||
# create directory if not exists
|
||||
if !File.exist?( location )
|
||||
FileUtils.mkdir_p( location )
|
||||
end
|
||||
location += file
|
||||
end
|
||||
|
||||
# unlink file from fs
|
||||
def self.unlink_from_fs(sha)
|
||||
if File.exist?( get_locaton(sha) )
|
||||
puts "NOTICE: storge remove '#{ get_locaton(sha) }'"
|
||||
File.delete( get_locaton(sha) )
|
||||
end
|
||||
end
|
||||
|
||||
# read file from fs
|
||||
def self.read_from_fs(sha)
|
||||
puts "read from fs #{ get_locaton(sha) }"
|
||||
if !File.exist?( get_locaton(sha) )
|
||||
raise "ERROR: No such file #{ get_locaton(sha) }"
|
||||
end
|
||||
data = File.open( get_locaton(sha), 'rb' )
|
||||
content = data.read
|
||||
|
||||
# check sha
|
||||
local_sha = Digest::SHA256.hexdigest( content )
|
||||
if local_sha != sha
|
||||
raise "ERROR: Corrupt file in fs #{ get_locaton(sha) }, sha should be #{sha} but is #{local_sha}"
|
||||
end
|
||||
content
|
||||
end
|
||||
|
||||
# write file to fs
|
||||
def self.write_to_fs(data, sha)
|
||||
|
||||
# install file
|
||||
permission = '600'
|
||||
if !File.exist?( get_locaton(sha) )
|
||||
puts "NOTICE: storge write '#{ get_locaton(sha) }' (#{permission})"
|
||||
file = File.new( get_locaton(sha), 'wb' )
|
||||
file.write( data )
|
||||
file.close
|
||||
end
|
||||
File.chmod( permission.to_i(8), get_locaton(sha) )
|
||||
|
||||
# check sha
|
||||
local_sha = Digest::SHA256.hexdigest( read_from_fs(sha) )
|
||||
if sha != local_sha
|
||||
raise "ERROR: Corrupt file in fs #{ get_locaton(sha) }, sha should be #{sha} but is #{local_sha}"
|
||||
end
|
||||
|
||||
true
|
||||
end
|
||||
# unlink file from fs
|
||||
def self.unlink_from_fs(sha)
|
||||
if File.exist?( get_locaton(sha) )
|
||||
puts "NOTICE: storge remove '#{ get_locaton(sha) }'"
|
||||
File.delete( get_locaton(sha) )
|
||||
end
|
||||
end
|
||||
|
||||
# read file from fs
|
||||
def self.read_from_fs(sha)
|
||||
puts "read from fs #{ get_locaton(sha) }"
|
||||
if !File.exist?( get_locaton(sha) )
|
||||
raise "ERROR: No such file #{ get_locaton(sha) }"
|
||||
end
|
||||
data = File.open( get_locaton(sha), 'rb' )
|
||||
content = data.read
|
||||
|
||||
# check sha
|
||||
local_sha = Digest::SHA256.hexdigest( content )
|
||||
if local_sha != sha
|
||||
raise "ERROR: Corrupt file in fs #{ get_locaton(sha) }, sha should be #{sha} but is #{local_sha}"
|
||||
end
|
||||
content
|
||||
end
|
||||
|
||||
# write file to fs
|
||||
def self.write_to_fs(data, sha)
|
||||
|
||||
# install file
|
||||
permission = '600'
|
||||
if !File.exist?( get_locaton(sha) )
|
||||
puts "NOTICE: storge write '#{ get_locaton(sha) }' (#{permission})"
|
||||
file = File.new( get_locaton(sha), 'wb' )
|
||||
file.write( data )
|
||||
file.close
|
||||
end
|
||||
File.chmod( permission.to_i(8), get_locaton(sha) )
|
||||
|
||||
# check sha
|
||||
local_sha = Digest::SHA256.hexdigest( read_from_fs(sha) )
|
||||
if sha != local_sha
|
||||
raise "ERROR: Corrupt file in fs #{ get_locaton(sha) }, sha should be #{sha} but is #{local_sha}"
|
||||
end
|
||||
|
||||
true
|
||||
end
|
||||
end
|
||||
|
|
Loading…
Reference in a new issue