root/lang/ruby/pit/lib/pit.rb @ 3968

require "yaml"
require "pathname"
require "tempfile"

module Pit
        Directory = Pathname.new("~/.pit").expand_path
        @@config  = Directory + "default.yaml"

        def self.set(name, opts={})
                if opts.key?(:data)
                        result = opts[:data]
                else
                        c = self.get(name).to_yaml
                        t = Tempfile.new("pit")
                        t << c
                        t.close
                        system(ENV["EDITOR"], t.path)
                        t.open
                        result = t.read
                        if result == c
                                $stderr.puts "No Changes"
                                return
                        end
                        result = YAML.load(result)
                end
                config = self.load
                config[name] = result
                @@config.open("w") {|f| YAML.dump(config, f) }
                nil
        end

        def self.get(name, opts={})
                self.load[name] || {"username"=>"", "password"=>""}
        end

        def self.switch(name, opts={})
                # TODO
                @@config = Directory + "#{name}.yaml"
        end

        def self.load
                Directory.mkpath unless Directory.exist?
                unless @@config.exist?
                        @@config.open("w") {|f| f << {}.to_yaml }
                        @@config.chmod(0600)
                end
                YAML.load(@@config.read) || {}
        end
end


__END__
p Pit.set("test")
p Pit.get("test")

config = Pit.get("twitter")
p config["password"]
p config["username"]