agave-client 0.1.0

data/lib/agave/cli.rb

@@ -0,0 +1,93 @@
+# frozen_string_literal: true
+require 'thor'
+require 'agave/dump/runner'
+require 'agave/dump/ssg_detector'
+require 'agave/watch/site_change_watcher'
+require 'listen'
+require 'thread'
+
+module Agave
+  class Cli < Thor
+    package_name 'AgaveCMS'
+
+    desc 'dump', 'dumps AgaveCMS content into local files'
+    option :config, default: 'agave.config.rb'
+    option :token, default: ENV['AGAVE_API_TOKEN'], required: true
+    option :preview, default: false, type: :boolean
+    option :watch, default: false, type: :boolean
+    def dump
+      if !ENV['AGAVE_API_TOKEN']
+        say 'Site token is not specified! Please configure AGAVE_API_TOKEN.'
+        exit 0
+      end
+      
+      config_file = File.expand_path(options[:config])
+      watch_mode = options[:watch]
+      preview_mode = options[:preview]
+
+      client = Agave::Site::Client.new(
+        extra_headers: {
+          'X-Reason' => 'dump',
+          'X-SSG' => Dump::SsgDetector.new(Dir.pwd).detect
+        }
+      )
+
+      if watch_mode
+        site_id = client.request(:get, '/api/site')['data']['id']
+
+        semaphore = Mutex.new
+
+        thread_safe_dump(semaphore, config_file, client, preview_mode)
+
+        Agave::Watch::SiteChangeWatcher.new(site_id).connect do
+          thread_safe_dump(semaphore, config_file, client, preview_mode)
+        end
+
+        watch_config_file(config_file) do
+          thread_safe_dump(semaphore, config_file, client, preview_mode)
+        end
+
+        sleep
+      else
+        Dump::Runner.new(config_file, client, preview_mode).run
+      end
+    end
+
+    desc 'check', 'checks the presence of a AgaveCMS token'
+    def check
+      exit 0 if ENV['AGAVE_API_TOKEN']
+
+      say 'Site token is not specified!'
+      token = ask "Please paste your AgaveCMS site read-only API token:\n>"
+
+      if !token || token.empty?
+        puts 'Missing token'
+        exit 1
+      end
+
+      File.open('.env', 'a') do |file|
+        file.puts "AGAVE_API_TOKEN=#{token}"
+      end
+
+      say 'Token added to .env file.'
+
+      exit 0
+    end
+
+    no_tasks do
+      def watch_config_file(config_file, &block)
+        Listen.to(
+          File.dirname(config_file),
+          only: /#{Regexp.quote(File.basename(config_file))}/,
+          &block
+        ).start
+      end
+
+      def thread_safe_dump(semaphore, config_file, client, preview_mode)
+        semaphore.synchronize do
+          Dump::Runner.new(config_file, client, preview_mode).run
+        end
+      end
+    end
+  end
+end

data/lib/agave/dump/dsl/add_to_data_file.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+require 'agave/dump/operation/add_to_data_file'
+
+module Agave
+  module Dump
+    module Dsl
+      module AddToDataFile
+        def add_to_data_file(*args)
+          operations.add Operation::AddToDataFile.new(operations, *args)
+        end
+      end
+    end
+  end
+end

data/lib/agave/dump/dsl/create_data_file.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+require 'agave/dump/operation/create_data_file'
+
+module Agave
+  module Dump
+    module Dsl
+      module CreateDataFile
+        def create_data_file(*args)
+          operations.add Operation::CreateDataFile.new(operations, *args)
+        end
+      end
+    end
+  end
+end

data/lib/agave/dump/dsl/create_post.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+require 'agave/dump/operation/create_post'
+
+module Agave
+  module Dump
+    module Dsl
+      class DataFile
+        def initialize(operation, &block)
+          @operation = operation
+          instance_eval(&block)
+        end
+
+        def frontmatter(format, value)
+          @operation.frontmatter_format = format
+          @operation.frontmatter_value = value
+        end
+
+        def content(value)
+          @operation.content = value
+        end
+      end
+
+      module CreatePost
+        def create_post(path, &block)
+          operation = Operation::CreatePost.new(operations, path)
+          DataFile.new(operation, &block)
+
+          operations.add operation
+        end
+      end
+    end
+  end
+end

data/lib/agave/dump/dsl/directory.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+require 'agave/dump/dsl/create_post'
+require 'agave/dump/dsl/create_data_file'
+require 'agave/dump/dsl/add_to_data_file'
+
+module Agave
+  module Dump
+    module Dsl
+      class Directory
+        include Dsl::CreateDataFile
+        include Dsl::CreatePost
+        include Dsl::AddToDataFile
+
+        attr_reader :agave, :operations
+
+        def initialize(agave, operations, &block)
+          @agave = agave
+          @operations = operations
+          @self_before_instance_eval = eval 'self', block.binding
+
+          instance_eval(&block)
+        end
+
+        def method_missing(method, *args, &block)
+          @self_before_instance_eval.send method, *args, &block
+        end
+      end
+    end
+  end
+end

data/lib/agave/dump/dsl/root.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+require 'agave/dump/dsl/directory'
+require 'agave/dump/dsl/create_post'
+require 'agave/dump/dsl/create_data_file'
+require 'agave/dump/dsl/add_to_data_file'
+
+require 'agave/dump/operation/directory'
+
+module Agave
+  module Dump
+    module Dsl
+      class Root
+        include Dsl::CreateDataFile
+        include Dsl::CreatePost
+        include Dsl::AddToDataFile
+
+        attr_reader :agave, :operations
+
+        def initialize(config_code, agave, operations)
+          @agave = agave
+          @operations = operations
+
+          # rubocop:disable Lint/Eval
+          eval(config_code)
+          # rubocop:enable Lint/Eval
+        end
+
+        def directory(path, &block)
+          operation = Operation::Directory.new(operations, path)
+          operations.add operation
+
+          Directory.new(agave, operation, &block)
+        end
+      end
+    end
+  end
+end

data/lib/agave/dump/format.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+require 'agave/dump/format/toml'
+require 'agave/dump/format/yaml'
+
+module Agave
+  module Dump
+    module Format
+      def self.dump(format, value)
+        converter_for(format).dump(value)
+      end
+
+      def self.frontmatter_dump(format, value)
+        converter_for(format).frontmatter_dump(value)
+      end
+
+      def self.converter_for(format)
+        case format.to_sym
+        when :toml
+          Format::Toml
+        when :yaml, :yml
+          Format::Yaml
+        when :json
+          Format::Json
+        end
+      end
+    end
+  end
+end

data/lib/agave/dump/format/json.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+require 'json'
+
+module Agave
+  module Dump
+    module Format
+      module Json
+        def self.dump(value)
+          JSON.dump(value)
+        end
+
+        def self.frontmatter_dump(value)
+          "#{dump(value)}\n"
+        end
+      end
+    end
+  end
+end

data/lib/agave/dump/format/toml.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+require 'active_support/core_ext/hash/keys'
+require 'toml'
+
+class Time
+  def to_toml(_path = '')
+    utc.strftime('%Y-%m-%dT%H:%M:%SZ')
+  end
+end
+
+class Date
+  def to_toml(_path = '')
+    strftime('%Y-%m-%d')
+  end
+end
+
+module Agave
+  module Dump
+    module Format
+      module Toml
+        def self.dump(value)
+          TOML::Generator.new(value).body
+        end
+
+        def self.frontmatter_dump(value)
+          "+++\n#{dump(value)}+++"
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+require 'active_support/core_ext/hash/keys'
+require 'yaml'
+
+require 'agave/local/item'
+
+class Array
+  def deep_stringify_keys
+    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
+
+module Agave
+  module Dump
+    module Format
+      module Yaml
+        def self.deep_hashify_items(value)
+          case value
+          when Array
+            value.map { |v| deep_hashify_items(v) }
+          when Hash
+            value.each_with_object({}) do |(k, v), acc|
+              acc[k] = deep_hashify_items(v)
+            end
+          when ::Agave::Local::Item
+            value.to_hash
+          else
+            if value.respond_to?(:to_hash)
+              value.to_hash
+            else
+              value
+            end
+          end
+        end
+
+        def self.dump(value)
+          plain = deep_hashify_items(value)
+          YAML.dump(plain.deep_stringify_keys).chomp.gsub(/^\-+\n/, '')
+        end
+
+        def self.frontmatter_dump(value)
+          "---\n#{dump(value)}\n---"
+        end
+      end
+    end
+  end
+end

data/lib/agave/dump/operation/add_to_data_file.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+require 'agave/dump/format'
+
+module Agave
+  module Dump
+    module Operation
+      class AddToDataFile
+        attr_reader :context, :path, :format, :value
+
+        def initialize(context, path, format, value)
+          @context = context
+          @path = path
+          @format = format
+          @value = value
+        end
+
+        def perform
+          complete_path = File.join(context.path, path)
+          FileUtils.mkdir_p(File.dirname(complete_path))
+
+          content_to_add = Format.dump(format, value)
+
+          old_content = if File.exist? complete_path
+                          ::File.read(complete_path)
+                        else
+                          ''
+                        end
+
+          new_content = old_content.sub(
+            /\n*(#\s*agavecms:start.*#\s*agavecms:end|\Z)/m,
+            "\n\n# agavecms:start\n#{content_to_add}\n# agavecms:end"
+          )
+
+          File.open(complete_path, 'w') do |f|
+            f.write new_content
+          end
+        end
+      end
+    end
+  end
+end

data/lib/agave/dump/operation/create_data_file.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+require 'fileutils'
+require 'agave/dump/format'
+
+module Agave
+  module Dump
+    module Operation
+      class CreateDataFile
+        attr_reader :context, :path, :format, :value
+
+        def initialize(context, path, format, value)
+          @context = context
+          @path = path
+          @format = format
+          @value = value
+        end
+
+        def perform
+          FileUtils.mkdir_p(File.dirname(path))
+
+          File.open(File.join(context.path, path), 'w') do |file|
+            file.write Format.dump(format, value)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/agave/dump/operation/create_post.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+require 'fileutils'
+require 'agave/dump/format'
+
+module Agave
+  module Dump
+    module Operation
+      class CreatePost
+        attr_reader :context, :path
+
+        attr_accessor :frontmatter_format, :frontmatter_value
+        attr_accessor :content
+
+        def initialize(context, path)
+          @context = context
+          @path = path
+        end
+
+        def perform
+          FileUtils.mkdir_p(File.dirname(path))
+
+          File.open(File.join(context.path, path), 'w') do |file|
+            file.write Format.frontmatter_dump(
+              frontmatter_format,
+              frontmatter_value
+            )
+            file.write "\n\n"
+            file.write content
+          end
+        end
+      end
+    end
+  end
+end