lib/chronicle/imessage/imessage_extractor.rb in chronicle-imessage-0.2.6 vs lib/chronicle/imessage/imessage_extractor.rb in chronicle-imessage-0.2.7

@@ -5,13 +5,14 @@
   module Imessage 
     class ImessageExtractor < Chronicle::ETL::Extractor
       register_connector do |r|
         r.provider = 'imessage'
         r.description = 'a local imessage database'
+        r.identifier = 'messages'
       end
 
-      setting :db, default: File.join(Dir.home, 'Library', 'Messages', 'chat.db'), required: true
+      setting :input, default: File.join(Dir.home, 'Library', 'Messages', 'chat.db'), required: true
       setting :load_attachments, default: false
       setting :only_attachments, default: false
       setting :my_phone_number
       setting :my_name
       setting :icloud_account_id
@@ -38,11 +39,15 @@
         @messages.count
       end
 
       private
 
+      def db_file
+        [@config.input].flatten.first
+      end
+
       def prepare_data
-        @db = SQLite3::Database.new(@config.db, results_as_hash: true)
+        @db = SQLite3::Database.new(db_file, results_as_hash: true)
         @local_contacts = LocalContacts.new
         @contacts = @local_contacts.contacts
 
         @messages = load_messages
         @chats = load_chats