lib/dotgpg/environment.rb in dotgpg-environment-0.1.1 vs lib/dotgpg/environment.rb in dotgpg-environment-0.2.0

@@ -7,14 +7,39 @@
     def read
       # use dotgpg here
       dir = Dotgpg::Dir.closest(@filename)
 
       fail "not in a dotgpg directory" unless dir
+
+      # if the file's not there assume we're creating a new one and return an
+      # empty string
+      return '' unless File.exists? @filename
+
       # make a new stringio object to pass in
       s = StringIO.new
       dir.decrypt @filename, s
       # have to rewind, otherwise read doesn't work
       s.rewind
       s.read
+    end
+
+    def write
+      # use dotgpg here
+      dir = Dotgpg::Dir.closest(@filename)
+
+      fail "not in a dotgpg directory" unless dir
+      # make a new stringio object to pass in
+      s = StringIO.new
+
+      sort.each do |k,v|
+        # if our value has newlines or #'s it needs to be double-quoted. in
+        # addition newlines need to be \n and not actual multi-line strings,
+        # see https://github.com/bkeepers/dotenv#usage
+        v = v.inspect if v.to_s.match(/\n|#/)
+        s.write "#{k}=#{v}\n"
+      end
+
+      s.rewind
+      dir.encrypt @filename, s
     end
   end
 end