dato 0.1.12 → 0.1.13

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 57f88cb5b563d207575df6cc877db833b84e6ba8
-  data.tar.gz: 18d368ed8bf4431445fef22bc81c894294802239
+  metadata.gz: 9ce7d382f884eb6efae6931aa28679144d557aa7
+  data.tar.gz: 61a5d88910e3a6f9478a489e98f7cfa08467b37c
 SHA512:
-  metadata.gz: 5413cc7f5c30e958d53a929b954409e0ae0c6b3a625648d0f38d5ccb253aa1a5d8643803de577177ac9d29cf4775029ea028121f61c83910d8f6e47817258be4
-  data.tar.gz: d026c30365f6d67f2739f2785913831cf320c06a4f2283c1e6aad517b20842d6fa72dbf876bb31ec87c29ed493ded7bdd7b55305aaa44d1a5479dc3f1690dfa5
+  metadata.gz: c76af4354a87d7c467d95628ea9d49a09769cd68d99a5099470c0c4bd924dd1f0b3f4f40d38d3e76d8c6635db789a8c73b362bc6e15ed1dd2311d33aa7743bea
+  data.tar.gz: 251ba83f8e43176dd94aebb694a669687390846d3618bb98d6c04c68458016dcae2ee7621b0659a268bcd147f29f6afc99043f5f8bb160d8115a7ecec7764ecf

data/lib/dato/account/client.rb

@@ -16,11 +16,16 @@ module Dato
         sites: Repo::Site
       }.freeze
 
-      attr_reader :token, :base_url, :schema
+      attr_reader :token, :base_url, :schema, :extra_headers
 
-      def initialize(token, base_url: 'https://account-api.datocms.com')
+      def initialize(
+        token,
+        base_url: 'https://account-api.datocms.com',
+        extra_headers: {}
+      )
         @base_url = base_url
         @token = token
+        @extra_headers = extra_headers
       end
 
       REPOS.each do |method_name, repo_klass|
@@ -44,11 +49,12 @@ module Dato
       def connection
         options = {
           url: base_url,
-          headers: {
+          headers: extra_headers.merge(
             'Accept' => 'application/json',
             'Content-Type' => 'application/json',
-            'Authorization' => "Bearer #{@token}"
-          }
+            'Authorization' => "Bearer #{@token}",
+            'User-Agent' => "ruby-client v#{Dato::VERSION}"
+          )
         }
 
         @connection ||= Faraday.new(options) do |c|

data/lib/dato/dump/format/yaml.rb

@@ -4,15 +4,15 @@ require 'yaml'
 
 class Array
   def deep_stringify_keys
-    inject([]) { |accum, value|
-      if (value.is_a?(Hash) or value.is_a?(Array))
+    each_with_object([]) do |value, accum|
+      if value.is_a?(Hash) || value.is_a?(Array)
         new_val = value.deep_stringify_keys
         accum.push new_val
       else
         accum.push value
       end
       accum
-    }
+    end
   end
 end
 

data/lib/dato/dump/runner.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 require 'dato/dump/dsl/root'
 require 'dato/dump/operation/root'
+require 'dato/dump/ssg_detector'
 
 module Dato
   module Dump
@@ -13,17 +14,18 @@ module Dato
       end
 
       def run
-        site.load
-
-        root = Operation::Root.new(Dir.pwd)
-
         Dsl::Root.new(
           File.read(config_path),
           site.items_repo,
-          root
+          operation
         )
 
-        root.perform
+        site.load
+        operation.perform
+      end
+
+      def operation
+        @operation ||= Operation::Root.new(Dir.pwd)
       end
 
       def site
@@ -31,7 +33,17 @@ module Dato
       end
 
       def client
-        @client ||= Dato::Site::Client.new(api_token)
+        @client ||= Dato::Site::Client.new(
+          api_token,
+          extra_headers: {
+            'X-Reason' => 'dump',
+            'X-SSG' => generator
+          }
+        )
+      end
+
+      def generator
+        SsgDetector.new(Dir.pwd).detect
       end
     end
   end

data/lib/dato/dump/ssg_detector.rb

@@ -0,0 +1,99 @@
+# frozen_string_literal: true
+require 'toml'
+require 'json'
+require 'yaml'
+
+module Dato
+  module Dump
+    class SsgDetector
+      attr_reader :path
+
+      RUBY = %w(middleman jekyll nanoc).freeze
+
+      NODE = %w(brunch assemble ember-cli hexo metalsmith react-scripts
+                roots docpad wintersmith gatsby harp grunt gulp).freeze
+
+      PYTHON = %w(mkdocs pelican cactus).freeze
+
+      HUGO = [
+        {
+          file: 'config.toml',
+          loader: ->(content) { TOML::Parser.new(content).parsed }
+        },
+        {
+          file: 'config.yaml',
+          loader: ->(content) { YAML.load(content) }
+        },
+        {
+          file: 'config.json',
+          loader: ->(content) { JSON.load(content) }
+        }
+      ].freeze
+
+      def initialize(path)
+        @path = path
+      end
+
+      def detect
+        ruby_generator ||
+          node_generator ||
+          python_generator ||
+          hugo ||
+          'unknown'
+      end
+
+      private
+
+      def ruby_generator
+        gemfile_path = File.join(path, 'Gemfile')
+        return unless File.exist?(gemfile_path)
+
+        gemfile = File.read(gemfile_path)
+
+        RUBY.find do |generator|
+          gemfile =~ /('#{generator}'|"#{generator}")/
+        end
+      end
+
+      def node_generator
+        package_path = File.join(path, 'package.json')
+        return unless File.exist?(package_path)
+
+        package = JSON.load(File.read(package_path))
+
+        deps = package.fetch('dependencies', {})
+        dev_deps = package.fetch('devDependencies', {})
+        all_deps = deps.merge(dev_deps)
+
+        NODE.find do |generator|
+          all_deps.key? generator
+        end
+      rescue JSON::ParserError
+        nil
+      end
+
+      def python_generator
+        requirements_path = File.join(path, 'requirements.txt')
+        return unless File.exist?(requirements_path)
+
+        requirements = File.read(requirements_path)
+
+        PYTHON.find do |generator|
+          requirements =~ /^#{generator}(==)?/
+        end
+      end
+
+      def hugo
+        HUGO.any? do |option|
+          path = File.join(path, option[:file])
+          config = option[:parser].call(File.read(path))
+          config.key? 'baseurl'
+        end && 'hugo'
+      rescue JSON::ParserError
+        nil
+      rescue Psych::SyntaxError
+        nil
+      end
+    end
+  end
+end

data/lib/dato/site/client.rb

@@ -29,11 +29,16 @@ module Dato
         items: Repo::Item
       }.freeze
 
-      attr_reader :token, :base_url, :schema
+      attr_reader :token, :base_url, :schema, :extra_headers
 
-      def initialize(token, base_url: 'https://site-api.datocms.com')
+      def initialize(
+        token,
+        base_url: 'https://site-api.datocms.com',
+        extra_headers: {}
+      )
         @base_url = base_url
         @token = token
+        @extra_headers = extra_headers
       end
 
       def upload_file(path_or_url)
@@ -67,11 +72,12 @@ module Dato
       def connection
         options = {
           url: base_url,
-          headers: {
+          headers: extra_headers.merge(
             'Accept' => 'application/json',
             'Content-Type' => 'application/json',
-            'Authorization' => "Bearer #{@token}"
-          }
+            'Authorization' => "Bearer #{@token}",
+            'User-Agent' => "ruby-client v#{Dato::VERSION}"
+          )
         }
 
         @connection ||= Faraday.new(options) do |c|

data/lib/dato/version.rb

@@ -1,4 +1,4 @@
 # frozen_string_literal: true
 module Dato
-  VERSION = '0.1.12'
+  VERSION = '0.1.13'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: dato
 version: !ruby/object:Gem::Version
-  version: 0.1.12
+  version: 0.1.13
 platform: ruby
 authors:
 - Stefano Verna
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2016-09-05 00:00:00.000000000 Z
+date: 2016-09-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -321,6 +321,7 @@ files:
 - lib/dato/dump/operation/directory.rb
 - lib/dato/dump/operation/root.rb
 - lib/dato/dump/runner.rb
+- lib/dato/dump/ssg_detector.rb
 - lib/dato/json_api_deserializer.rb
 - lib/dato/json_api_serializer.rb
 - lib/dato/local/entities_repo.rb
@@ -379,4 +380,3 @@ signing_key:
 specification_version: 4
 summary: Ruby client for DatoCMS API
 test_files: []
-has_rdoc: