59dc3d8 - teagles/wagon

-

require_relative 'helpers/check_path'

-

require_relative 'helpers/force_color'

-

require_relative 'generate'

-

module Locomotive

-

module Wagon

-

module CLI

-

class Generate < Thor

-

include Helpers::CheckPath

-

class_option :path, aliases: '-p', type: :string, default: '.', optional: true, desc: 'if your LocomotiveCMS site is not in the current path'

-

desc 'content_type SLUG FIELDS', 'Creates a content type with the specified slug and fields.'

-

long_desc <<-LONGDESC

-

Creates a content type with the specified slug and fields.

-

SLUG should be plural, lowercase, and underscored.

-

FIELDS format: field_1[:TYPE][:REQUIRED] field_2[:TYPE][:REQUIRED] ...

-

TYPE values: string, text, integer, float, boolean, email, date, date_time, file, tags, select, belongs_to, has_many, or many_to_many. Default is string.

-

To require a field, set REQUIRED to true. The first field is required by default.

-

Examples:

-

* wagon generate content_type posts title published_at:date_time:true body:text

-

* wagon generate content_type products title price:float photo:file category:belongs_to:true

-

LONGDESC

-

def content_type(name, *fields)

-

say('The fields are missing', :red) and return false if fields.empty?

-

if check_path!

-

Locomotive::Wagon.generate :content_type, name, self.options['path'], fields

-

end

-

end

-

desc 'page FULLPATH', 'Create a page. No need to pass an extension to the FULLPATH arg'

-

long_desc <<-LONGDESC

-

Create a page. The generator will ask for the extension (liquid or haml) and also

-

if the page is localized or not.

-

Examples:

-

* wagon generate page contact

-

* wagon generate page about_us/me

-

LONGDESC

-

def page(fullpath)

-

if path = check_path!

-

locales = self.site_config(path)['locales']

-

Locomotive::Wagon.generate :page, fullpath, self.options['path'], locales

-

end

-

end

-

desc 'snippet SLUG', 'Create a snippet'

-

long_desc <<-LONGDESC

-

Create a snippet. The generator will ask for the extension (liquid or haml) and also

-

if the snippet is localized or not.

-

Example:

-

* wagon generate snippet footer

-

LONGDESC

-

def snippet(slug)

-

if path = check_path!

-

locales = self.site_config(path)['locales']

-

Locomotive::Wagon.generate :snippet, slug, self.options['path'], locales

-

end

-

end

-

protected

-

# Read the YAML config file of a LocomotiveCMS site.

-

# The path should be returned by the check_path! method first.

-

#

-

# @param [ String ] path The full path to a LocomotiveCMS site.

-

#

-

# @return [ Hash ] The site

-

#

-

def site_config(path = nil)

-

YAML.load_file(File.join(path, 'config', 'site.yml'))

-

end

-

end

-

class Main < Thor

-

include Helpers::CheckPath

-

include Helpers::ForceColor

-

class_option :force_color, aliases: '-c', type: :boolean, default: false, desc: 'Whether or not to use ANSI color in the output.'

-

desc 'version', 'Version of the LocomotiveCMS Wagon'

-

def version

-

require 'locomotive/wagon/version'

-

say Locomotive::Wagon::VERSION

-

end

-

desc 'generate RESOURCE ARGUMENTS', 'Generates a content_type, page, or snippet'

-

long_desc <<-LONGDESC

-

Generates a content_type, page, or snippet

-

RESOURCE can be set to content_type, page, or snippet.

-

Use wagon generate help [RESOURCE] for usage information and examples.

-

LONGDESC

-

subcommand 'generate', Generate

-

desc 'init NAME [PATH] [OPTIONS]', 'Create a brand new site'

-

method_option :template, aliases: '-t', type: 'string', default: 'blank', desc: 'instead of building from a blank site, you can also have a pre-fetched site from a template (see the templates command)'

-

method_option :lib, aliases: '-l', type: 'string', desc: 'Path to an external ruby lib or generator'

-

method_option :skip_bundle, type: 'boolean', default: false, desc: "Don't run bundle install"

-

method_option :verbose, aliases: '-v', type: 'boolean', default: false, desc: 'display the full error stack trace if an error occurs'

-

def init(name, path = '.', generator_options = nil)

-

force_color_if_asked(options)

-

require 'locomotive/wagon/generators/site'

-

require File.expand_path(options[:lib]) if options[:lib]

-

generator = Locomotive::Wagon::Generators::Site.get(options[:template])

-

if generator.nil?

-

say "Unknown site template '#{options[:template]}'", :red

-

else

-

begin

-

if Locomotive::Wagon.init(name, path, options[:skip_bundle].to_s, generator, generator_options)

-

self.print_next_instructions_when_site_created(name, path, options[:skip_bundle])

-

end

-

rescue GeneratorException => e

-

self.print_exception(e, options[:verbose])

-

exit(1)

-

end

-

end

-

end

-

desc 'clone NAME HOST [PATH]', 'Clone a remote LocomotiveCMS site'

-

method_option :verbose, aliases: '-v', type: 'boolean', default: false, desc: 'display the full error stack trace if an error occurs'

-

method_option :email, aliases: '-e', desc: 'email of an administrator account'

-

method_option :password, aliases: '-p', desc: 'password of an administrator account'

-

method_option :api_key, aliases: '-a', desc: 'api key of an administrator account'

-

def clone(name, host, path = '.')

-

begin

-

if Locomotive::Wagon.clone(name, path, { host: host }.merge(options))

-

self.print_next_instructions_when_site_created(name, path)

-

end

-

rescue Exception => e

-

self.print_exception(e, options[:verbose])

-

exit(1)

-

end

-

end

-

desc 'list_templates', 'List all the templates to create either a site or a content type'

-

method_option :lib, aliases: '-l', type: 'string', desc: 'Path to an external ruby lib or generator'

-

method_option :json, aliases: '-j', type: :boolean, default: false, desc: 'Output the list in JSON'

-

def list_templates

-

force_color_if_asked(options)

-

require 'locomotive/wagon/generators/site'

-

require File.expand_path(options[:lib]) if options[:lib]

-

if Locomotive::Wagon::Generators::Site.empty?

-

say 'No templates', :red

-

elsif !options[:json]

-

Locomotive::Wagon::Generators::Site.list.each do |info|

-

say info.name, :bold, false

-

say " - #{info.description}" unless info.description.blank?

-

end

-

else

-

say Locomotive::Wagon::Generators::Site.list_to_json

-

end

-

end

-

desc 'serve [PATH]', 'Serve a site from the file system'

-

method_option :host, aliases: '-h', type: 'string', default: '0.0.0.0', desc: 'The host (address) of the Thin server'

-

method_option :port, aliases: '-p', type: 'string', default: '3333', desc: 'The port of the Thin server'

-

method_option :daemonize, aliases: '-d', type: 'boolean', default: false, desc: 'Run daemonized Thin server in the background'

-

method_option :force, aliases: '-f', type: 'boolean', default: false, desc: 'Stop the current daemonized Thin server if found before starting a new one'

-

method_option :verbose, aliases: '-v', type: 'boolean', default: false, desc: 'display the full error stack trace if an error occurs'

-

def serve(path = '.')

-

parent_pid = Process.pid

-

force_color_if_asked(options)

-

if check_path!(path)

-

begin

-

Locomotive::Wagon.serve(path, options)

-

rescue SystemExit => e

-

if parent_pid == Process.pid

-

say "Your site is served now.", :green

-

end

-

rescue Exception => e

-

self.print_exception(e, options[:verbose])

-

exit(1)

-

end

-

end

-

end

-

desc 'stop [PATH]', 'Stop serving a site previously launched by the serve command with the -d option'

-

def stop(path = '.')

-

force_color_if_asked(options)

-

if check_path!(path)

-

begin

-

Locomotive::Wagon.stop(path)

-

say "Your site is not served anymore.", :green

-

rescue Exception => e

-

say e.message, :red

-

exit(1)

-

end

-

end

-

end

-

desc 'push ENV [PATH]', 'Push a site to a remote LocomotiveCMS Engine'

-

method_option :resources, aliases: '-r', type: 'array', default: nil, desc: 'Only push the resource(s) passed in argument'

-

method_option :force, aliases: '-f', type: 'boolean', default: false, desc: 'Force the push of a resource'

-

method_option :translations, aliases: '-t', type: 'boolean', default: false, desc: 'Push the local translations (by default, they are not)'

-

method_option :data, aliases: '-d', type: 'boolean', default: false, desc: 'Push the content entries and the editable elements (by default, they are not)'

-

method_option :verbose, aliases: '-v', type: 'boolean', default: false, desc: 'display the full error stack trace if an error occurs'

-

def push(env, path = '.')

-

if check_path!(path)

-

if connection_info = self.retrieve_connection_info(env, path)

-

begin

-

Locomotive::Wagon.push(path, connection_info, options)

-

rescue Exception => e

-

self.print_exception(e, options[:verbose])

-

exit(1)

-

end

-

end

-

end

-

end

-

desc 'pull ENV [PATH]', 'Pull a site from a remote LocomotiveCMS Engine'

-

method_option :resources, aliases: '-r', type: 'array', default: nil, desc: 'Only pull the resource(s) passed in argument'

-

method_option :verbose, aliases: '-v', type: 'boolean', default: false, desc: 'display the full error stack trace if an error occurs'

-

def pull(env, path = '.')

-

if check_path!(path)

-

if connection_info = self.retrieve_connection_info(env, path)

-

begin

-

Locomotive::Wagon.pull(path, connection_info, options)

-

rescue Exception => e

-

self.print_exception(e, options[:verbose])

-

exit(1)

-

end

-

end

-

end

-

end

-

desc 'destroy ENV [PATH]', 'Destroy a remote LocomotiveCMS Engine'

-

def destroy(env, path = '.')

-

if check_path!(path)

-

if connection_info = self.retrieve_connection_info(env, path)

-

if ask('Are you sure ?', limited_to: %w(yes no)) == 'yes'

-

Locomotive::Wagon.destroy(path, connection_info)

-

else

-

say 'The destroy operation has been cancelled', :red

-

exit(1)

-

end

-

end

-

end

-

end

-

protected

-

# Print a nice message when a site has been created.

-

#

-

# @param [ String ] name The name of the site

-

# @param [ String ] path The path of the local site

-

# @param [ Boolean ] skip_bundle Do not run bundle install

-

#

-

def print_next_instructions_when_site_created(name, path, skip_bundle)

-

say "\nCongratulations, your site \"#{name}\" has been created successfully !", :green

-

say 'Next steps:', :bold

-

next_instructions = "\tcd #{path}/#{name}\n\t"

-

next_instructions += "bundle install\n\t" if skip_bundle

-

next_instructions += "bundle exec wagon serve\n\topen http://0.0.0.0:3333"

-

say next_instructions

-

end

-

# Print the exception.

-

#

-

# @param [ Object ] exception The raised exception

-

# @param [ Boolean ] verbose Print the full backtrace if true

-

#

-

def print_exception(exception, verbose)

-

say exception.message, :red

-

if verbose

-

say "\t" + exception.backtrace.join("\n\t")

-

end

-

end

-

# From a site specified by a path, retrieve the information of the connection

-

# for a environment located in the config/deploy.yml file of the site.

-

#

-

# @param [ String ] env The environment (development, staging, production, ...etc)

-

# @param [ String ] path The path of the local site

-

#

-

# @return [ Hash ] The information of the connection or nil if errors

-

#

-

def retrieve_connection_info(env, path)

-

require 'active_support/core_ext/hash'

-

require 'erb'

-

connection_info = nil

-

begin

-

path_to_deploy_file = File.join(path, 'config', 'deploy.yml')

-

env_parsed_deploy_file = ERB.new(File.open(path_to_deploy_file).read).result

-

connection_info = YAML::load(env_parsed_deploy_file)[env.to_s].with_indifferent_access

-

if connection_info[:ssl] && !connection_info[:host].start_with?('https')

-

connection_info[:host] = 'https://' + connection_info[:host]

-

end

-

if connection_info.nil?

-

raise "No #{env.to_s} environment found in the config/deploy.yml file"

-

end

-

rescue Exception => e

-

say "Unable to read the information about the remote LocomotiveCMS site (#{e.message})", :red

-

end

-

connection_info

-

end

-

end

-

end

-

end

-

end

@@	@@ -1,320 +0,0 @@
-	require_relative 'helpers/check_path'
-	require_relative 'helpers/force_color'
-	require_relative 'generate'
-
-	module Locomotive
-	module Wagon
-	module CLI
-	class Generate < Thor
-	include Helpers::CheckPath
-
-	class_option :path, aliases: '-p', type: :string, default: '.', optional: true, desc: 'if your LocomotiveCMS site is not in the current path'
-
-	desc 'content_type SLUG FIELDS', 'Creates a content type with the specified slug and fields.'
-	long_desc <<-LONGDESC
-	Creates a content type with the specified slug and fields.
-
-	SLUG should be plural, lowercase, and underscored.
-
-	FIELDS format: field_1[:TYPE][:REQUIRED] field_2[:TYPE][:REQUIRED] ...
-
-	TYPE values: string, text, integer, float, boolean, email, date, date_time, file, tags, select, belongs_to, has_many, or many_to_many. Default is string.
-
-	To require a field, set REQUIRED to true. The first field is required by default.
-
-	Examples:
-
-	* wagon generate content_type posts title published_at:date_time:true body:text
-
-	* wagon generate content_type products title price:float photo:file category:belongs_to:true
-	LONGDESC
-	def content_type(name, *fields)
-	say('The fields are missing', :red) and return false if fields.empty?
-
-	if check_path!
-	Locomotive::Wagon.generate :content_type, name, self.options['path'], fields
-	end
-	end
-
-	desc 'page FULLPATH', 'Create a page. No need to pass an extension to the FULLPATH arg'
-	long_desc <<-LONGDESC
-	Create a page. The generator will ask for the extension (liquid or haml) and also
-	if the page is localized or not.
-
-	Examples:
-
-	* wagon generate page contact
-
-	* wagon generate page about_us/me
-	LONGDESC
-	def page(fullpath)
-	if path = check_path!
-	locales = self.site_config(path)['locales']
-	Locomotive::Wagon.generate :page, fullpath, self.options['path'], locales
-	end
-	end
-
-	desc 'snippet SLUG', 'Create a snippet'
-	long_desc <<-LONGDESC
-	Create a snippet. The generator will ask for the extension (liquid or haml) and also
-	if the snippet is localized or not.
-
-	Example:
-
-	* wagon generate snippet footer
-	LONGDESC
-	def snippet(slug)
-	if path = check_path!
-	locales = self.site_config(path)['locales']
-	Locomotive::Wagon.generate :snippet, slug, self.options['path'], locales
-	end
-	end
-
-	protected
-
-	# Read the YAML config file of a LocomotiveCMS site.
-	# The path should be returned by the check_path! method first.
-	#
-	# @param [ String ] path The full path to a LocomotiveCMS site.
-	#
-	# @return [ Hash ] The site
-	#
-	def site_config(path = nil)
-	YAML.load_file(File.join(path, 'config', 'site.yml'))
-	end
-	end
-
-	class Main < Thor
-
-	include Helpers::CheckPath
-	include Helpers::ForceColor
-
-	class_option :force_color, aliases: '-c', type: :boolean, default: false, desc: 'Whether or not to use ANSI color in the output.'
-
-	desc 'version', 'Version of the LocomotiveCMS Wagon'
-	def version
-	require 'locomotive/wagon/version'
-	say Locomotive::Wagon::VERSION
-	end
-
-	desc 'generate RESOURCE ARGUMENTS', 'Generates a content_type, page, or snippet'
-	long_desc <<-LONGDESC
-	Generates a content_type, page, or snippet
-
-	RESOURCE can be set to content_type, page, or snippet.
-
-	Use wagon generate help [RESOURCE] for usage information and examples.
-	LONGDESC
-	subcommand 'generate', Generate
-
-	desc 'init NAME [PATH] [OPTIONS]', 'Create a brand new site'
-	method_option :template, aliases: '-t', type: 'string', default: 'blank', desc: 'instead of building from a blank site, you can also have a pre-fetched site from a template (see the templates command)'
-	method_option :lib, aliases: '-l', type: 'string', desc: 'Path to an external ruby lib or generator'
-	method_option :skip_bundle, type: 'boolean', default: false, desc: "Don't run bundle install"
-	method_option :verbose, aliases: '-v', type: 'boolean', default: false, desc: 'display the full error stack trace if an error occurs'
-	def init(name, path = '.', generator_options = nil)
-	force_color_if_asked(options)
-	require 'locomotive/wagon/generators/site'
-	require File.expand_path(options[:lib]) if options[:lib]
-	generator = Locomotive::Wagon::Generators::Site.get(options[:template])
-	if generator.nil?
-	say "Unknown site template '#{options[:template]}'", :red
-	else
-	begin
-	if Locomotive::Wagon.init(name, path, options[:skip_bundle].to_s, generator, generator_options)
-	self.print_next_instructions_when_site_created(name, path, options[:skip_bundle])
-	end
-	rescue GeneratorException => e
-	self.print_exception(e, options[:verbose])
-	exit(1)
-	end
-	end
-	end
-
-	desc 'clone NAME HOST [PATH]', 'Clone a remote LocomotiveCMS site'
-	method_option :verbose, aliases: '-v', type: 'boolean', default: false, desc: 'display the full error stack trace if an error occurs'
-	method_option :email, aliases: '-e', desc: 'email of an administrator account'
-	method_option :password, aliases: '-p', desc: 'password of an administrator account'
-	method_option :api_key, aliases: '-a', desc: 'api key of an administrator account'
-	def clone(name, host, path = '.')
-	begin
-	if Locomotive::Wagon.clone(name, path, { host: host }.merge(options))
-	self.print_next_instructions_when_site_created(name, path)
-	end
-	rescue Exception => e
-	self.print_exception(e, options[:verbose])
-	exit(1)
-	end
-	end
-
-	desc 'list_templates', 'List all the templates to create either a site or a content type'
-	method_option :lib, aliases: '-l', type: 'string', desc: 'Path to an external ruby lib or generator'
-	method_option :json, aliases: '-j', type: :boolean, default: false, desc: 'Output the list in JSON'
-	def list_templates
-	force_color_if_asked(options)
-	require 'locomotive/wagon/generators/site'
-	require File.expand_path(options[:lib]) if options[:lib]
-	if Locomotive::Wagon::Generators::Site.empty?
-	say 'No templates', :red
-	elsif !options[:json]
-	Locomotive::Wagon::Generators::Site.list.each do \|info\|
-	say info.name, :bold, false
-	say " - #{info.description}" unless info.description.blank?
-	end
-	else
-	say Locomotive::Wagon::Generators::Site.list_to_json
-	end
-	end
-
-	desc 'serve [PATH]', 'Serve a site from the file system'
-	method_option :host, aliases: '-h', type: 'string', default: '0.0.0.0', desc: 'The host (address) of the Thin server'
-	method_option :port, aliases: '-p', type: 'string', default: '3333', desc: 'The port of the Thin server'
-	method_option :daemonize, aliases: '-d', type: 'boolean', default: false, desc: 'Run daemonized Thin server in the background'
-	method_option :force, aliases: '-f', type: 'boolean', default: false, desc: 'Stop the current daemonized Thin server if found before starting a new one'
-	method_option :verbose, aliases: '-v', type: 'boolean', default: false, desc: 'display the full error stack trace if an error occurs'
-	def serve(path = '.')
-	parent_pid = Process.pid
-	force_color_if_asked(options)
-	if check_path!(path)
-	begin
-	Locomotive::Wagon.serve(path, options)
-	rescue SystemExit => e
-	if parent_pid == Process.pid
-	say "Your site is served now.", :green
-	end
-	rescue Exception => e
-	self.print_exception(e, options[:verbose])
-	exit(1)
-	end
-	end
-	end
-
-	desc 'stop [PATH]', 'Stop serving a site previously launched by the serve command with the -d option'
-	def stop(path = '.')
-	force_color_if_asked(options)
-	if check_path!(path)
-	begin
-	Locomotive::Wagon.stop(path)
-	say "Your site is not served anymore.", :green
-	rescue Exception => e
-	say e.message, :red
-	exit(1)
-	end
-	end
-	end
-
-	desc 'push ENV [PATH]', 'Push a site to a remote LocomotiveCMS Engine'
-	method_option :resources, aliases: '-r', type: 'array', default: nil, desc: 'Only push the resource(s) passed in argument'
-	method_option :force, aliases: '-f', type: 'boolean', default: false, desc: 'Force the push of a resource'
-	method_option :translations, aliases: '-t', type: 'boolean', default: false, desc: 'Push the local translations (by default, they are not)'
-	method_option :data, aliases: '-d', type: 'boolean', default: false, desc: 'Push the content entries and the editable elements (by default, they are not)'
-	method_option :verbose, aliases: '-v', type: 'boolean', default: false, desc: 'display the full error stack trace if an error occurs'
-	def push(env, path = '.')
-	if check_path!(path)
-	if connection_info = self.retrieve_connection_info(env, path)
-	begin
-	Locomotive::Wagon.push(path, connection_info, options)
-	rescue Exception => e
-	self.print_exception(e, options[:verbose])
-	exit(1)
-	end
-	end
-	end
-	end
-
-	desc 'pull ENV [PATH]', 'Pull a site from a remote LocomotiveCMS Engine'
-	method_option :resources, aliases: '-r', type: 'array', default: nil, desc: 'Only pull the resource(s) passed in argument'
-	method_option :verbose, aliases: '-v', type: 'boolean', default: false, desc: 'display the full error stack trace if an error occurs'
-	def pull(env, path = '.')
-	if check_path!(path)
-	if connection_info = self.retrieve_connection_info(env, path)
-	begin
-	Locomotive::Wagon.pull(path, connection_info, options)
-	rescue Exception => e
-	self.print_exception(e, options[:verbose])
-	exit(1)
-	end
-	end
-	end
-	end
-
-	desc 'destroy ENV [PATH]', 'Destroy a remote LocomotiveCMS Engine'
-	def destroy(env, path = '.')
-	if check_path!(path)
-	if connection_info = self.retrieve_connection_info(env, path)
-	if ask('Are you sure ?', limited_to: %w(yes no)) == 'yes'
-	Locomotive::Wagon.destroy(path, connection_info)
-	else
-	say 'The destroy operation has been cancelled', :red
-	exit(1)
-	end
-	end
-	end
-	end
-
-	protected
-
-	# Print a nice message when a site has been created.
-	#
-	# @param [ String ] name The name of the site
-	# @param [ String ] path The path of the local site
-	# @param [ Boolean ] skip_bundle Do not run bundle install
-	#
-	def print_next_instructions_when_site_created(name, path, skip_bundle)
-	say "\nCongratulations, your site \"#{name}\" has been created successfully !", :green
-	say 'Next steps:', :bold
-
-	next_instructions = "\tcd #{path}/#{name}\n\t"
-	next_instructions += "bundle install\n\t" if skip_bundle
-	next_instructions += "bundle exec wagon serve\n\topen http://0.0.0.0:3333"
-
-	say next_instructions
-	end
-
-	# Print the exception.
-	#
-	# @param [ Object ] exception The raised exception
-	# @param [ Boolean ] verbose Print the full backtrace if true
-	#
-	def print_exception(exception, verbose)
-	say exception.message, :red
-	if verbose
-	say "\t" + exception.backtrace.join("\n\t")
-	end
-	end
-
-	# From a site specified by a path, retrieve the information of the connection
-	# for a environment located in the config/deploy.yml file of the site.
-	#
-	# @param [ String ] env The environment (development, staging, production, ...etc)
-	# @param [ String ] path The path of the local site
-	#
-	# @return [ Hash ] The information of the connection or nil if errors
-	#
-	def retrieve_connection_info(env, path)
-	require 'active_support/core_ext/hash'
-	require 'erb'
-	connection_info = nil
-	begin
-	path_to_deploy_file = File.join(path, 'config', 'deploy.yml')
-	env_parsed_deploy_file = ERB.new(File.open(path_to_deploy_file).read).result
-	connection_info = YAML::load(env_parsed_deploy_file)[env.to_s].with_indifferent_access
-
-	if connection_info[:ssl] && !connection_info[:host].start_with?('https')
-	connection_info[:host] = 'https://' + connection_info[:host]
-	end
-
-	if connection_info.nil?
-	raise "No #{env.to_s} environment found in the config/deploy.yml file"
-	end
-	rescue Exception => e
-	say "Unable to read the information about the remote LocomotiveCMS site (#{e.message})", :red
-	end
-	connection_info
-	end
-
-	end
-
-	end
-	end
-	end

remove duplicate task