Pigeon-Ruby/pigeon-cli

#!/usr/bin/env ruby

require_relative File.join("dist", "pigeon")

require "thor"

module Pigeon
  class Identity < Thor
    class ConfigAlreadyExists < StandardError; end

    desc "new", "Creates a new identiy in `.pigeon` directory if none exists"

    def new
      # TODO: --force flag
      # TODO: --seed flag
      if Dir.exist?(Pigeon::Storage::ROOT_DIR)
        puts "Pigeon has detected a `.pigeon` directory.
          Refusing to overwrite existing Pigeon config.
          Remove config dir or switch to a different directory."
        raise ConfigAlreadyExists
      end
      kp = Pigeon::KeyPair.new()
      kp.save!
      puts kp.public_key
    end

    desc "show", "Prints a base64 identiy string to STDOUT"

    def show
      puts Pigeon::KeyPair.current.public_key
    end
  end

  class Blob < Thor
    desc "set", "Copy arbitrary binary data into the database"

    def set(data = "")
      blob = (data != "") ? data : STDIN.read
      puts Pigeon::Storage.current.set_blob(blob)
    end

    desc "get", "Read arbitrary data from the database"

    def get(hex_digest)
      puts Pigeon::Storage.current.get_blob(hex_digest)
    end
  end

  class Peer < Thor
    desc "add", "Begin following a Pigeon peer"

    def add(identity)
      puts Pigeon::Storage.current.add_peer(identity)
    end

    desc "remove", "Stop following a Pigeon peer"

    def remove(identity)
      puts Pigeon::Storage.current.remove_peer(identity)
    end

    desc "block", "Stop following a Pigeon peer AND refuse to replicate"

    def block(identity)
      puts Pigeon::Storage.current.block_peer(identity)
    end

    desc "all", "List all Pigeon peers"

    def all
      puts Pigeon::Storage.current.all_peers
    end
  end

  class PigeonMessage < Thor
    desc "create", "Begin a new Pigeon message"

    def create(kind)
      puts Pigeon::Message.create(kind: kind).serialize
    end
  end

  class CLI < Thor
    desc "status", "Show various information about the `.pigeon` directory"

    def status
      puts "
      ,.------------------------------------
      |    -`.   Pigeon Protocol Ruby Client
      |  '( @ >    Version: #{Config::VERSION}
      |   _) (     Peers:   #{Dir[File.join(".pigeon", "peers", "*")].count}
      |  /    )    Blocked: #{Dir[File.join(".pigeon", "blocked", "*")].count}
      | /_,'  /
      |   \  /
      |===m" "m===
      '-,___________________________________
      "
    end

    desc "identity SUBCOMMAND ...ARGS", "Manage `.pigeon` identity"
    subcommand "identity", Identity

    desc "blob SUBCOMMAND ...ARGS", "Manage blob storage"
    subcommand "blob", Blob

    desc "peer SUBCOMMAND ...ARGS", "Manage blob storage"
    subcommand "peer", Peer

    desc "message SUBCOMMAND ...ARGS", "Manage messages"
    subcommand "message", PigeonMessage
  end
end

Pigeon::CLI.start(ARGV)