slackup 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 217865baf5660d9ee09b17a815f9849c8f714f1e
-  data.tar.gz: 27907d862fd07bbb93b7aa1cc4bdaa8f7ea4133b
+  metadata.gz: 9f775a52592ebb2b334b49c07577edb1d63ae909
+  data.tar.gz: c8f07567b097c4138ddccf4d1359c711d24dc6c4
 SHA512:
-  metadata.gz: 5e83bd95a1104f5a24eb818f2554163af2ac4218ba595ed876558b7a72f776948b54062b78c054e52dbff76d4b84d2a0f1ec4f32a1d9927d1a6207559578285b
-  data.tar.gz: 69a6419c81695b8a83d62e239881b7843278181558058fa46405494a8c99cb9a26d90367e1aa30545f0f6209367e2827dda3c90e1ffe9052f9a91011f44bb340
+  metadata.gz: a8991c5c8a2b0227974db3c2e0ccb33a3d6a69df8154145a1b9e90bfe9aa17a64fadb158999cf9867119a9b0f152744c9c6555359aec2dc0fd0c532d199ad494
+  data.tar.gz: 6dadcf2bc8c6adc5d0bd692800b541176a90664a7f93b251d9786e36e5727b9e09c8ee208c47bca2cef21e45cefa05d2a7da4c945e83638415cf53109fcea8cd

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    slackup (0.0.2)
+    slackup (0.0.3)
       slack-api (~> 1.1, >= 1.1.3)
 
 GEM

data/README.md

@@ -45,6 +45,8 @@ This gem is does the bare basics of backup and works for me.
 
 It has no tests.
 
+It depends on the ['slack-api' gem](https://github.com/aki017/slack-ruby-gem)
+
 ## Contributing
 
 1. Fork it ( https://github.com/[my-github-username]/slackup/fork )

data/lib/slackup/version.rb

@@ -1,3 +1,3 @@
 module Slackup
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/lib/slackup.rb

@@ -42,12 +42,18 @@ class Slackup
   def execute
     SEMAPHORE.synchronize do
       authorize! &&
-      Dir.chdir(name) do
-        channels.each do |channel|
-          write_channel_messages(channel)
+        Dir.chdir(name) do
+          channels.each do |channel|
+            write_channel_messages(channel)
+          end
+          write_stars
+          write_users
+          Dir.chdir(ims_dir) do
+            im_list.each do |im|
+              write_im_messages(im)
+            end
+          end
         end
-        write_stars
-      end
     end
   end
 
@@ -67,45 +73,162 @@ class Slackup
     end
   end
 
+  User = Struct.new(:user_hash) do
+    def id; user_hash["id"]; end
+
+    def name; user_hash["name"]; end
+
+    def deleted; user_hash["deleted"]; end
+
+    def color; user_hash["color"]; end
+
+    def profile; user_hash["profile"]; end
+
+    def admin?; user_hash["is_admin"]; end
+
+    def owner?; user_hash["is_owner"]; end
+
+    def has_2fa?; user_hash["has_2fa"]; end
+
+    def has_files?; user_hash["has_files"]; end
+
+    def to_hash; user_hash; end
+  end
+  # {
+  #   "ok": true,
+  #   "members": [
+  #     {
+  #       "id": "U023BECGF",
+  #       "name": "bobby",
+  #       "deleted": false,
+  #       "color": "9f69e7",
+  #       "profile": {
+  #         "first_name": "Bobby",
+  #         "last_name": "Tables",
+  #         "real_name": "Bobby Tables",
+  #         "email": "bobby@slack.com",
+  #         "skype": "my-skype-name",
+  #         "phone": "+1 (123) 456 7890",
+  #         "image_24": "https:\/\/...",
+  #         "image_32": "https:\/\/...",
+  #         "image_48": "https:\/\/...",
+  #         "image_72": "https:\/\/...",
+  #         "image_192": "https:\/\/..."
+  #       },
+  #       "is_admin": true,
+  #       "is_owner": true,
+  #       "has_2fa": false,
+  #       "has_files": true
+  #     },
+  #   ]
+  # }
   def users
-    @users ||= Slack.users_list["members"]
+    @users ||= Slack.users_list["members"].map { |member| User.new(member) }
   end
 
   def channels
     @channels ||= Slack.channels_list["channels"]
   end
 
+  Im = Struct.new(:im_hash) do
+    def id; im_hash["id"]; end
+
+    def user; im_hash["user"]; end
+  end
+  # @return [Hash]
+  # @example
+  # {
+  #   "ok"=>true,
+  #   "ims"=>[
+  #     {"id"=>"D1234567890", "is_im"=>true, "user"=>"USLACKBOT", "created"=>1372105335, "is_user_deleted"=>false},
+  #   ]
+  # }
+  def im_list
+    @im_list ||= Slack.im_list["ims"].map { |im| Im.new(im) }
+  end
+
+  # @param im_id [String] is the 'channel' of the im, e.g. "D1234567890"
+  # @return [Hash]
+  # @example return
+  # {
+  #   "ok": true,
+  #   "latest": "1358547726.000003",
+  #   "messages": [
+  #     {
+  #       "type": "message",
+  #       "ts": "1358546515.000008",
+  #       "user": "U2147483896",
+  #       "text": "Hello"
+  #     },
+  #     ]
+  #   "has_more": false
+  def im_history(im_id)
+    Slack.im_history(channel: im_id)
+  end
+
   def write_channel_messages(channel)
-    messages = Slack.channels_history({channel: channel["id"], count: "1000"})["messages"]
-    File.open(backup_filename(channel['name']),"w")  do |f|
-      f.write format_channel_messages(messages)
+    messages = Slack.channels_history(channel: channel["id"], count: "1000")["messages"]
+    File.open(backup_filename(channel["name"]), "w")  do |f|
+      formatted_messages = format_channel_messages(messages)
+      f.write serialize(formatted_messages)
     end
   end
 
-  def format_channel_messages(messages)
+  def format_messages(messages)
     messages.reverse.map { |msg|
-      if (msg.has_key?("text") && msg.has_key?("user"))
-        users.each do |user|
-          if user["id"] == msg["user"]
-            break msg["user"] = user["name"]
-          end
-        end
+      if msg.has_key?("text") && msg.has_key?("user")
+        msg["user"] = user_name(msg["user"])
         msg
       else
         nil
       end
-    }.compact.to_yaml
+    }.compact
+  end
+
+  alias_method :format_channel_messages, :format_messages
+  alias_method :format_im_messages, :format_messages
+
+  # gets user name for an id, if mapping is known, else returns the input
+  def user_name(user_id)
+    @user_names ||= users.each_with_object({}) {|user, lookup|
+      lookup[user.id] = user.name
+    }
+    @user_names.fetch(user_id) {
+      user_id
+    }
   end
 
   def write_stars
-    File.open(backup_filename("stars"),"w")  do |f|
+    File.open(backup_filename("stars"), "w")  do |f|
       stars = Slack.stars_list(count: "1000", page: "1")
-      f.write(format_stars(stars))
+      f.write(serialize(stars))
+    end
+  end
+
+  def write_users
+    File.open(backup_filename("users"), "w")  do |f|
+      f.write(serialize(users.map(&:to_hash)))
+    end
+  end
+
+  def serialize(obj)
+    obj.to_yaml
+  end
+
+  def write_im_messages(im)
+    messages = im_history(im.id)["messages"]
+    im_username = user_name(im.user).downcase.gsub(/\s+/, "-")
+    formatted_messages = format_im_messages(messages)
+    return if formatted_messages.empty?
+    File.open(backup_filename(im_username), "w")  do |f|
+      f.write serialize(formatted_messages)
     end
   end
 
-  def format_stars(stars)
-    stars.to_yaml
+  def ims_dir
+    @ims_dir ||= "ims"
+    FileUtils.mkdir_p(@ims_dir)
+    @ims_dir
   end
 
   def backup_filename(name)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: slackup
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Benjamin Fleischer
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2015-05-14 00:00:00.000000000 Z
+date: 2015-05-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: slack-api