Use miniterm instead of screen
Andre Richter
4 years ago
parent
098e19ecc4
commit
e1473099ff
@ -0,0 +1,138 @@
|
|||||||
|
#!/usr/bin/env ruby
|
||||||
|
# frozen_string_literal: true
|
||||||
|
|
||||||
|
# SPDX-License-Identifier: MIT OR Apache-2.0
|
||||||
|
#
|
||||||
|
# Copyright (c) 2020 Andre Richter <andre.o.richter@gmail.com>
|
||||||
|
|
||||||
|
require 'rubygems'
|
||||||
|
require 'bundler/setup'
|
||||||
|
require 'io/console'
|
||||||
|
require 'colorize'
|
||||||
|
require 'serialport'
|
||||||
|
|
||||||
|
class ConnectionError < StandardError; end
|
||||||
|
|
||||||
|
# The main class
|
||||||
|
class MiniTerm
|
||||||
|
def initialize(serial_name)
|
||||||
|
@name_short = 'MT'
|
||||||
|
@target_serial_name = serial_name
|
||||||
|
@target_serial = nil
|
||||||
|
@host_console = IO.console
|
||||||
|
end
|
||||||
|
|
||||||
|
private
|
||||||
|
|
||||||
|
def serial_connected?
|
||||||
|
File.exist?(@target_serial_name)
|
||||||
|
end
|
||||||
|
|
||||||
|
def wait_for_serial
|
||||||
|
loop do
|
||||||
|
break if serial_connected?
|
||||||
|
|
||||||
|
print "\r[#{@name_short}] ⏳ Waiting for #{@target_serial_name}"
|
||||||
|
sleep(1)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def open_serial
|
||||||
|
wait_for_serial
|
||||||
|
|
||||||
|
@target_serial = SerialPort.new(@target_serial_name, 230_400, 8, 1, SerialPort::NONE)
|
||||||
|
|
||||||
|
# Ensure all output is immediately flushed to the device.
|
||||||
|
@target_serial.sync = true
|
||||||
|
rescue Errno::EACCES => e
|
||||||
|
puts
|
||||||
|
puts "[#{@name_short}] 🚫 #{e.message} - Maybe try with 'sudo'"
|
||||||
|
exit
|
||||||
|
else
|
||||||
|
puts
|
||||||
|
puts "[#{@name_short}] ✅ Connected"
|
||||||
|
end
|
||||||
|
|
||||||
|
def terminal
|
||||||
|
@host_console.raw!
|
||||||
|
|
||||||
|
Thread.abort_on_exception = true
|
||||||
|
Thread.report_on_exception = false
|
||||||
|
|
||||||
|
# Receive from target and print on host console.
|
||||||
|
target_to_host = Thread.new do
|
||||||
|
loop do
|
||||||
|
char = @target_serial.getc
|
||||||
|
|
||||||
|
raise ConnectionError if char.nil?
|
||||||
|
|
||||||
|
# Translate incoming newline to newline + carriage return.
|
||||||
|
@host_console.putc("\r") if char == "\n"
|
||||||
|
@host_console.putc(char)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
# Transmit host console input to target.
|
||||||
|
loop do
|
||||||
|
c = @host_console.getc
|
||||||
|
|
||||||
|
# CTRL + C in raw mode was pressed.
|
||||||
|
if c == "\u{3}"
|
||||||
|
target_to_host.kill
|
||||||
|
break
|
||||||
|
end
|
||||||
|
|
||||||
|
@target_serial.putc(c)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def connetion_reset
|
||||||
|
@target_serial&.close
|
||||||
|
@target_serial = nil
|
||||||
|
@host_console.cooked!
|
||||||
|
end
|
||||||
|
|
||||||
|
# When the serial lost power or was removed during R/W operation.
|
||||||
|
def handle_reconnect
|
||||||
|
connetion_reset
|
||||||
|
|
||||||
|
puts
|
||||||
|
puts "[#{@name_short}] ⚡ #{'Connection Error: Reinsert the USB serial again'.light_red}"
|
||||||
|
end
|
||||||
|
|
||||||
|
def handle_unexpected(error)
|
||||||
|
connetion_reset
|
||||||
|
|
||||||
|
puts
|
||||||
|
puts "[#{@name_short}] ⚡ #{"Unexpected Error: #{error.inspect}".light_red}"
|
||||||
|
end
|
||||||
|
|
||||||
|
public
|
||||||
|
|
||||||
|
def run
|
||||||
|
open_serial
|
||||||
|
terminal
|
||||||
|
rescue ConnectionError, EOFError, Errno::EIO
|
||||||
|
handle_reconnect
|
||||||
|
retry
|
||||||
|
rescue StandardError => e
|
||||||
|
handle_unexpected(e)
|
||||||
|
ensure
|
||||||
|
connetion_reset
|
||||||
|
puts
|
||||||
|
puts "[#{@name_short}] Bye 👋"
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
if __FILE__ == $PROGRAM_NAME
|
||||||
|
puts 'Miniterm 1.0'.cyan
|
||||||
|
puts
|
||||||
|
|
||||||
|
# CTRL + C handler. Only here to suppress Ruby's default exception print.
|
||||||
|
trap('INT') do
|
||||||
|
# The `ensure` block from `MiniTerm::run` will run after exit, restoring console state.
|
||||||
|
exit
|
||||||
|
end
|
||||||
|
|
||||||
|
MiniTerm.new(ARGV[0]).run
|
||||||
|
end
|
||||||
Loading…
Reference in New Issue