Pigeon-Ruby/pigeon-cli

#!/usr/bin/env ruby

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

require "thor"

def bail(msg)
  $stderr.puts msg
  exit 1
end

module Pigeon
  class ThorBase < Thor
    no_commands do
      def exit_on_failure?
        true
      end
    end
  end

  class Identity < ThorBase
    class ConfigAlreadyExists < StandardError; end

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

    def new
      if File.file?(Pigeon::PIGEON_DB_PATH)
        puts "Pigeon has detected a `pigeon.db` file.
          Refusing to overwrite existing Pigeon config.
          Remove config dir or switch to a different directory."
        raise ConfigAlreadyExists
      end
      kp = Pigeon::LocalIdentity.new()
      kp.save!
      puts kp.multihash
    end

    desc "show", "Prints your identiy string to STDOUT"

    def show
      puts Pigeon::LocalIdentity.current.multihash
    end
  end

  class Blob < ThorBase
    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 < ThorBase
    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 PigeonDraft < ThorBase
    desc "create", "Begin a new Pigeon message"

    def create(kind)
      puts Pigeon::Draft.create(kind: kind).render_as_draft
    end

    desc "append", "Add a key/value pair to the current DRAFT"

    def append(key, raw_value = "")
      v = (raw_value != "") ? raw_value : STDIN.read
      draft = Pigeon::Draft.current
      if draft
        puts draft[key] = v
      else
        bail("You must create a draft first")
      end
    end

    desc "show", "Print a message to STDOUT. If message_id is missing, current draft will be displayed."

    def show(message_id = "")
      if message_id == ""
        puts Pigeon::Draft.current.render_as_draft
      else
        bail("You must create a draft first")
      end
    end

    desc "sign", "Commit current DRAFT to local feed."

    def sign
      puts Pigeon::Draft.current.publish.render
    end
  end

  class PigeonBundle < ThorBase
    desc "create", "Create a pigeon bundle file"

    def create(file_path = Pigeon::DEFAULT_BUNDLE_PATH)
      Pigeon::Bundle.create(file_path)
    end
  end

  class PigeonMessage < ThorBase
    desc "find", "Find a pigeon message in the local DB"

    def find(multihash)
      puts Pigeon::Storage.current.read_message(multihash).render
    end

    desc "find-all", "Find a pigeon message in the local DB"

    def find_all(author = Pigeon::LocalIdentity.current.multihash)
      # TODO: Ability to find-all messages by author ID
      puts Pigeon::Storage
             .current
             .find_all(author)
             .join(Pigeon::CR) + Pigeon::CR
    end

    desc "last", "Grab your last message. INTERNAL USE ONLY"

    def last
      me = Pigeon::LocalIdentity.current
      store = Pigeon::Storage.current
      mcount = store.get_message_count_for(me.multihash)
      multihash = store.get_message_by_depth(me.multihash, mcount - 1)
      puts multihash
    end
  end

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

    def status
      store = Pigeon::Storage.current
      me = Pigeon::LocalIdentity.current.multihash
      mcount = store.get_message_count_for(me)
      puts "
          -`.   Pigeon Protocol Ruby Client
        '( @ >  Version:   #{Pigeon::VERSION}
         _) (   Peers:     #{store.all_peers.count}
        /    )  Blocked:   #{store.all_blocks.count}
       /_,'  /  Published: #{mcount}
         \\  /   Ident:     #{me}
      ===m" "m===
      "
    end

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

    desc "bundle SUBCOMMAND ...ARGS", "Consume and create bundle files"
    subcommand "bundle", PigeonBundle

    desc "draft SUBCOMMAND ...ARGS", "Manage drafts"
    subcommand "draft", PigeonDraft

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

    desc "message SUBCOMMAND ...ARGS", "Manage blob storage"
    subcommand "message", PigeonMessage

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

Pigeon::CLI.start(ARGV)