From 9f5d3cfa8d15f5d54e6ca1e2d0643e4d8eaa0854 Mon Sep 17 00:00:00 2001
From: Pixelastic <tim@pixelastic.com>
Date: Tue, 9 Jun 2015 13:03:09 +0200
Subject: [PATCH] feat(push): Push items to index.

---
 algoliasearch-jekyll.gemspec |  1 +
 lib/algoliasearch-jekyll.rb  |  6 ++--
 lib/push.rb                  | 70 ++++++++++++++++++++++++++++++++----
 3 files changed, 66 insertions(+), 11 deletions(-)

diff --git a/algoliasearch-jekyll.gemspec b/algoliasearch-jekyll.gemspec
index 4670f19..9dc762c 100644
--- a/algoliasearch-jekyll.gemspec
+++ b/algoliasearch-jekyll.gemspec
@@ -13,6 +13,7 @@ Gem::Specification.new do |s|
   s.add_runtime_dependency 'nokogiri'
   s.add_runtime_dependency 'json'
   s.add_runtime_dependency 'awesome_print'
+  s.add_runtime_dependency 'algoliasearch'
   s.homepage    = 'https://github.com/algolia/algoliasearch-jekyll'
   s.license     = 'MIT'
 end
diff --git a/lib/algoliasearch-jekyll.rb b/lib/algoliasearch-jekyll.rb
index 0c5a313..f50f21b 100644
--- a/lib/algoliasearch-jekyll.rb
+++ b/lib/algoliasearch-jekyll.rb
@@ -19,7 +19,7 @@ class AlgoliaSearchJekyll < Jekyll::Command
 
           subcommand.action do |args, options|
             @config = configuration_from_options(options)
-            AlgoliaSearchJekyllPush.process(args, options)
+            AlgoliaSearchJekyllPush.process(args, options, @config)
           end
         end
       end
@@ -32,9 +32,7 @@ class AlgoliaSearchJekyll < Jekyll::Command
       if File.exist?(key_file) && File.size(key_file) > 0
         return File.open(key_file).read.strip
       end
-
-      
-
+      nil
     end
   end
 end
diff --git a/lib/push.rb b/lib/push.rb
index bf23411..e5967f0 100644
--- a/lib/push.rb
+++ b/lib/push.rb
@@ -1,3 +1,4 @@
+require 'algoliasearch'
 require 'nokogiri'
 require 'json'
 
@@ -7,11 +8,14 @@ class AlgoliaSearchJekyllPush < Jekyll::Command
     def init_with_program(_prog)
     end
 
-    def process(args = [], options = {})
-      index_name = args[0]
-      puts "Pushing to #{index_name} with options #{options}"
+    def process(args = [], options = {}, config = {})
+      @args = args
+      @options = options
+      @config = config
 
-      @config = configuration_from_options(options)
+      index_name = args[0]
+
+      @config['algolia']['index_name'] = index_name if index_name
       site = Jekyll::Site.new(@config)
 
       # Instead of writing generated website to disk, we will push it to the
@@ -34,8 +38,7 @@ class AlgoliaSearchJekyllPush < Jekyll::Command
       @config['markdown_ext'].split(',').include?(ext)
     end
 
-    def push(items)
-      api_key = AlgoliaSearchJekyll.api_key
+    def check_credentials(api_key, application_id, index_name)
       unless api_key
         Jekyll.logger.error 'Algolia Error: No API key defined'
         Jekyll.logger.warn '  You have two ways to configure your API key:'
@@ -43,7 +46,60 @@ class AlgoliaSearchJekyllPush < Jekyll::Command
         Jekyll.logger.warn '    - A file named ./_algolia_api_key'
         exit 1
       end
-      p items
+
+      unless application_id
+        Jekyll.logger.error 'Algolia Error: No application ID defined'
+        Jekyll.logger.warn '  Please set your application id in the '\
+                           '_config.yml file, like so:'
+        puts ''
+        # The spaces are needed otherwise the text is centered
+        Jekyll.logger.warn '  algolia:         '
+        Jekyll.logger.warn '    application_id: \'{your_application_id}\''
+        puts ''
+        Jekyll.logger.warn '  Your application ID can be found in your algolia'\
+                           ' dashboard'
+        Jekyll.logger.warn '    https://www.algolia.com/licensing'
+        exit 1
+      end
+
+      unless index_name
+        Jekyll.logger.error 'Algolia Error: No index name defined'
+        Jekyll.logger.warn '  Please set your index name in the _config.yml'\
+                           ' file, like so:'
+        puts ''
+        # The spaces are needed otherwise the text is centered
+        Jekyll.logger.warn '  algolia:         '
+        Jekyll.logger.warn '    index_name: \'{your_index_name}\''
+        puts ''
+        Jekyll.logger.warn '  You can edit your indices in your dashboard'
+        Jekyll.logger.warn '    https://www.algolia.com/explorer'
+        exit 1
+      end
+      true
+    end
+
+    def push(items)
+      api_key = AlgoliaSearchJekyll.api_key
+      application_id = @config['algolia']['application_id']
+      index_name = @config['algolia']['index_name']
+      check_credentials(api_key, application_id, index_name)
+
+      Algolia.init(application_id: application_id, api_key: api_key)
+      index = Algolia::Index.new(index_name)
+      index.delete_index
+
+      items.each_slice(1000) do |batch|
+        Jekyll.logger.info "Indexing #{batch.size} items"
+        begin
+          index.add_objects(batch)
+        rescue StandardError => error
+          Jekyll.logger.error 'Algolia Error: HTTP Error'
+          Jekyll.logger.warn error.message
+          exit 1
+        end
+      end
+
+      Jekyll.logger.info "Indexing of #{items.size} items done."
     end
 
     def get_items_from_file(file)