summaryrefslogtreecommitdiff
path: root/lib/chef/file_store.rb
blob: b834624090fae78cc61040f27b6028f98f75edce (plain)
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
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
99
require File.join(File.dirname(__FILE__), "mixin", "params_validate")
require 'digest/sha2'
require 'json'

class Chef
  class FileStore
    class << self
      include Chef::Mixin::ParamsValidate
  
      def store(obj_type, name, object)
        validate(
          {
            :obj_type => obj_type,
            :name => name,
            :object => object,
          },
          {
            :object => { :respond_to => :to_json },
          }
        )
      
        store_path = create_store_path(obj_type, name)
        io = File.open(store_path, "w")
        io.puts object.to_json
        io.close
      end
    
      def load(obj_type, name)
        validate(
          {
            :obj_type => obj_type,
            :name => name,
          },
          {
            :obj_type => { :kind_of => String },
            :name => { :kind_of => String },
          }
        )
        store_path = create_store_path(obj_type, name)
        raise "Cannot find #{store_path} for #{obj_type} #{name}!" unless File.exists?(store_path)
      
        object = JSON.parse(IO.read(store_path))
      end
      
      def delete(obj_type, name)
        validate(
          {
            :obj_type => obj_type,
            :name => name,
          },
          {
            :obj_type => { :kind_of => String },
            :name => { :kind_of => String },
          }
        )
        store_path = create_store_path(obj_type, name)
        if File.exists?(store_path)
          File.unlink(store_path)
        end
      end
      
      def list(obj_type)
        validate(
          { 
            :obj_type => obj_type,
          },
          {
            :obj_type => { :kind_of => String }
          }
        )
        keys = Array.new
        Dir[File.join(Chef::Config[:file_store_path], obj_type, '**', '*')].each do |f|
          if File.file?(f)
            keys << File.basename(f)
          end
        end
        keys
      end
      
      def create_store_path(obj_type, key)
        shadigest = Digest::SHA2.hexdigest("#{obj_type}#{key}")
        
        file_path = [
          Chef::Config[:file_store_path],
          obj_type,
          shadigest[0,1],
          shadigest[1,3]
        ]
        file_path.each_index do |i|
          create_path = File.join(file_path[0, i + 1])
          Dir.mkdir(create_path) unless File.directory?(create_path) 
        end
        file_path << key
        File.join(*file_path)
      end
  
    end
  end
end