278 #92 automatic-command merge into automatic command

id774 · id774 · commit 8cfeccd58a3f · 2013-03-19T10:57:22.000+09:00
diff --git a/bin/automatic b/bin/automatic
@@ -3,24 +3,108 @@
 # Name::      Automatic::Ruby
 # Author::    774 <http://id774.net>
 # Created::   Feb 18, 2012
-# Updated::   Jan 22, 2013
-# Copyright:: 774 Copyright (c) 2012
+# Updated::   Mar 19, 2013
+# Copyright:: 774 Copyright (c) 2012-2013
 # License::   Licensed under the GNU GENERAL PUBLIC LICENSE, Version 3.0.
 
 root_dir = File.expand_path("..", File.dirname(__FILE__))
 $:.unshift root_dir
 $:.unshift root_dir + '/lib'
+
 require 'automatic'
+require 'rubygems'
 require 'optparse'
+require 'fileutils'
+require 'feedbag'
+require 'pp'
 
 recipe_path = ""
-parser = OptionParser.new { |parser|
+
+def abort_with_usage(subcommand, message)
+  top_filename = File.basename($0)
+  abort("Usage: #{top_filename} #{subcommand} #{message}")
+end
+
+subparsers = {
+  'scaffold' => lambda {|argv|
+    Dir::entries(root_dir + '/plugins').sort.each {|path|
+      dir = (File.expand_path('~/.automatic/plugins/' + path))
+      unless File.exist?(dir)
+        FileUtils.mkdir_p(dir)
+        puts "Creating #{dir}"
+      end
+    }
+    dir = (File.expand_path('~/.automatic/assets'))
+    unless File.exist?(dir)
+      FileUtils.mkdir_p(dir)
+      FileUtils.cp_r(root_dir + '/assets/siteinfo', dir + '/siteinfo')
+      puts "Creating #{dir}"
+    end
+    dir = (File.expand_path('~/.automatic/config'))
+    unless File.exist?(dir)
+      FileUtils.mkdir_p(dir)
+      FileUtils.cp_r(root_dir + '/config', dir + '/example')
+      puts "Creating #{dir}"
+    end
+    dir = (File.expand_path('~/.automatic/db'))
+    unless File.exist?(dir)
+      FileUtils.mkdir_p(dir)
+      puts "Creating #{dir}"
+    end
+  },
+  'unscaffold' => lambda {|argv|
+    dir = (File.expand_path('~/.automatic'))
+    if File.directory?(dir)
+      puts "Removing #{dir}"
+      FileUtils.rm_r(dir)
+    end
+  },
+  'autodiscovery' => lambda {|argv|
+    url = argv.shift || abort_with_usage("autodiscovery", "<url>")
+    pp Feedbag.find(url)
+  },
+  'feedparser' => lambda {|argv|
+    require 'automatic/feed_parser'
+    url = argv.shift || abort_with_usage("feedparser", "<url>")
+    rss_results = Automatic::FeedParser.get(url)
+    pp rss_results
+  },
+  'inspect' => lambda {|argv|
+    require 'automatic/feed_parser'
+    url = argv.shift || abort_with_usage("inspect", "<url>")
+    feeds = Feedbag.find(url)
+    pp feeds
+    rss_results = Automatic::FeedParser.get(feeds.pop)
+    pp rss_results
+  },
+  'opmlparser' => lambda {|argv|
+    require 'automatic/opml'
+    path = argv.shift
+    if path.nil?
+      abort_with_usage("opmlparser", "<opml path>")
+    end
+    parser = Automatic::OPML::Parser.new(File.read(path))
+    parser.each_outline {|opml, o|
+      puts "#{o.xmlUrl}"
+    }
+  },
+  'log' => lambda {|argv|
+    require 'automatic/log'
+    level = argv.shift || abort_with_usage("log", "<level> <message>")
+    message = ARGV.shift
+    Automatic::Log.puts(level, message)
+  }
+}
+
+parser = OptionParser.new {|parser|
   parser.banner = "Usage: automatic [options] arg"
   parser.version = Automatic.const_get(:VERSION)
-  parser.separator "options:"
+  parser.separator "Args: #{subparsers.keys.join(", ")}"
+  parser.separator "Options:"
   parser.on('-c', '--config FILE', String,
             "recipe YAML file"){|c| recipe_path = c}
   parser.on('-h', '--help', "show this message") {
+
     puts parser
     exit
   }
@@ -35,15 +119,19 @@ rescue OptionParser::ParseError => err
   exit 1
 end
 
-if recipe_path.to_s.empty?
-  if ENV["AUTOMATIC_RUBY_ENV"] == "test"
-    recpie_path = File.join(root_dir, 'config/default.yml')
+unless recipe_path.to_s.empty?
+  recipe = Automatic::Recipe.new(recipe_path)
+  Automatic.run(:recipe   => recipe,
+                :root_dir => root_dir)
+else
+  parser.order!(ARGV)
+  unless ARGV.empty?
+    Hash.new {|h, k|
+      puts "No such subcommand: #{k}"
+      exit 1
+    }.update(subparsers)[ARGV.shift].call(ARGV)
   else
     puts parser
     exit 0
   end
 end
-
-recipe = Automatic::Recipe.new(recipe_path)
-Automatic.run(:recipe   => recipe,
-              :root_dir => root_dir)
diff --git a/bin/automatic-config b/bin/automatic-config
diff --git a/script/build b/script/build
@@ -21,7 +21,7 @@ kickstart() {
     export RACK_ENV="test"
     export DATABASE_URL="$RACK_ROOT/db"
     ruby -v
-    test_config log info started.
+    test_subcommand log info started.
 }
 
 exec_rspec() {
@@ -34,7 +34,7 @@ run_test() {
     while [ $# -gt 0 ]
     do
         $RACK_ROOT/bin/automatic -c $1
-        test_config log info $?
+        test_subcommand log info $?
         shift
     done
 }
@@ -47,21 +47,21 @@ load_tests() {
     unset YAML_FILE
 }
 
-test_config() {
-    $RACK_ROOT/bin/automatic-config $*
+test_subcommand() {
+    $RACK_ROOT/bin/automatic $*
 }
 
 test_scaffold() {
-    test_config scaffold
-    test_config unscaffold
+    test_subcommand scaffold
+    test_subcommand unscaffold
 }
 
 added_tests() {
     test -f $HOME/.automatic || test_scaffold
-    test_config autodiscovery http://blog.id774.net/blogs/
-    test_config opmlparser "$RACK_ROOT/test/fixtures/sampleOPML.xml"
-    test_config feedparser http://blog.id774.net/blogs/feed/ > /dev/null
-    test_config log info finished.
+    test_subcommand autodiscovery http://blog.id774.net/blogs/
+    test_subcommand opmlparser "$RACK_ROOT/test/fixtures/sampleOPML.xml"
+    test_subcommand feedparser http://blog.id774.net/blogs/feed/ > /dev/null
+    test_subcommand log info finished.
 }
 
 integration() {
