contentful_bootstrap 1.6.0 → 2.0.2

data/lib/contentful/bootstrap/commands/generate_token.rb

@@ -0,0 +1,52 @@
+require 'contentful/management'
+require 'contentful/management/error'
+require 'contentful/management/request'
+require 'contentful/bootstrap/version'
+require 'contentful/bootstrap/commands/base'
+
+module Contentful
+  module Bootstrap
+    module Commands
+      class GenerateToken < Base
+        attr_reader :token_name
+        def initialize(token, space, token_name = 'Bootstrap Token', trigger_oauth = true)
+          super(token, space, trigger_oauth)
+          @token_name = token_name
+          @actual_space = space unless space.is_a?(String)
+        end
+
+        def run
+          puts 'Creating Delivery API Token'
+
+          fetch_space
+
+          access_token = fetch_access_token
+
+          puts "Token '#{token_name}' created! - '#{access_token}'"
+          print 'Do you want to write the Delivery Token to your configuration file? (Y/n): '
+          unless gets.chomp.downcase == 'n'
+            @token.write_access_token(@actual_space.name, access_token)
+            @token.write_space_id(@actual_space.name, @actual_space.id)
+          end
+
+          access_token
+        end
+
+        def fetch_space
+          @actual_space ||= Contentful::Management::Space.find(@space)
+        end
+
+        def fetch_access_token
+          response = Contentful::Management::Request.new(
+            "/#{@actual_space.id}/api_keys",
+            'name' => token_name,
+            'description' => "Created with 'contentful_bootstrap.rb v#{Contentful::Bootstrap::VERSION}'"
+          ).post
+          fail response if response.object.is_a?(Contentful::Management::Error)
+
+          response.object['accessToken']
+        end
+      end
+    end
+  end
+end

data/lib/contentful/bootstrap/constants.rb

@@ -1,8 +1,8 @@
 module Contentful
   module Bootstrap
     module Constants
-      OAUTH_APP_ID = "a19770bea126e6d596d8599ff42e14173409e3e252895b78d0cb289c10586276".freeze
-      OAUTH_CALLBACK_URL = "http://localhost:5123/oauth_callback".freeze
+      OAUTH_APP_ID = 'a19770bea126e6d596d8599ff42e14173409e3e252895b78d0cb289c10586276'.freeze
+      OAUTH_CALLBACK_URL = 'http://localhost:5123/oauth_callback'.freeze
     end
   end
 end

data/lib/contentful/bootstrap/generator.rb

@@ -0,0 +1,94 @@
+require 'contentful'
+require 'inifile'
+require 'json'
+require 'zlib'
+
+module Contentful
+  module Bootstrap
+    class Generator
+      def initialize(space_id, access_token)
+        @client = Contentful::Client.new(access_token: access_token, space: space_id)
+      end
+
+      def generate_json
+        template = {}
+        template['content_types'] = content_types
+        template['assets'] = assets
+        template['entries'] = entries
+        JSON.pretty_generate(template)
+      end
+
+      private
+
+      def assets
+        proccessed_assets = @client.assets.map do |asset|
+          result = { 'id' => asset.sys[:id], 'title' => asset.title }
+          result['file'] = {
+            'filename' => ::File.basename(asset.file.file_name, '.*'),
+            'url' => "https:#{asset.file.url}"
+          }
+          result
+        end
+        proccessed_assets.sort_by { |item| item['id'] }
+      end
+
+      def content_types
+        proccessed_content_types = @client.content_types.map do |type|
+          result = { 'id' => type.sys[:id], 'name' => type.name }
+          result['display_field'] = type.display_field unless type.display_field.nil?
+
+          result['fields'] = type.fields.map do |field|
+            map_field_properties(field.properties)
+          end
+
+          result
+        end
+        proccessed_content_types.sort_by { |item| item['id'] }
+      end
+
+      def entries
+        entries = {}
+
+        @client.entries.each do |entry|
+          result = { 'id' => entry.sys[:id] }
+
+          entry.fields.each do |key, value|
+            value = map_field(value)
+            result[key] = value unless value.nil?
+          end
+
+          ct_id = entry.content_type.sys[:id]
+          entries[ct_id] = [] if entries[ct_id].nil?
+          entries[ct_id] << result
+        end
+
+        entries
+      end
+
+      def map_field(value)
+        return value.map { |v| map_field(v) } if value.is_a? ::Array
+
+        if value.is_a?(Contentful::Asset) || value.is_a?(Contentful::Entry)
+          return {
+            'link_type' => value.class.name.split('::').last,
+            'id' => value.sys[:id]
+          }
+        end
+
+        return nil if value.is_a?(Contentful::Link)
+
+        value
+      end
+
+      def map_field_properties(properties)
+        properties['link_type'] = properties.delete(:linkType) unless properties[:linkType].nil?
+
+        items = properties[:items]
+        properties[:items] = map_field_properties(items.properties) unless items.nil?
+
+        properties.delete_if { |k, v| v.nil? || [:required, :localized].include?(k) }
+        properties
+      end
+    end
+  end
+end

data/lib/contentful/bootstrap/server.rb

@@ -1,8 +1,8 @@
-require "thread"
-require "webrick"
-require "launchy"
-require "contentful/bootstrap/constants"
-require "contentful/bootstrap/token"
+require 'thread'
+require 'webrick'
+require 'launchy'
+require 'contentful/bootstrap/constants'
+require 'contentful/bootstrap/token'
 
 module Contentful
   module Bootstrap
@@ -45,40 +45,48 @@ module Contentful
     end
 
     class IndexController < WEBrick::HTTPServlet::AbstractServlet
-      def do_GET(request, response)
+      def do_GET(_request, response)
         client_id = Contentful::Bootstrap::Constants::OAUTH_APP_ID
         redirect_uri = Contentful::Bootstrap::Constants::OAUTH_CALLBACK_URL
-        scope = "content_management_manage"
+        scope = 'content_management_manage'
         Launchy.open("https://be.contentful.com/oauth/authorize?response_type=token&client_id=#{client_id}&redirect_uri=#{redirect_uri}&scope=#{scope}")
         response.status = 200
-        response.body = ""
+        response.body = ''
       end
     end
 
     class OAuthCallbackController < WEBrick::HTTPServlet::AbstractServlet
-      def do_GET(request, response)
+      def do_GET(_request, response)
         response.status = 200
-        response.content_type = "text/html"
+        response.content_type = 'text/html'
         response.body = OAuthEchoView.new.render
       end
     end
 
     class SaveTokenController < WEBrick::HTTPServlet::AbstractServlet
+      attr_reader :token
+
+      def initialize(server, token, *options)
+        super(server, options)
+        @token = token
+      end
+
       def do_GET(request, response)
-        Token.write(request.query["token"])
+        @token.write(request.query['token'])
         response.status = 200
-        response.content_type = "text/html"
+        response.content_type = 'text/html'
         response.body = ThanksView.new.render
       end
     end
 
     class Server
       attr_reader :server
-      def initialize
-        @server = WEBrick::HTTPServer.new(:Port => 5123)
-        @server.mount "/", IndexController
-        @server.mount "/oauth_callback", OAuthCallbackController
-        @server.mount "/save_token", SaveTokenController
+
+      def initialize(token)
+        @server = WEBrick::HTTPServer.new(Port: 5123)
+        @server.mount '/', IndexController
+        @server.mount '/oauth_callback', OAuthCallbackController
+        @server.mount '/save_token', SaveTokenController, token
       end
 
       def start

data/lib/contentful/bootstrap/support.rb

@@ -1,10 +1,11 @@
-require "stringio"
+require 'stringio'
 
 module Contentful
   module Bootstrap
     module Support
       def silence_stderr
-        old_stderr, $stderr = $stderr, StringIO.new
+        old_stderr = $stderr
+        $stderr = StringIO.new
         yield
       ensure
         $stderr = old_stderr

data/lib/contentful/bootstrap/templates.rb

@@ -1,4 +1,4 @@
-require "contentful/bootstrap/templates/blog"
-require "contentful/bootstrap/templates/catalogue"
-require "contentful/bootstrap/templates/gallery"
-require "contentful/bootstrap/templates/json_template"
+require 'contentful/bootstrap/templates/blog'
+require 'contentful/bootstrap/templates/catalogue'
+require 'contentful/bootstrap/templates/gallery'
+require 'contentful/bootstrap/templates/json_template'

data/lib/contentful/bootstrap/templates/base.rb

@@ -1,5 +1,6 @@
-require "contentful/management"
-require "contentful/bootstrap/templates/links/base"
+require 'json'
+require 'contentful/management'
+require 'contentful/bootstrap/templates/links/base'
 
 module Contentful
   module Bootstrap
@@ -12,9 +13,18 @@ module Contentful
         end
 
         def run
-          create_content_types
-          create_assets
-          create_entries
+          begin
+            create_content_types
+            create_assets
+            create_entries
+          rescue Contentful::Management::Error => e
+            error = e.error
+            puts "Error at: #{error[:url]}"
+            puts "Message: #{error[:message]}"
+            puts "Details: #{error[:details]}"
+
+            raise e
+          end
         end
 
         def content_types
@@ -30,6 +40,7 @@ module Contentful
         end
 
         protected
+
         def create_image(name, url)
           image = Contentful::Management::File.new
           image.properties[:contentType] = 'image/jpeg'
@@ -39,27 +50,28 @@ module Contentful
         end
 
         private
+
         def create_content_types
           content_types.each do |ct|
-            puts "Creating Content Type '#{ct["name"]}'"
+            puts "Creating Content Type '#{ct['name']}'"
 
             fields = []
             content_type = space.content_types.new
-            content_type.id = ct["id"]
-            content_type.name = ct["name"]
-            content_type.display_field = ct["display_field"]
+            content_type.id = ct['id']
+            content_type.name = ct['name']
+            content_type.display_field = ct['display_field']
 
-            ct["fields"].each do |f|
+            ct['fields'].each do |f|
               field = Contentful::Management::Field.new
-              field.id = f["id"]
-              field.name = f["name"]
-              field.type = f["type"]
-              field.link_type = f["link_type"] if is_link?(f)
+              field.id = f['id']
+              field.name = f['name']
+              field.type = f['type']
+              field.link_type = f['link_type'] if link?(f)
 
-              if is_array?(f)
+              if array?(f)
                 array_field = Contentful::Management::Field.new
-                array_field.type = f["items"]["type"]
-                array_field.link_type = f["items"]["link_type"]
+                array_field.type = f['items']['type']
+                array_field.link_type = f['items']['link_type']
                 field.items = array_field
               end
 
@@ -72,25 +84,34 @@ module Contentful
           end
         end
 
-        def is_link?(field)
-          field.has_key?("link_type")
+        def link?(field)
+          field.key?('link_type')
         end
 
-        def is_array?(field)
-          field.has_key?("items")
+        def array?(field)
+          field.key?('items')
         end
 
         def create_assets
           assets.each do |asset|
-            puts "Creating Asset '#{asset["title"]}'"
+            puts "Creating Asset '#{asset['title']}'"
             asset = space.assets.create(
-              id: asset["id"],
-              title: asset["title"],
-              file: asset["file"]
+              id: asset['id'],
+              title: asset['title'],
+              file: asset['file']
             )
             asset.process_file
-            sleep(1) # Wait for Process
-            asset.publish
+
+            attempts = 0
+            while attempts < 10
+              unless space.assets.find(asset.id).file.url.nil?
+                asset.publish
+                break
+              end
+
+              sleep(1) # Wait for Process
+              attempts += 1
+            end
           end
         end
 
@@ -106,8 +127,8 @@ module Contentful
               array_fields = []
               regular_fields = []
               e.each do |field_name, value|
-                if value.is_a? Array
-                  array_fields << field_name if value.is_a? Array
+                if value.is_a? ::Array
+                  array_fields << field_name
                   next
                 end
 
@@ -116,7 +137,7 @@ module Contentful
 
               array_fields.each do |af|
                 e[af].map! do |item|
-                  if item.is_a? Links::Base
+                  if item.is_a? ::Contentful::Bootstrap::Templates::Links::Base
                     item.to_management_object
                   else
                     item

data/lib/contentful/bootstrap/templates/blog.rb

@@ -1,5 +1,5 @@
-require "contentful/bootstrap/templates/base"
-require "contentful/bootstrap/templates/links"
+require 'contentful/bootstrap/templates/base'
+require 'contentful/bootstrap/templates/links'
 
 module Contentful
   module Bootstrap
@@ -8,37 +8,37 @@ module Contentful
         def content_types
           [
             {
-              "id" => 'author',
-              "name" => 'Author',
-              "display_field" => 'name',
-              "fields" => [
+              'id' => 'author',
+              'name' => 'Author',
+              'display_field' => 'name',
+              'fields' => [
                 {
-                  "id" => 'name',
-                  "name" => "Author Name",
-                  "type" => "Symbol"
+                  'id' => 'name',
+                  'name' => 'Author Name',
+                  'type' => 'Symbol'
                 }
               ]
             },
             {
-              "id" => 'post',
-              "name" => 'Post',
-              "display_field" => 'title',
-              "fields" => [
+              'id' => 'post',
+              'name' => 'Post',
+              'display_field' => 'title',
+              'fields' => [
                 {
-                  "id" => 'title',
-                  "name" => "Post Title",
-                  "type" => "Symbol"
+                  'id' => 'title',
+                  'name' => 'Post Title',
+                  'type' => 'Symbol'
                 },
                 {
-                  "id" => 'content',
-                  "name" => "Content",
-                  "type" => "Text"
+                  'id' => 'content',
+                  'name' => 'Content',
+                  'type' => 'Text'
                 },
                 {
-                  "id" => 'author',
-                  "name" => "Author",
-                  "type" => "Link",
-                  "link_type" => "Entry"
+                  'id' => 'author',
+                  'name' => 'Author',
+                  'type' => 'Link',
+                  'link_type' => 'Entry'
                 }
               ]
             }
@@ -49,24 +49,24 @@ module Contentful
           {
             'author' => [
               {
-                "id" => "dan_brown",
-                "name" => "Dan Brown"
+                'id' => 'dan_brown',
+                'name' => 'Dan Brown'
               },
               {
-                "id" => "pablo_neruda",
-                "name" => "Pablo Neruda"
+                'id' => 'pablo_neruda',
+                'name' => 'Pablo Neruda'
               }
             ],
             'post' => [
               {
-                "title" => "Inferno",
-                "content" => "Inferno is the last book in Dan Brown's collection...",
-                "author" => Links::Entry.new("dan_brown")
+                'title' => 'Inferno',
+                'content' => 'Inferno is the last book in Dan Brown\'s collection...',
+                'author' => Links::Entry.new('dan_brown')
               },
               {
-                "title" => "Alturas de Macchu Picchu",
-                "content" => "Alturas de Macchu Picchu is one of Pablo Neruda's most famous poetry books...",
-                "author" => Links::Entry.new("pablo_neruda")
+                'title' => 'Alturas de Macchu Picchu',
+                'content' => 'Alturas de Macchu Picchu is one of Pablo Neruda\'s most famous poetry books...',
+                'author' => Links::Entry.new('pablo_neruda')
               }
             ]
           }