yuyi 0.0.3

data/lib/yuyi/core.rb

@@ -0,0 +1,49 @@
+require 'tsort'
+
+class Array
+  def to_yaml_style
+    :inline
+  end
+end
+
+class Hash
+  include TSort
+  alias tsort_each_node each_key
+  def tsort_each_child(node, &block)
+    fetch(node).each(&block)
+  end
+
+  # https://github.com/rails/rails/blob/c48a0cac626b4e32d7abfa9f4f1fae16568157d9/activesupport/lib/active_support/core_ext/hash/keys.rb
+  #
+  # Destructively convert all keys to symbols, as long as they respond
+  # to +to_sym+. This includes the keys from the root hash and from all
+  # nested hashes.
+  #
+  def deep_symbolize_keys!
+    deep_transform_keys!{ |key| key.to_sym rescue key }
+  end
+
+  def deep_stringify_keys!
+    deep_transform_keys!{ |key| key.to_s rescue key }
+  end
+
+  # Destructively convert all keys by using the block operation.
+  # This includes the keys from the root hash and from all
+  # nested hashes.
+  #
+  def deep_transform_keys! &block
+    keys.each do |key|
+      value = delete(key)
+
+      self[yield(key)] = case value
+      when Hash
+        value.deep_transform_keys!(&block)
+      when Array
+        value.each{ |e| e.deep_transform_keys!(&block) rescue value }
+      else
+        value
+      end
+    end
+    self
+  end
+end

data/lib/yuyi/menu.rb

@@ -0,0 +1,202 @@
+class Yuyi::Menu
+  @rolls = {}
+  @pre_installs = []
+  @post_installs = []
+
+  # DSL Helper methods
+  #
+  def self.instance; @@instance; end
+
+  def self.object; @object; end
+  def object; self.class.object; end
+
+  def self.sources; @@instance.sources; end
+  def sources; @sources; end
+
+  def self.rolls; @rolls; end
+  def rolls; self.class.rolls; end
+
+  def self.path
+    @@instance.path
+  end
+  def path; @path; end
+
+  def self.on_the_menu? roll
+    @@instance.on_the_menu? roll
+  end
+
+  def self.find_roll name, options = {}
+    @@instance.find_roll name, options
+  end
+
+  def self.set_sources
+    @@instance.set_sources
+  end
+
+  def self.upgrade? boolean = nil
+    @@instance.upgrade? boolean
+  end
+
+  def self.options roll
+    @@instance.options roll
+  end
+
+  # Attempt to load a menu from a file path
+  # defaults to previously stored @path value
+  #
+  def self.load_from_file path = path
+    @object = begin
+      YAML.load(File.open(File.expand_path(path))).deep_symbolize_keys!
+    rescue
+      nil
+    end
+  end
+
+  # Add rolls to hash in format of {file_name: RollInstance}
+  # Called from yuyi/roll.rb#self.inherited
+  #
+  def self.add_roll file_name, klass
+    @rolls[file_name] = klass.new
+  end
+
+  def self.add_pre_install block
+    @pre_installs << block
+  end
+
+  def self.add_post_install block
+    @post_installs << block
+  end
+
+  def self.pre_install
+    @pre_installs.each { |b| b.call }
+  end
+
+  def self.post_install
+    @post_installs.each { |b| b.call }
+  end
+
+  # Get/Set upgrade flag
+  #
+  def upgrade? boolean = nil
+    @upgrade = boolean.nil? ? @upgrade : boolean
+  end
+
+  # Check if a roll is on the menu
+  #
+  def on_the_menu? roll
+    object[:rolls].keys.include? roll
+  end
+
+  # Download & create all sources on the menu
+  #
+  def set_sources
+    if object[:sources].empty?
+      Yuyi.say 'No rolls could be found because no sources were set in your menu.'
+      return
+    end
+
+    @sources = []
+    object[:sources].each do |source|
+      source.each do |name, path|
+        @sources << Yuyi::Source.new(name, path)
+      end
+    end
+
+    @sources
+  end
+
+  def options roll
+    object[:rolls][roll.file_name] || {}
+  end
+
+  # Initialize all the rolls in order
+  #
+  def order_rolls
+    Yuyi::Menu.pre_install
+
+    sorted_rolls.each do |file_name|
+      rolls[file_name].order
+    end
+
+    Yuyi::Menu.post_install
+  end
+
+  # Find the best roll in the source to be added
+  #
+  def find_roll name, options = {}
+    options ||= {}
+    # return specific source roll if specified in the menu
+    #
+    if source = options[:source]
+      require_roll name, File.join(source, name.to_s)
+      return
+
+    # look through the sources for the first roll that matches.
+    # sources are listed in the menu in order of priority
+    else
+      sources.each do |source|
+        if path = source.available_rolls[name]
+          require_roll name, path
+          return
+        end
+      end
+    end
+
+    # no roll was found
+    Yuyi.say "You ordered the '#{name}' roll off the menu, but we are fresh out...", :type => :fail
+    Yuyi.say 'Check your menu to make sure a source with your roll is listed.', :type => :warn
+    Yuyi.say
+  end
+
+private
+
+    def initialize path
+      return unless Yuyi::Menu.load_from_file path
+
+      @path = path
+      @@instance = self
+
+      set_sources
+      set_rolls
+    end
+
+    # Create all rolls on the menu
+    #
+    def set_rolls
+      object[:rolls].each do |name, options|
+        find_roll name, options
+      end
+    end
+
+    # Require a single roll
+    #
+    def require_roll name, path
+      # check if already on the roll for when requiring dependencies
+      return if roll_loaded? name
+
+      begin
+        require path
+      rescue LoadError
+        Yuyi.say "There was a problem loading the `#{name}` roll from `#{path}`", :type => :fail
+        Yuyi.say 'If this problem continues, please log an issue on the Yuyi github page.', :type => :warn
+        Yuyi.say
+      end
+    end
+
+    # Check if roll has already been required & loaded
+    #
+    def roll_loaded? roll
+      rolls.keys.include? roll
+    end
+
+    # Return an array of the topologically sorted rolls from the menu
+    #
+    def sorted_rolls
+      tsort_hash = {}
+      rolls.each do |file_name, roll|
+        tsort_hash[file_name.to_s] = roll.dependencies.map(&:to_s)
+      end
+
+      tsort_hash.tsort.map(&:to_sym)
+    end
+end

data/lib/yuyi/roll.rb

@@ -0,0 +1,153 @@
+# encoding: utf-8
+
+class Yuyi::Roll
+
+  # DSL API Methods
+  #
+  def self.title title = nil
+    title ? @title = title : @title
+  end
+
+  def self.file_name file_name = nil
+    file_name ? @file_name = file_name : @file_name
+  end
+
+  def self.pre_install &block
+    Yuyi::Menu.add_pre_install block
+  end
+
+  def self.install &block
+    @install ||= block
+  end
+
+  def self.post_install &block
+    Yuyi::Menu.add_post_install block
+  end
+
+  def self.uninstall &block
+    @uninstall ||= block
+  end
+
+  def self.upgrade &block
+    @upgrade ||= block
+  end
+
+  def self.installed? &block
+    @installed ||= block
+  end
+
+  def self.dependencies *dependencies
+    @dependencies ||= []
+
+    unless dependencies.empty?
+      @dependencies |= dependencies
+      require_dependencies
+    end
+
+    @dependencies
+  end
+
+  # set option definitions
+  def self.options arg = {}
+    @option_defs ||= arg
+  end
+
+  # DSL Helper methods
+  #
+  def title; self.class.title; end
+  def file_name; self.class.file_name; end
+
+  # return option definitions
+  def option_defs
+    self.class.options
+  end
+
+  def write_to_file file, *text
+    Yuyi.write_to_file file, *text
+  end
+
+  def delete_from_file file, *text
+    Yuyi.delete_from_file file, *text
+  end
+
+  def options
+    option_defaults = {}
+    option_defs.each do |roll, option_settings|
+      option_defaults[roll] = option_settings[:default]
+    end
+
+    option_defaults.merge Yuyi::Menu.options(self)
+  end
+
+  def dependencies
+    self.class.dependencies
+  end
+
+  # Run the roll
+  #
+  def order
+    if installed?
+      if options[:uninstall]
+        Yuyi.say "🍣\s Uninstalling #{title}...", :color => 33
+        uninstall
+      elsif upgrade?
+        Yuyi.say "🍣\s Upgrading #{title}", :color => 36
+        upgrade
+      end
+    else
+      Yuyi.say "🍣\s Installing #{title}...", :color => 32
+      install
+    end
+  end
+
+private
+
+    def upgrade?
+      Yuyi::Menu.upgrade?
+    end
+
+    # Add to global collection of rolls
+    #
+    def self.inherited klass
+      # convert class name to a human readble title-cased string
+      klass.title klass.to_s.gsub(/(?=[A-Z])/, ' ').strip
+
+      # convert absolute path to a file name symbol
+      klass.file_name caller.first[/[a-z_]+?(?=\.rb)/].to_sym
+
+      Yuyi::Menu.add_roll klass.file_name, klass
+    end
+
+    def self.require_dependencies
+      @dependencies.each do |roll|
+        unless Yuyi::Menu.on_the_menu? roll
+          Yuyi::Menu.find_roll roll
+        end
+      end
+    end
+
+    def install
+      instance_eval(&self.class.install)
+    end
+
+    def uninstall
+      instance_eval(&self.class.uninstall)
+    end
+
+    def upgrade
+      instance_eval(&self.class.upgrade)
+    end
+
+    def installed?
+      !!instance_eval(&self.class.installed?)
+    end
+
+    # Helpers for Yuyi Cli methods
+    def say *args; Yuyi.say *args; end
+    def run *args; Yuyi.run *args; end
+    def command? *args; Yuyi.command? *args; end
+
+    def on_the_menu? roll
+      Yuyi::Menu.on_the_menu? roll
+    end
+end

data/lib/yuyi/source.rb

@@ -0,0 +1,107 @@
+require 'open-uri'
+require 'pathname'
+require 'tmpdir'
+
+class Yuyi::Source
+  ROLL_FILE_GLOB = '**/*.rb'
+
+  def available_rolls; @available_rolls end
+
+private
+
+    def self.create_main_tmp_dir
+      @@root_tmp_dir = Dir.mktmpdir
+      $: << @@root_tmp_dir
+    end
+
+    def initialize name, path
+      @available_rolls = {}
+      @name = name
+      @path = path
+
+      create_tmp_dir
+      download_source
+      get_available_rolls
+    end
+
+    def create_tmp_dir
+      dir = FileUtils.mkdir_p(File.join(@@root_tmp_dir, @name.to_s))
+
+      # RUBY_VERSION
+      # mkdir_p <= 1.8 returns a string
+      # mkdir_p >= 1.9 returns an array
+      #
+      dir = case dir
+      when Array
+        dir.first
+      else
+        dir
+      end
+
+      @tmp_dir = dir
+    end
+
+    def download_source
+      # if remote source
+      if URI.parse(@path).scheme
+        zip_file = File.join(@tmp_dir, @name.to_s)
+
+        # Download the file and save it to the tmp directory
+        open zip_file, 'w' do |save_file|
+          begin
+            save_file << open(@path).read
+          rescue
+            Yuyi.say "Could not open the `#{@name}` source.  Please check that the path is correct.", :type => :fail
+            Yuyi.say "If you continue to have issues, your source server may be down.", :type => :warn
+            return
+          end
+        end
+
+        unzip_source zip_file
+
+      # if local source
+      else
+        path = File.expand_path(@path)
+
+        # if file, expect archive file and unzip it
+        if File.file? path
+          unzip_source path if File.file? path
+
+        # if dir, copy rolls out of it
+        else
+          FileUtils.cp_r Dir.glob(File.join(path, ROLL_FILE_GLOB)), @tmp_dir
+        end
+      end
+    end
+
+    def unzip_source file
+      # unzip file
+      unzip = if RUBY_PLATFORM =~ /darwin/
+        Yuyi.run "tar -xf #{file} -C #{@tmp_dir}", :boolean => true
+      else
+        Yuyi.run "unzip -o #{file} -d #{@tmp_dir}", :boolean => true
+      end
+
+      if unzip == false
+        Yuyi.say "The `#{@name}` source is an unrecognized archive format.", :type => :fail
+        Yuyi.say "Make sure it is a format supported by tar (tar, pax, cpio, zip, jar, ar, or ISO 9660 image)", :type => :warn
+        exit
+      end
+    end
+
+    # Get source rolls from tmp directory
+    #
+    def get_available_rolls
+      Dir.glob(File.join(@tmp_dir, ROLL_FILE_GLOB)).map do |r|
+        name = File.basename(r, '.rb').to_sym
+        tmp_path = Pathname.new @@root_tmp_dir
+        full_path = Pathname.new r
+        require_path = full_path.relative_path_from(tmp_path).to_s.chomp('.rb')
+
+        @available_rolls[name] = require_path
+      end
+    end
+
+  # Run when class is loaded
+  create_main_tmp_dir
+end

data/lib/yuyi.rb

@@ -0,0 +1,15 @@
+class Yuyi; end
+
+require 'yuyi/cli'
+require 'yuyi/core'
+require 'yuyi/menu'
+require 'yuyi/roll'
+require 'yuyi/source'
+
+class Yuyi
+  extend Yuyi::Cli
+
+  NAME = 'Yuyi'
+  VERSION = '0.1.1'
+  DEFAULT_MENU = File.expand_path('~/yuyi_menu')
+end

data/spec/fixtures/menu.yaml

@@ -0,0 +1,6 @@
+sources:
+- local: spec/fixtures/roll_dir
+- yuyi: spec/fixtures/roll_zip.zip
+rolls:
+  foo_roll:
+    foo: bar

data/spec/fixtures/menu2.yaml

@@ -0,0 +1,6 @@
+sources:
+- local: spec/fixtures/roll_dir
+- yuyi: spec/fixtures/roll_zip.zip
+rolls:
+  foo_roll:
+    bar: foo

data/spec/fixtures/roll_dir/nested/bar_roll.rb

@@ -0,0 +1,2 @@
+class FooRoll < Yuyi::Roll
+end

data/spec/fixtures/roll_dir/nested/foo_roll.rb

@@ -0,0 +1,2 @@
+class FooRoll < Yuyi::Roll
+end