Class: KeyDB

Inherits:
Object
  • Object
show all
Defined in:
lib/keydb.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(filename) ⇒ KeyDB

Returns a new instance of KeyDB.



4
5
6
7
8
9
10
11
 
# File 'lib/keydb.rb', line 4

def initialize filename
  @filename = filename.split(".")
  @filename = "#{@filename[0]}.db"
  unless Dir.entries('.').detect {|f| f.match "#{@filename}"}
    @db_file = File.new(@filename, "w")
    @db_file.close
  end
end

Dynamic Method Handling

This class handles dynamic methods through the method_missing method

#method_missing(keyname, *value) ⇒ Object 



63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
 
# File 'lib/keydb.rb', line 63

def method_missing(keyname, *value)
  keyname = keyname.to_s
  if keyname =~ /=/
    keyname.gsub!("=", "")
    value = value[0]
    if keyname =~ /\/\/\/|\(---\)|-\*\*\*-/
      raise "Invalid Keyname!"
      return
    end
    File.open(@filename, "r").each_line do |line|
      if Marshal.load(line).split("///")[0] == keyname
        raise "Keynames must be unique"
        return
      end
    end
    @db_file = File.open(@filename, "a")
    @to_write = (Marshal.dump("#{keyname}///#{value}"))
    @db_file.write(@to_write)
    @db_file.write("\n")
    @db_file.close
  else
    File.open(@filename, "r").each_line do |line|
      if Marshal.load(line).split("(---)")[1] == keyname
        return "Key has been deleted"
      elsif Marshal.load(line).split("-***-")[1] == keyname
        return Marshal.load(line).split("-***-")[2]
      end
    end
    File.open(@filename, "r").each_line do |line|
      line = Marshal.load(line)
      if line.split("///")[0] == keyname
        return line.split("///")[1]
      end
    end
  end 
end

Instance Attribute Details

#filename ⇒ Object (readonly)

Returns the value of attribute filename.



2
3
4
 
# File 'lib/keydb.rb', line 2

def filename
  @filename
end

#version ⇒ Object (readonly)

Returns the value of attribute version.



2
3
4
 
# File 'lib/keydb.rb', line 2

def version
  @version
end

Instance Method Details

#change_key(keyname, value) ⇒ Object 



13
14
15
16
17
18
 
# File 'lib/keydb.rb', line 13

def change_key keyname, value
  @db_file = File.open(@filename, "a")
  @db_file.write(Marshal.dump("CK-***-#{keyname}-***-#{value}"))
  @db_file.write("\n")
  @db_file.close
end

#delete_db ⇒ Object 



54
55
56
 
# File 'lib/keydb.rb', line 54

def delete_db
  system("rm #{@filename}")
end

#delete_key(keyname) ⇒ Object 



20
21
22
23
24
25
 
# File 'lib/keydb.rb', line 20

def delete_key keyname
  @db_file = File.open(@filename, "a")
  @db_file.write(Marshal.dump("DK(---)#{keyname}"))
  @db_file.write("\n")
  @db_file.close
end

#keys ⇒ Object 



31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
 
# File 'lib/keydb.rb', line 31

def keys
  to_ignore = Array.new
  File.open(@filename).each_line do |line|
    if Marshal.load(line).split("(---)")[1] != nil
      to_ignore.push(Marshal.load(line).split("(---)")[1])
    elsif Marshal.load(line).split("-***-")[1] != nil
      to_ignore.push(Marshal.load(line).split("-***-")[1])
    end
  end
  puts to_ignore
  File.open(@filename).each_line do |line|
    line = Marshal.load(line)
    @keys_and_vals = line.split('///')
    if line.split("(---)").length == 2 or to_ignore.include? line.split("///")[0]
      print
    elsif to_ignore.include? line.split("-***-")[1] 
      print  
    else
      puts @keys_and_vals[0]
    end
  end
end

#to_s ⇒ Object 



27
28
29
 
# File 'lib/keydb.rb', line 27

def to_s
  return "KeyDB-#{@filename}"
end

#wipe_db ⇒ Object 



58
59
60
61
 
# File 'lib/keydb.rb', line 58

def wipe_db
  delete_db
  File.new(@filename, "w")
end