1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
module Magic
class Database
attr_reader :flags
# Creates an instance of +Magic::Database+ using given database
# file and flags
def initialize(*args)
options = args.last.is_a?(Hash) ? args.pop : {} # extract options
database = options.delete(:database)
open(*args)
load(database)
end
# Opens magic db using given flags
def open(*flags)
@flags = calculate_flags(*flags)
@magic_set = Api.magic_open(@flags)
end
# Closes the database
def close
Api.magic_close(@magic_set)
end
# Loads given database file (or default if +nil+ given)
def load(database = nil)
Api.magic_load(@magic_set, database)
end
# Determine type of a file at given path
def file(filename)
raise Errno::ENOENT, filename unless File.exists?(filename)
result = Api.magic_file(@magic_set, filename.to_s)
if result.null?
raise Error, error
else
result.get_string(0)
end
end
# Determine type of given string
def buffer(string)
result = Api.magic_buffer(@magic_set, string, string.bytesize)
if result.null?
raise Error, error
else
result.get_string(0)
end
end
# Returns the last error occured
def error
Api.magic_error(@magic_set)
end
# Sets the flags
def flags=(*flags)
@flags = calculate_flags(*flags)
Api.magic_setflags(@magic_set, @flags)
end
protected
def calculate_flags(*flags)
flags.inject(0) { |calculated, flag| calculated |= Constants::Flag.const_get(flag.to_s.upcase) }
end
end
end
|
