build_and_make.rake

require File.expand_path(File.dirname(__FILE__) + "/../init.rb")
require 'ruby_to_ansi_c'

C_VAR_TYPES = "unsigned|int|long|double|str|char|byte"

# incredibly primitive tests 
# rake test:compile or rake test:upload
# runs through all sketches in the example directory

def run_tests(sketch, type)
  sh %{rake make:#{type} sketch=examples/#{sketch}}
end

namespace :test do
  
  desc "iterate through all the sketches in the example directory"
  task :upload => :gather do 
    @examples.each {|e| run_tests(e, "upload")}
  end
  
  task :compile => :gather do 
    @examples.each {|e| run_tests(e, "compile")}
    end
  end
  
  desc "gather all tests"
  task :gather do # => "make:upload" do
    @examples = []
    @test_results = []
    Dir.entries( File.expand_path("#{RAD_ROOT}/examples/") ).each do |f|
      if (f =~ /\.rb$/)
        @examples << f.split('.').first
      end
  end

end


namespace :make do
  
  desc "compile the sketch and then upload it to your Arduino board"
  task :upload => :compile do
    if Makefile.hardware_params['physical_reset']
      puts "Reset the Arduino and hit enter.\n==If your board doesn't need it, you can turn off this prompt in config/software.yml=="
      STDIN.gets.chomp
    end
    sh %{cd #{RAD_ROOT}/#{@test_dir + @sketch_name}; make upload}
  end
    
  desc "generate a makefile and use it to compile the .cpp"
  task :compile => [:clean_sketch_dir, "build:sketch"] do # should also depend on "build:sketch"
    Makefile.compose_for_sketch( @test_dir + @sketch_name )
    # not allowed? sh %{export PATH=#{Makefile.software_params[:arduino_root]}/tools/avr/bin:$PATH}
    sh %{cd #{RAD_ROOT}/#{@test_dir + @sketch_name}; make depend; make}
  end
  
  desc "generate a makefile and use it to compile the .cpp using the current .cpp file"
  task :compile_cpp => ["build:sketch_dir", "build:gather_required_plugins", "build:plugin_setup", "build:setup", :clean_sketch_dir] do # should also depend on "build:sketch"
    Makefile.compose_for_sketch( @test_dir + @sketch_name )
    # not allowed? sh %{export PATH=#{Makefile.software_params[:arduino_root]}/tools/avr/bin:$PATH}
    sh %{cd #{RAD_ROOT}/#{@test_dir + @sketch_name}; make depend; make}
  end
  
  desc "generate a makefile and use it to compile the .cpp and upload it using current .cpp file"
  task :upload_cpp => ["build:sketch_dir", "build:gather_required_plugins", "build:plugin_setup", "build:setup", :clean_sketch_dir] do # should also depend on "build:sketch"
    Makefile.compose_for_sketch( @test_dir + @sketch_name )
    # not allowed? sh %{export PATH=#{Makefile.software_params[:arduino_root]}/tools/avr/bin:$PATH}
    sh %{cd #{RAD_ROOT}/#{@test_dir + @sketch_name}; make depend; make upload}
  end
  
  task :clean_sketch_dir => ["build:file_list", "build:sketch_dir"] do
    @sketch_name = @sketch_class.split(".").first
    FileList.new(Dir.entries("#{RAD_ROOT}/#{@test_dir + @sketch_name}")).exclude("#{@test_dir + @sketch_name}.cpp").exclude(/^\./).each do |f|
      sh %{rm #{RAD_ROOT}/#{@test_dir + @sketch_name}/#{f}}
    end
  end
  
end

namespace :build do

  desc "actually build the sketch"
  task :sketch => [:file_list, :sketch_dir, :gather_required_plugins, :plugin_setup, :setup] do
    klass = @sketch_class.split(".").first.split("_").collect{|c| c.capitalize}.join("")    
    eval ArduinoSketch.pre_process(File.read(@test_dir + @sketch_class))
    c_methods = []
    sketch_signatures = []
    # until we better understand RubyToC let's see what's happening on errors
    $sketch_methods.each do |meth| 
      raw_rtc_meth = RADProcessor.translate(constantize(klass), meth) 
      puts "Translator Error: #{raw_rtc_meth.inspect}" if raw_rtc_meth[0..8] == "// ERROR:"
      c_methods << raw_rtc_meth unless meth == "setup"
      # treat the setup method differently
      @additional_setup = [] if meth == "setup"
      raw_rtc_meth.each {|m| @additional_setup << ArduinoSketch.add_to_setup(m) } if meth == "setup"
    end
    c_methods.each {|meth| sketch_signatures << "#{meth.scan(/^\w*\s?\*?\n.*\)/)[0].gsub("\n", " ")};" }
    clean_c_methods = []
    # remove external variables that were previously injected
    c_methods.join("\n").each { |meth| clean_c_methods << ArduinoSketch.post_process_ruby_to_c_methods(meth) }
    c_methods_with_timer = clean_c_methods.join.gsub(/loop\(\)\s\{/,"loop() {\ntrack_total_loop_time();")
    # last chance to add/change setup
    @setup[2] << sketch_signatures.join("\n") unless sketch_signatures.empty?
    # add special setup method to existing setup if present
    if @additional_setup
      @setup[2] << "void additional_setup();" # declaration
      @setup[4] << "additional_setup();" # call from setup
      @setup[5] << @additional_setup.join("") # 
    end
    result = "#{@setup.join( "\n" )}\n#{c_methods_with_timer}\n"
    name = @sketch_class.split(".").first
    File.open("#{@test_dir}#{name}/#{name}.cpp", "w"){|f| f << result}
  end

  # needs to write the library include and the method signatures
  desc "build setup function"
  task :setup do
    klass = @sketch_class.split(".").first.split("_").collect{|c| c.capitalize}.join("")
    eval "class #{klass} < ArduinoSketch; end;"
    
    @@as = ArduinoSketch.new
    
    delegate_methods = @@as.methods - Object.new.methods
    delegate_methods.reject!{|m| m == "compose_setup"}
        
    delegate_methods.each do |meth|
       constantize(klass).module_eval <<-CODE
       def self.#{meth}(*args)
       @@as.#{meth}(*args)
       end
       CODE
    end
    # allow variable declaration without quotes: @foo = int 
    ["long","unsigned","int","byte","short"].each do |type|
      constantize(klass).module_eval <<-CODE
       def self.#{type}
        return "#{type}"
       end
       CODE
    end   
    eval File.read(@test_dir + @sketch_class)
    @@as.process_external_vars(constantize(klass))
    @setup = @@as.compose_setup
  end
  
  desc "add plugin methods"
  task :plugin_setup do
    $plugins_to_load.each do |name|
       klass = name.split(".").first.split("_").collect{|c| c.capitalize}.join("")
       eval "class #{klass} < ArduinoPlugin; end;"
    
       @@ps = ArduinoPlugin.new
       plugin_delegate_methods = @@ps.methods - Object.new.methods
       plugin_delegate_methods.reject!{|m| m == "compose_setup"}
    
       plugin_delegate_methods.each do |meth|
         constantize(klass).module_eval <<-CODE
         def self.#{meth}(*args)
           @@ps.#{meth}(*args)
         end
         CODE
       end

      eval ArduinoPlugin.process(File.read("vendor/plugins/#{name}"))
      
    end
    @@no_plugins = ArduinoPlugin.new if @plugin_names.empty?
  end
  
  desc "determine which plugins to load based on use of methods in sketch"
  task :gather_required_plugins do
    @plugin_names.each do |name|
       ArduinoPlugin.check_for_plugin_use(File.read(@test_dir + @sketch_class), File.read("vendor/plugins/#{name}"), name )
    end
    puts "#{$plugins_to_load.length} of #{$plugin_methods_hash.length} plugins are being loaded:  #{$plugins_to_load.join(", ")}"
  end
  
  desc "setup target directory named after your sketch class"
  task :sketch_dir => [:file_list] do
    mkdir_p "#{@test_dir + @sketch_class.split(".").first}"
  end

  task :file_list do
    # take another look at this, since if the root directory name is changed, everything breaks
    # perhaps we generate a constant when the project is generated an pop it here or in the init file
    @sketch_directory = File.expand_path("#{File.dirname(__FILE__)}/../../../").split("/").last
    # multiple sketches are possible with rake make:upload sketch=new_sketch
    @test_dir = ""
    if ENV['sketch'] =~ /^examples\//
      # strip the example and set a directory variable
      ENV['sketch'] = ENV['sketch'].gsub(/^examples\//, "")
      @test_dir = "examples/"
    end
    @sketch_class = ENV['sketch'] ? "#{ENV['sketch']}.rb" : "#{@sketch_directory}.rb"
    $sketch_file_location = @test_dir + @sketch_class
    @file_names = []
    @plugin_names = []
    Dir.entries( File.expand_path(RAD_ROOT) ).each do |f|
      if (f =~ /\.rb$/)
        @file_names << f
      end
    end
    Dir.entries( File.expand_path("#{RAD_ROOT}/vendor/plugins/") ).each do |f|
      if (f =~ /\.rb$/)
        @plugin_names << f
      end
    end
  end
end

#yoinked from Rails
def constantize(camel_cased_word)
  unless /\A(?:::)?([A-Z]\w*(?:::[A-Z]\w*)*)\z/ =~ camel_cased_word
    raise NameError, "#{camel_cased_word.inspect} is not a valid constant name!"
  end

  Object.module_eval("::#{$1}", __FILE__, __LINE__)
end