require 'thor'
require 'thor/group'

module Artoo
  module Commands
    class Connect < Commands
      package_name "connect"

      desc "scan", "Scan for connected devices"
      option :type, :aliases => "-t", :default => "bluetooth", :desc => "type of scan [bluetooth, serial]"
      def scan
        case os
        when :linux
          case options[:type]
          when 'bluetooth'
            run("hcitool scan")
          when 'serial'
            run("ls /dev/tty*")
          when 'usb'
            run("lsusb")            
          else
            say "ERROR: scan type '#{options[:type]}' not supported!"
          end
        when :macosx
          run("ls /dev/tty*")          
        else
          say "OS not yet supported..."
        end
      end

      desc "bind [ADDRESS] [NAME]", "Binds a Bluetooth device to some connected hardware"
      option :comm, :aliases => "-c", :default => 0, :desc => "Comm number"
      option :radio, :aliases => "-r", :default => "hci0", :desc => "Bluetooth radio address"
      def bind(address, name)
        case os
        when :linux
          run("sudo rfcomm -i #{options[:radio]} bind /dev/rfcomm#{options[:comm]} #{address} 1")
          run("sudo ln -s /dev/rfcomm#{options[:comm]} /dev/#{name}")
        when :macosx
          say "OSX binds devices on its own volition."
        else
          say "OS not yet supported..."
        end
      end

      desc "serial [NAME] [PORT]", "Connect a serial device to a TCP socket using socat"
      option :retries, :aliases => "-r", :default => 0, :desc => "Number of times to retry connecting on failure"
      option :baudrate, :aliases => "-b", :default => 57600, :desc => "Baud rate to use to connect to the serial device"
      def serial(name, port)
        attempts = 1 + options[:retries].to_i
        
        case os
        when :linux
          run("sudo chmod a+rw /dev/#{name}")
          
          while(attempts > 0) do
            run("socat -d -d FILE:/dev/#{name},nonblock,raw,b#{options[:baudrate]},echo=0 TCP-LISTEN:#{port},fork")
            break unless $? == 1
            attempts -= 1
          end

        when :macosx
          while(attempts > 0) do
            run("socat -d -d -b#{options[:baudrate]} FILE:/dev/#{name},nonblock,raw,echo=0 TCP-LISTEN:#{port},fork")
            break unless $? == 1
            attempts -= 1
          end

        else
          say "OS not yet supported..."
        end
      end
    end
  end
end