shutwork 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 27c8118a2da9dde8199128291e41636ded1192012ec449bc723afc9df5101de7
-  data.tar.gz: 9dce3362c8013fdd214ec2957cda9fdb5407c02bf7ff5cd77978478df03b2626
+  metadata.gz: 3ff7c92d1e6570df34fa40f2fbcb6e7296a98439ecc067b1bb0435a93998ea01
+  data.tar.gz: 4280596b8d939771b5d4d7b85f5dbff32a38d1a780e2d78b6ef819d766327e00
 SHA512:
-  metadata.gz: 65d2c5f4622333fe47c50fdf1296c261e7298784fca38166aaa632dbe94427ea851b131da610bbc7f24c2cccd92370f62dc4cf65b3f1d7c9d18a24613f952a50
-  data.tar.gz: 23e38ebca5b76c5472928c001e1b08e84763fe20b2ae50c6b522520eb4ab3646c14ae6e3ae2f35f88c00851e74ecf2d0d63e7434d5f2e0cee0c90d9b3fddc0ea
+  metadata.gz: eaa3804849dd7e30754697ba72a46c9465f1a7fc673c05dee40a3420937aba805a0c52586d56a73b5d58d17482c903eebf126c8a006db39d9631ea7387b9cb7d
+  data.tar.gz: 780e63d70d1b8d359260b8b2da437b1b8c15db4dc3fc296172e042768648800edb0ad31ecde2e315ac855762b58aa598df1487fb95ee17a9934eb98a5de79e21

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    shutwork (0.1.0)
+    shutwork (0.2.0)
       faraday (>= 1.0.0, < 2.0)
       thor (>= 1.0.0, < 2.0)
 

data/README.md

@@ -1,6 +1,7 @@
 # Shutwork
 
 [![Build Status](https://travis-ci.com/kayhide/shutwork.svg?branch=master)](https://travis-ci.com/kayhide/shutwork)
+[![Gem Version](https://badge.fury.io/rb/shutwork.svg)](https://badge.fury.io/rb/shutwork)
 
 ## Installation
 
@@ -10,7 +11,69 @@ Install `shutwork` as:
 
 ## Usage
 
-TBD
+### About me
+
+To get me:
+
+    $ shutwork me
+
+For the first time, it will complain about the absense of token and tell you what to do.
+Follow the displayed instruction and supply your token.
+
+When token is prepared, `me` command will show your account id and your name.
+
+    $ shutwork me
+        123456  Alice
+
+### List rooms and messages
+
+To get rooms:
+
+    $ shutwork rooms
+      12000000  A room
+      29000000  Bob
+      47000000  Another room
+     188000000  Yet anoter room
+    ....
+
+To get messages of a room:
+
+    $ shutwork rooms 12000000
+    ----------------
+    2019-10-23 01:19:50 +0900  Alice
+    Hey, it is working!
+    ----------------
+    2019-10-23 12:49:16 +0900  Bob
+    Wow, unbelievable!
+    ----------------
+    ....
+
+### List members and files
+
+To list memebers at a room:
+
+    $ shutwork rooms 12000000 --members
+        123456  Alice
+      29000000  Bob
+    ...
+
+To list files at a room:
+
+    $ shutwork rooms 12000000 --files
+    2016-08-30 16:12:31 +0900   544KB  help_me.jpg
+    2016-09-02 12:26:26 +0900   229KB  maybe_broken.jpg
+    2016-10-15 19:59:53 +0900   1.0MB  smoking_melting.jpg
+    2016-11-23 21:52:26 +0900   253KB  seriously_damaged.jpg
+    2017-01-28 16:05:37 +0900   204KB  rip.jpg
+    ...
+
+
+### Note
+
+Because of the limitation of Chatwork API, messages cannot be retrieved more than 100 items.
+
+It only collects the most recent 100 messages.
+
 
 
 ## Development

data/lib/shutwork/cli.rb

@@ -31,11 +31,11 @@ module Shutwork
 
       case command
       when "me"
-        Shutwork::Command::Me.new.run
+        Shutwork::Command::Me.new.run ARGV.clone
       when "rooms"
-        Shutwork::Command::Rooms.new.run
+        Shutwork::Command::Rooms.new.run ARGV.clone
       when "token"
-        Shutwork::Command::Token.new.run
+        Shutwork::Command::Token.new.run ARGV.clone
       when nil
         Shutwork::Command::Help.new.run
       else
@@ -56,6 +56,10 @@ module Shutwork
       $stderr.puts "3. Check if it works as:"
       $stderr.puts "  shutwork me"
       $stderr.puts
+
+    rescue Errno::EPIPE
+      exit 0
+
     end
   end
 end

data/lib/shutwork/client.rb

@@ -31,6 +31,18 @@ module Shutwork
       process @conn.get("rooms/#{room_id}/messages?force=1")
     end
 
+    def room_members room_id
+      process @conn.get("rooms/#{room_id}/members")
+    end
+
+    def room_files room_id
+      process @conn.get("rooms/#{room_id}/files")
+    end
+
+    def file room_id, file_id
+      process @conn.get("rooms/#{room_id}/files/#{file_id}?create_download_url=1")
+    end
+
     def process res
       if @verbose
         $stderr.puts res.headers.to_json

data/lib/shutwork/command/base.rb

@@ -1,7 +1,9 @@
 require "forwardable"
-
+require "open-uri"
 require "thor/shell/color"
 
+require "shutwork/command/display"
+
 module Shutwork
   module Command
     class Base
@@ -9,9 +11,23 @@ module Shutwork
 
       def_delegators :@shell, :say, :say_status
 
+      include Display
+
       def initialize
         @shell = Thor::Shell::Color.new
       end
+
+      def download_file url, dst, at = nil
+        FileUtils.mkdir_p dst.dirname
+        open(url) do |i|
+          open(dst, 'wb') do |o|
+            o.write i.read
+          end
+          if at
+            File.utime at, at, dst
+          end
+        end
+      end
     end
   end
 end

data/lib/shutwork/command/display.rb

@@ -0,0 +1,54 @@
+module Shutwork
+  module Command
+    module Display
+      def display_account item
+        puts ("%10s  %s" % [item["account_id"], item["name"]])
+      end
+
+      def display_room item
+        puts ("%10s  %s" % [item["room_id"], item["name"]])
+      end
+
+      def display_message item
+        puts "----------------"
+        puts ("%s  %s" % [
+          format_datetime(item["send_time"]),
+          item.dig("account", "name")
+        ])
+        puts item["body"]
+      end
+
+      def display_file item
+        puts ("%s  %6s  %s" % [
+          format_datetime(item["upload_time"]),
+          format_filesize(item["filesize"]),
+          item["filename"]]
+        )
+      end
+
+      def format_datetime x
+        Time.at(x).to_s
+      end
+
+      def format_filesize x
+        to_filesize_human x
+      end
+
+      def to_filesize_human x
+        x = %w(KB MB GB TB).inject([x.to_f, 'B']) do |acc, u|
+          v = acc.first / 1024
+          if v.round(2) < 0.90
+            acc
+          else
+            [v, u]
+          end
+        end
+        if x.first >= 10.0 || x.last == 'B'
+          "%.0f%s" % x
+        else
+          "%.1f%s" % x
+        end
+      end
+    end
+  end
+end

data/lib/shutwork/command/help.rb

@@ -10,11 +10,11 @@ Usage: shutwork COMMAND
 Available options:
   -h,--help                Show this help text
      --version             Show version
-  -r,--raw                 Show result in raw format
 
 Available commands:
-  me                       Me
-  rooms                    Rooms
+  me                       Show me
+  rooms                    Show Rooms and messages
+  token                    Manage token
 EOS
       end
     end

data/lib/shutwork/command/me.rb

@@ -8,16 +8,16 @@ require "shutwork/token"
 module Shutwork
   module Command
     class Me < Base
-      def parse_args
+      def parse_args args
         opts = OptionParser.new
         opts.program_name = "shutwork #{self.class.name.split(/::/).last.downcase}"
         opts.on("-r", "--raw", "Show results in raw format") { @raw = true }
         opts.on("-v", "--verbose", "Verbose") { @verbose = true }
-        opts.parse ARGV
+        opts.parse args
       end
 
-      def run
-        parse_args
+      def run args  = []
+        parse_args args
         token = Shutwork::Token.read
         @client = Shutwork::Client.new token: token, verbose: @verbose
 
@@ -25,8 +25,7 @@ module Shutwork
         if @raw
           puts item
         else
-          me = JSON.parse(item)
-          puts ("%10s  %s" % [me["account_id"], me["name"]])
+          display_account JSON.parse(item)
         end
       end
     end

data/lib/shutwork/command/rooms.rb

@@ -8,21 +8,29 @@ require "shutwork/token"
 module Shutwork
   module Command
     class Rooms < Base
-      def parse_args
+      def parse_args args
         opts = OptionParser.new
         opts.program_name = "shutwork #{self.class.name.split(/::/).last.downcase}"
         opts.on("-r", "--raw", "Show results in raw format") { @raw = true }
         opts.on("-v", "--verbose", "Verbose") { @verbose = true }
-        opts.parse ARGV
+        opts.on("--members", "Fetch members instead of messages") { @target = :members }
+        opts.on("--files", "Fetch files instead of messages") { @target = :files }
+        opts.on("--download", "Download files") { @download = true }
+        opts.on("--download-dir DIR", "Set download dir (default: download)") { |dir| @download_dir = dir }
+        opts.parse args
       end
 
-      def run
-        @args = parse_args
+      def run args = []
+        @args = parse_args args
         token = Shutwork::Token.read
         @client = Shutwork::Client.new token: token, verbose: @verbose
 
         if @args.first
-          show @args.first
+          if @download
+            download @args.first
+          else
+            show @args.first
+          end
         else
           list
         end
@@ -33,23 +41,44 @@ module Shutwork
         if @raw
           puts items
         else
-          rooms = JSON.parse(items)
-          rooms.each do |room|
-            puts ("%10s  %s" % [room["room_id"], room["name"]])
-          end
+          JSON.parse(items).each(&method(:display_room))
         end
       end
 
       def show room_id
-        items = @client.room_messages room_id
+        items =
+          case @target
+          when :members
+            @display = :display_account
+            @client.room_members room_id
+          when :files
+            @display = :display_file
+            @client.room_files room_id
+          else
+            @display = :display_message
+            @client.room_messages room_id
+          end
         if @raw
           puts items
         else
-          messages = JSON.parse(items)
-          messages.each do |message|
-            puts "----------------"
-            puts ("%s  %s" % [Time.at(message["send_time"]), message.dig("account", "name")])
-            puts message["body"]
+          JSON.parse(items).each(&method(@display))
+        end
+      end
+
+      def download room_id
+        dir = Pathname.new(@download_dir || "download").join(room_id.to_s)
+        items = @client.room_files room_id
+        files = JSON.parse(items)
+        files.each do |file|
+          dst = dir.join file["filename"]
+          if dst.exist?
+            say_status "exist", dst, :cyan
+          else
+            say_status "download", dst, :yellow
+            x = @client.file room_id, file["file_id"]
+            url = JSON.parse(x)["download_url"]
+            at = Time.at file["upload_time"]
+            download_file url, dst, at
           end
         end
       end

data/lib/shutwork/command/token.rb

@@ -4,16 +4,16 @@ require "shutwork/token"
 module Shutwork
   module Command
     class Token < Base
-      def parse_args
+      def parse_args args
         opts = OptionParser.new
-        opts.program_name = "shutwork #{self.class.name.split(/::/).last.downcase}"
+        opts.program_name = "shutwork #{self.class.name.split(/::/).last.downcase} [TOKEN]"
         opts.on("--skip-verify", "Skips validity check") { @skip_verify = true }
         opts.on("-v", "--verbose", "Verbose") { @verbose = true }
-        opts.parse ARGV
+        opts.parse args
       end
 
-      def run
-        @token = parse_args.first&.strip
+      def run args = []
+        @token = parse_args(args).first&.strip
 
         if @token
           unless @skip_verify

data/lib/shutwork/version.rb

@@ -1,3 +1,3 @@
 module Shutwork
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: shutwork
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Hideaki Kawai
@@ -115,6 +115,7 @@ files:
 - lib/shutwork/cli.rb
 - lib/shutwork/client.rb
 - lib/shutwork/command/base.rb
+- lib/shutwork/command/display.rb
 - lib/shutwork/command/help.rb
 - lib/shutwork/command/me.rb
 - lib/shutwork/command/rooms.rb