Shell.get_device

Author: hasumikin

mrbgems/picoruby-ble/mrblib/ble.rb

@@ -110,7 +110,7 @@ def heartbeat_callback
   end
 
   def blink_led
-    @led ||= CYW43::GPIO.new(CYW43::GPIO::LED_PIN)
+    @led ||= Shell.get_device(:gpio, 'LED_BLE')
     @led.write(@led.low? ? 1 : 0)
   end
 

mrbgems/picoruby-shell/mrbgem.rake

@@ -14,6 +14,7 @@ MRuby::Gem::Specification.new('picoruby-shell') do |spec|
   spec.add_dependency 'picoruby-env'
   spec.add_dependency 'picoruby-crc'
   spec.add_dependency 'picoruby-machine'
+  spec.add_dependency 'picoruby-yaml'
   if build.posix?
     if build.vm_mrubyc?
       spec.add_dependency('picoruby-dir')

mrbgems/picoruby-shell/mrblib/shell.rb

@@ -1,9 +1,11 @@
 require "env"
+require 'gpio'
 require "metaprog"
 require "picorubyvm"
 require "sandbox"
 require "crc"
 require "machine"
+require 'yaml'
 begin
   require "filesystem-fat"
   require "vfs"
@@ -17,6 +19,25 @@
 
 
 class Shell
+
+  DeviceInstances = {}
+
+  def self.get_device(type, name)
+    key = "#{type}_#{name}".upcase
+    DeviceInstances[key] ||= case key
+    when 'GPIO_TRIGGER_NMBLE'
+      GPIO.new((ENV[key] || 22).to_i, GPIO::IN|GPIO::PULL_UP)
+    when 'GPIO_LED_BLE', 'GPIO_LED_WIFI'
+      if ENV[key].nil? || ENV[key] == 'cyw43_led'
+        CYW43::GPIO.new(CYW43::GPIO::LED_PIN)
+      else
+        GPIO.new(ENV[key].to_i, GPIO::OUT)
+      end
+    else
+      raise "Unknown GPIO key: #{key}"
+    end
+  end
+
   def self.setup_root_volume(device, label: "PicoRuby")
     sleep 1 if device == :sd
     return if VFS.volume_index("/")
@@ -89,6 +110,38 @@ def self.setup_system_files(root = nil, force: false)
       self.ensure_system_file(path, Machine.unique_id, nil)
     end
     Dir.chdir ENV['HOME']
+
+    config_file = "/etc/config.yml"
+    # example of `config.yml`:
+    #
+    # device:
+    #   gpio:
+    #     trigger_nmble: 22
+    #     led_ble: cyw43_led
+    #     led_wifi: 23
+    begin
+      config = YAML.load_file(config_file)
+      # @type var config: Hash[String, untyped]
+      device = config['device']
+      if device&.respond_to?(:each)
+        device.each do |type, values|
+          values&.each do |key, value|
+            ENV["#{type}_#{key}".upcase] = value.to_s
+          end
+        end
+      end
+    rescue => e
+      puts "Failed to load config file: #{config_file}"
+      puts "  #{e.message} (#{e.class})"
+    end
+
+    begin
+      require "cyw43"
+      if CYW43.respond_to?(:enable_sta_mode)
+        ENV['WIFI_MODULE'] = "cwy43"
+      end
+    rescue
+    end
   end
 
   def self.bootstrap(file)
@@ -178,6 +231,7 @@ def initialize(clean: false)
       '|_|   |_|\___\___/|_| \_\\___,_|_.__/ \__, |',
       "               #{AUTHOR_COLOR}by hasumikin#{LOGO_COLOR}          |___/"
     ]
+    SHORT_LOGO_LINES = ["PicoRuby", "   by", "hasumikin"]
   elsif RUBY_ENGINE == "mruby"
     LOGO_LINES = [
       ' __  __ _                ____        _',
@@ -187,8 +241,8 @@ def initialize(clean: false)
       '|_|  |_|_|\___|_|  \___/|_| \_\\\\__,_|_.__/ \__, |',
       "                  #{AUTHOR_COLOR}by hasumikin#{LOGO_COLOR}             |___/"
     ]
+    SHORT_LOGO_LINES = ["MicroRuby", "   by", "hasumikin"]
   end
-  SHORT_LOGO_LINES = ["PicoRuby", "   by", "hasumikin"]
 
   def show_logo
     return nil if ENV['TERM'] == "dumb"

mrbgems/picoruby-shell/shell_executables/r2p2.rb

@@ -1,16 +1,8 @@
-begin
-  require "cyw43"
-  if CYW43.respond_to?(:enable_sta_mode)
-    ENV['WIFI_MODULE'] = "cwy43"
-    require 'gpio'
-    pin = GPIO.new(22, GPIO::IN|GPIO::PULL_UP)
-    if pin.low?
-      system "nmble"
-    end
-    system "wifi_connect --check-auto-connect"
+if ENV['WIFI_MODULE'] == "cwy43"
+  if Shell.get_device(:gpio, 'TRIGGER_NMBLE').low?
+    system "nmble"
   end
-rescue => e
-  puts "No WiFi module. Ignore: #{e.message}"
+  system "wifi_connect --check-auto-connect"
 end
 
 if File.exist?("#{ENV['HOME']}/app.mrb")

mrbgems/picoruby-shell/sig/shell.rbs

@@ -9,6 +9,17 @@ $LOAD_PATH: Array[String]
 ARGV: Array[String]
 
 class Shell
+  type rtc_t = PCF8523
+
+  type device_t = GPIO | ADC | I2C | SPI | UART | PWM
+  DeviceInstances: Hash[String, device_t]
+  def self.get_device: (:gpio, String | Symbol name) -> GPIO
+                     | (:adc,  String | Symbol name) -> ADC
+                     | (:i2c,  String | Symbol name) -> I2C
+                     | (:spi,  String | Symbol name) -> SPI
+                     | (:uart, String | Symbol name) -> UART
+                     | (:pwm,  String | Symbol name) -> PWM
+
   LOGO_LINES: [String]
   SHORT_LOGO_LINES: [String]
   LOGO_COLOR: String