Pigeon-Ruby/bin/pigeon-cli

#!/usr/bin/env ruby

require "pigeon"
require "thor"

def db
  if File.file?(Pigeon::PIGEON_DB_PATH)
    $db ||= Pigeon::Database.new
  else
    STDERR.puts("You must first run `pigeon-cli identity new`.")
    exit 1
  end
end

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 <<-END
          Pigeon has detected a `pigeon.db` file.
          Refusing to overwrite existing Pigeon config.
          Remove config dir or switch to a different directory."
        END
        raise ConfigAlreadyExists
      end
      $db = Pigeon::Database.new
      puts db.local_identity.multihash
    end

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

    def show
      puts db.local_identity.multihash
    end
  end

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

    def set(data = "")
      blob = (data != "") ? data : STDIN.read
      puts db.put_blob(blob)
    end

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

    def get(hex_digest)
      puts db.get_blob(hex_digest)
    end
  end

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

    def add(identity)
      puts db.add_peer(identity)
    end

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

    def remove(identity)
      puts db.remove_peer(identity)
    end

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

    def block(identity)
      puts db.block_peer(identity)
    end

    desc "all", "List all Pigeon peers"

    def all
      puts db.all_peers
    end
  end

  class PigeonDraft < ThorBase
    desc "create", "Begin a new Pigeon message"

    def create(kind)
      puts db.create_draft(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 = db.current_draft
      if draft
        puts draft.put(db, 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 db.current_draft.render_as_draft
      else
        bail("You must create a draft first")
      end
    end

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

    def sign
      puts db.publish_draft(db.current_draft).render
    end
  end

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

    def create(file_path = Pigeon::DEFAULT_BUNDLE_PATH)
      db.create_bundle(file_path)
    end

    desc "ingest", "Ingest a pigeon bundle file"

    def ingest(file_path = Pigeon::DEFAULT_BUNDLE_PATH)
      db.ingest_bundle(file_path)
    end
  end

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

    def find(multihash)
      puts db.read_message(multihash).render
    end

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

    def find_all(author = db.local_identity.multihash)
      puts db.find_all_messages(author).join(Pigeon::CR) + Pigeon::CR
    end

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

    def last
      me = db.local_identity
      mcount = db.get_message_count_for(me.multihash)
      multihash = db.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
      me = db.local_identity.multihash
      mcount = db.get_message_count_for(me)
      puts "
          -`.   Pigeon Protocol Ruby Client
        '( @ >  Version:   #{Pigeon::VERSION}
         _) (   Peers:     #{db.all_peers.count}
        /    )  Blocked:   #{db.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)