lib/kitkat/database.rb in kitkat-0.0.1 vs lib/kitkat/database.rb in kitkat-0.0.2

@@ -2,10 +2,12 @@
 
 module Kitkat
   # Database-level operations.
   class Database
     def initialize(path)
+      ensure_dir_exists(path)
+
       @connection = SQLite3::Database.new(path)
 
       load_schema
 
       freeze
@@ -27,9 +29,15 @@
     end
 
     private
 
     attr_reader :connection
+
+    def ensure_dir_exists(path)
+      dir = File.dirname(path)
+
+      FileUtils.mkdir_p(dir) unless File.exist?(dir)
+    end
 
     def sql_statement
       'INSERT OR IGNORE INTO files VALUES (?, ?, ?, ?, ?, ?, ?)'
     end