logging in to mastodon

.gitignore

···
+config/*.yml

Gemfile

···
+source 'https://rubygems.org'
+
+gem 'mastodon-api', git: 'https://github.com/mastodon/mastodon-api.git'
+
+gem 'io-console', '~> 0.5'
+gem 'json', '~> 2.5'
+gem 'net-http', '~> 0.2'
+gem 'uri', '~> 0.13'
+gem 'yaml', '~> 0.1'

Gemfile.lock

···
+GIT
+  remote: https://github.com/mastodon/mastodon-api.git
+  revision: 60b0ed09c3b979cbeb833db0a320b30e1b022a1e
+  specs:
+    mastodon-api (2.0.0)
+      addressable (~> 2.6)
+      buftok (~> 0)
+      http (~> 4.0)
+      oj (~> 3.7)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    addressable (2.8.6)
+      public_suffix (>= 2.0.2, < 6.0)
+    bigdecimal (3.1.7)
+    buftok (0.3.0)
+    domain_name (0.6.20240107)
+    ffi (1.16.3)
+    ffi-compiler (1.3.2)
+      ffi (>= 1.15.5)
+      rake
+    http (4.4.1)
+      addressable (~> 2.3)
+      http-cookie (~> 1.0)
+      http-form_data (~> 2.2)
+      http-parser (~> 1.2.0)
+    http-cookie (1.0.5)
+      domain_name (~> 0.5)
+    http-form_data (2.3.0)
+    http-parser (1.2.3)
+      ffi-compiler (>= 1.0, < 2.0)
+    io-console (0.7.2)
+    json (2.7.1)
+    net-http (0.4.1)
+      uri
+    oj (3.16.3)
+      bigdecimal (>= 3.0)
+    public_suffix (5.0.4)
+    rake (13.1.0)
+    uri (0.13.0)
+    yaml (0.3.0)
+
+PLATFORMS
+  arm64-darwin-21
+  ruby
+
+DEPENDENCIES
+  io-console (~> 0.5)
+  json (~> 2.5)
+  mastodon-api!
+  net-http (~> 0.2)
+  uri (~> 0.13)
+  yaml (~> 0.1)
+
+BUNDLED WITH
+   2.5.6

app/mastodon_account.rb

···
+require 'mastodon'
+require 'yaml'
+
+require_relative 'mastodon_api'
+
+class MastodonAccount
+  APP_NAME = "tootify"
+  CONFIG_FILE = File.expand_path(File.join(__dir__, '..', 'config', 'mastodon.yml'))
+  OAUTH_SCOPES = 'read:accounts read:statuses write:media write:statuses'
+
+  def initialize
+    @config = File.exist?(CONFIG_FILE) ? YAML.load(File.read(CONFIG_FILE)) : {}
+  end
+
+  def save_config
+    File.write(CONFIG_FILE, YAML.dump(@config))
+  end
+
+  def oauth_login(handle, email, password)
+    instance = handle.split('@').last
+    app_response = register_oauth_app(instance, OAUTH_SCOPES)
+
+    api = MastodonAPI.new(instance)
+
+    json = api.oauth_login_with_password(
+      app_response.client_id,
+      app_response.client_secret,
+      email, password, OAUTH_SCOPES
+    )
+
+    api.access_token = json['access_token']
+    info = api.account_info
+
+    @config['handle'] = handle
+    @config['access_token'] = api.access_token
+    @config['user_id'] = info['id']
+    save_config
+  end
+
+  def register_oauth_app(instance, scopes)
+    client = Mastodon::REST::Client.new(base_url: "https://#{instance}")
+    client.create_app(APP_NAME, 'urn:ietf:wg:oauth:2.0:oob', scopes)
+  end
+end

app/mastodon_api.rb

···
+require 'json'
+require 'net/http'
+require 'uri'
+
+class MastodonAPI
+  class UnauthenticatedError < StandardError
+  end
+
+  class UnexpectedResponseError < StandardError
+  end
+
+  class APIError < StandardError
+    attr_reader :response
+
+    def initialize(response)
+      @response = response
+      super("APIError #{response.code}: #{response.body}")
+    end
+
+    def status
+      response.code.to_i
+    end
+  end
+
+  attr_accessor :access_token
+
+  def initialize(host, access_token = nil)
+    @host = host
+    @root = "https://#{@host}/api/v1"
+    @access_token = access_token
+  end
+
+  def oauth_login_with_password(client_id, client_secret, email, password, scopes)
+    url = URI("https://#{@host}/oauth/token")
+
+    params = {
+      client_id: client_id,
+      client_secret: client_secret,
+      grant_type: 'password',
+      scope: scopes,
+      username: email,
+      password: password
+    }
+
+    response = Net::HTTP.post_form(url, params)
+    status = response.code.to_i
+
+    if status / 100 == 2
+      JSON.parse(response.body)
+    else
+      raise APIError.new(response)
+    end
+  end
+
+  def account_info
+    raise UnauthenticatedError.new unless @access_token
+    get_json("/accounts/verify_credentials")
+  end
+
+  def lookup_account(username)
+    json = get_json("/accounts/lookup", { acct: username })
+    raise UnexpectedResponseError.new unless json.is_a?(Hash) && json['id'].is_a?(String)
+    json
+  end
+
+  def account_statuses(user_id, params = {})
+    get_json("/accounts/#{user_id}/statuses", params)
+  end
+
+  def get_json(path, params = {})
+    url = URI(path.start_with?('https://') ? path : @root + path)
+    url.query = URI.encode_www_form(params) if params
+
+    headers = {}
+    headers['Authorization'] = "Bearer #{@access_token}" if @access_token
+
+    response = Net::HTTP.get_response(url, headers)
+    status = response.code.to_i
+
+    if status / 100 == 2
+      JSON.parse(response.body)
+    elsif status / 100 == 3
+      get_json(response['Location'])
+    else
+      raise APIError.new(response)
+    end
+  end
+end

app/tootify.rb

···
+require 'io/console'
+require_relative 'mastodon_account'
+
+class Tootify
+  def initialize
+    @mastodon = MastodonAccount.new
+  end
+
+  def login_bluesky
+  end
+
+  def login_mastodon(handle)
+    print "Email: "
+    email = STDIN.gets.chomp
+
+    print "Password: "
+    password = STDIN.noecho(&:gets).chomp
+    puts
+
+    @mastodon.oauth_login(handle, email, password)
+  end
+end

tootify

···
+#!/usr/bin/env ruby
+
+require 'bundler/setup'
+require_relative 'app/tootify'
+
+$app = Tootify.new
+
+def run(argv)
+  options, args = argv.partition { |x| x.start_with?('-') }
+
+  case args.first
+  when 'login'
+    login(args[1])
+  when 'check'
+  else
+    print_help
+  end
+end
+
+def print_help
+  puts "Usage: #{$PROGRAM_NAME} login mastodon@account | login @bluesky"
+  puts "       #{$PROGRAM_NAME} check"
+  exit 1
+end
+
+def login(name)
+  if name =~ /\A[^@]+@[^@]+\z/
+    $app.login_mastodon(name)
+  elsif name =~ /\A@[^@]+\z/
+    $app.login_bluesky(name)
+  elsif name.nil?
+    print_help
+  else
+    puts "Invalid handle: #{name.inspect}"
+    exit 1
+  end
+end
+
+run(ARGV)