Class: Nanoc3::CLI::Base

237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258

# File 'lib/nanoc3/cli/base.rb', line 237

def handle_option(option)
  case option
  when :version
    gem_info = defined?(Gem) ? "with RubyGems #{Gem::VERSION}" : "without RubyGems"
    engine   = defined?(RUBY_ENGINE) ? RUBY_ENGINE : "ruby"

    puts "nanoc #{Nanoc3::VERSION} (c) 2007-2011 Denis Defreyne."
    puts "Running #{engine} #{RUBY_VERSION} (#{RUBY_RELEASE_DATE}) on #{RUBY_PLATFORM} #{gem_info}"
    exit 0
  when :verbose
    Nanoc3::CLI::Logger.instance.level = :low
  when :debug
    $DEBUG = true
  when :warn
    $-w = true
  when :no-color'
    Nanoc3::CLI::Logger.instance.color = false
  when :help
    show_help
    exit 0
  end
end

90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131

# File 'lib/nanoc3/cli/base.rb', line 90

def print_error(error)
  $stderr.puts

  # Header
  $stderr.puts '+--- /!\ ERROR /!\ -------------------------------------------+'
  $stderr.puts '| An exception occured while running nanoc. If you think this |'
  $stderr.puts '| is a bug in nanoc, please do report it at                   |'
  $stderr.puts '| <http://projects.stoneship.org/trac/nanoc/newticket> --     |'
  $stderr.puts '| thanks in advance!                                          |'
  $stderr.puts '+-------------------------------------------------------------+'

  # Exception and resolution (if any)
  $stderr.puts
  $stderr.puts '=== MESSAGE:'
  $stderr.puts
  $stderr.puts "#{error.class}: #{error.message}"
  resolution = self.resolution_for(error)
  $stderr.puts "#{resolution}" if resolution

  # Compilation stack
  $stderr.puts
  $stderr.puts '=== COMPILATION STACK:'
  $stderr.puts
  if ((self.site && self.site.compiler.stack) || []).empty?
    $stderr.puts "  (empty)"
  else
    self.site.compiler.stack.reverse.each do |obj|
      if obj.is_a?(Nanoc3::ItemRep)
        $stderr.puts "  - [item]   #{obj.item.identifier} (rep #{obj.name})"
      else # layout
        $stderr.puts "  - [layout] #{obj.identifier}"
      end
    end
  end

  # Backtrace
  require 'enumerator'
  $stderr.puts
  $stderr.puts '=== BACKTRACE:'
  $stderr.puts
  $stderr.puts error.backtrace.to_enum(:each_with_index).map { |item, index| "  #{index}. #{item}" }.join("\n")
end

40
41
42
43
44
45
46

# File 'lib/nanoc3/cli/base.rb', line 40

def require_site
  if site.nil?
    $stderr.puts 'The current working directory does not seem to be a ' +
                 'valid/complete nanoc site directory; aborting.'
    exit 1
  end
end

139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177

# File 'lib/nanoc3/cli/base.rb', line 139

def resolution_for(error)
  # FIXME this should probably go somewhere else so that 3rd-party code can add other gem names too
  gem_names = {
    'adsf'           => 'adsf',
    'bluecloth'      => 'bluecloth',
    'builder'        => 'builder',
    'coderay'        => 'coderay',
    'cri'            => 'cri',
    'erubis'         => 'erubis',
    'haml'           => 'haml',
    'json'           => 'json',
    'less'           => 'less',
    'markaby'        => 'markaby',
    'maruku'         => 'maruku',
    'mime/types'     => 'mime-types',
    'rack'           => 'rack',
    'rack/cache'     => 'rack-cache',
    'rainpress'      => 'rainpress',
    'rdiscount'      => 'rdiscount',
    'redcloth'       => 'redcloth',
    'rubypants'      => 'rubypants',
    'sass'           => 'sass',
    'w3c_validators' => 'w3c_validators'
  }

  case error
  when LoadError
    # Get gem name
    matches = error.message.match(/no such file to load -- ([^\s]+)/)
    return nil if matches.empty?
    lib_name = matches[1]
    gem_name = gem_names[$1]

    # Build message
    if gem_name
      "Try installing the '#{gem_name}' gem (`gem install #{gem_name}`) and then re-running the command."
    end
  end
end

67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

# File 'lib/nanoc3/cli/base.rb', line 67

def run(args)
  # Set exit handler
  [ 'INT', 'TERM' ].each do |signal|
    Signal.trap(signal) do
      puts
      exit!(0)
    end
  end

  super(args)
rescue Interrupt => e
  exit(1)
rescue StandardError, ScriptError => e
  print_error(e)
  exit(1)
end

186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204

# File 'lib/nanoc3/cli/base.rb', line 186

def set_vcs(vcs_name)
  # Skip if not possible
  return if vcs_name.nil? || site.nil?

  # Find VCS
  vcs_class = Nanoc3::Extra::VCS.named(vcs_name.to_sym)
  if vcs_class.nil?
    $stderr.puts "A VCS named #{vcs_name} was not found; aborting."
    exit 1
  end

  site.data_sources.each do |data_source|
    # Skip if not possible
    next if !data_source.respond_to?(:vcs=)

    # Set VCS
    data_source.vcs = vcs_class.new
  end
end

52
53
54
55
56
57
58
59
60
61
62
63
64

# File 'lib/nanoc3/cli/base.rb', line 52

def site
  # Load site if possible
  if File.file?('config.yaml') && (!self.instance_variable_defined?(:@site) || @site.nil?)
    begin
      @site = Nanoc3::Site.new('.')
    rescue Nanoc3::Errors::UnknownDataSource => e
      $stderr.puts "Unknown data source: #{e}"
      exit 1
    end
  end

  @site
end