TokiCLI 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 24126edaca561eab0623a449445132dddf6697a0
+  data.tar.gz: bec8ef8903fd53c79b8ef96696dc04fde69e9e7b
+SHA512:
+  metadata.gz: 985a22eda651b23467d18db8e7198edd24c5454275a09958e31b4d38a74f9a4d44031e30033302dd7cba00a3485b0b7301470464f605ce94170d2e1f7a8ab2f3
+  data.tar.gz: 8168f8fa325678e6e4d332476ee0d889a9819bbcf9a917e0cf89d7442b29c5295350f049a0f1ad5f19168f69193fb723df3ff0354767d00dab3247970d71dbac

data/.gitignore

@@ -0,0 +1,23 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+*.bundle
+*.so
+*.o
+*.a
+mkmf.log
+token

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in TokiCLI.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 Eric Dejonckheere
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,13 @@
+# TokiCLI
+
+TODO: Write a gem description
+
+## Installation
+
+    $ gem install TokiCLI
+
+## Usage
+
+TODO: Write usage instructions here
+
+

data/Rakefile

@@ -0,0 +1,2 @@
+require "bundler/gem_tasks"
+

data/TokiCLI.gemspec

@@ -0,0 +1,30 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'TokiCLI/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "TokiCLI"
+  spec.version       = TokiCLI::VERSION
+  spec.authors       = ["Eric Dejonckheere"]
+  spec.email         = ["eric@aya.io"]
+  spec.summary       = %q{Toki command-line client}
+  spec.description   = %q{Toki command-line client: read your Toki data from the local database or from the App.net backup channel}
+  spec.homepage      = ""
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.required_ruby_version = '>= 1.9.3'
+
+  spec.add_dependency "thor", ">= 0.18"
+  spec.add_dependency "rest-client", ">= 1.6"
+  spec.add_dependency "amalgalite", ">= 1.3"
+  spec.add_dependency "terminal-table", ">= 1.4"
+
+  spec.add_development_dependency "bundler", "~> 1.6"
+  spec.add_development_dependency "rake"
+end

data/bin/toki

@@ -0,0 +1,10 @@
+#!/usr/bin/env ruby
+# encoding: utf-8
+# Toki command-line client
+# Author: Eric Dejonckheere (http://app.net/ericd)
+# Toki by: Keita Kobayashi (http://app.net/keita)
+
+$PROGRAM_NAME = 'toki'
+require_relative '../lib/TokiCLI'
+
+TokiCLI::App.start ARGV

data/lib/TokiCLI.rb

@@ -0,0 +1,7 @@
+# encoding: utf-8
+require 'ostruct'
+require 'thor'
+require 'amalgalite'
+require 'terminal-table/import'
+require_relative "TokiCLI/version"
+require_relative "TokiCLI/app"

data/lib/TokiCLI/app.rb

@@ -0,0 +1,114 @@
+# encoding: utf-8
+module TokiCLI
+  class App < Thor
+    package_name "TokiCLI"
+    %w{get_messages search_messages get_channels module status view}.each {|r| require_relative "#{r}"}
+
+    desc "total", "Shows the total usage of all apps"
+    option :adn, aliases: '-a', type: :boolean, desc: 'Select ADN channel as source (toki total -a)'
+    def total
+      if options[:adn]
+        toki = total_adn
+      else
+        toki = total_db
+      end
+      get_total(toki, options)
+    end
+
+    desc "auth", "Input your App.net token for authorization"
+    def auth
+      puts Status.paste_token
+      token = STDIN.gets.chomp!
+      unless token.nil?
+        save_token(token)
+      else
+        abort(Status.no_token)
+      end
+      puts Status.done
+    end
+
+    desc "log APP", "Shows the complete log for an app"
+    option :adn, aliases: '-a', type: :boolean, desc: 'Select ADN channel as source (toki total -a)'
+    def log(*args)
+      abort(Status.specify_name) if args.empty?
+      if options[:adn]
+        toki = total_adn
+      else
+        toki = total_db
+      end
+      asked = args[0]
+      synced = toki.get_content(options)
+      app_data = toki.get_app(asked, synced)
+      v = TokiCLI::View.new
+      puts v.app_table(asked, app_data)
+      puts "\n"
+    end
+
+    private
+
+    def total_adn
+      clear
+      puts Status.get_all
+      init_toki_adn
+    end
+
+    def total_db
+      clear
+      puts "getting data from db\n"
+      init_toki_db
+    end
+
+    def get_total(toki, options)
+      begin
+        apps = get_all(toki, options)
+      rescue Interrupt
+        puts Status.canceled
+        exit
+      rescue => e
+        puts Status.error(e)
+        exit
+      end
+      clear
+      puts TokiCLI::View.new.total_table(apps)
+      puts "\n"
+    end
+
+    def init_toki_db
+      TokiCLI::Toki.new(nil, nil)
+    end
+
+    def init_toki_adn(channel_id = get_channel_id)
+      TokiCLI::Toki.new("#{get_token}", channel_id)
+    end
+
+    def save_token(token)
+      File.write(Dir.home + '/.toki_token', token)
+    end
+    def get_token
+      if File.exist?(Dir.home + '/.toki_token')
+        File.read(Dir.home + '/.toki_token').chomp
+      else
+        abort Status.run_auth
+      end
+    end
+    def get_channel_id #TODO return ch['counts']['messages'] too
+      channels = ADNChannels::GetChannels.new(get_token).get_channels
+      channel_id = ''
+      channels.each do |ch|
+        if ch['type'] == 'us.kkob.toki.sync-b'
+          channel_id = ch['id']
+          break
+        end
+      end
+      return channel_id.to_i unless channel_id == ''
+      puts Status.no_channel
+      exit
+    end
+    def clear
+      puts "\e[H\e[2J"
+    end
+    def get_all(t, options)
+      t.all_data(t.get_content(options))
+    end
+  end
+end

data/lib/TokiCLI/get_channels.rb

@@ -0,0 +1,22 @@
+require 'json'
+require 'rest_client'
+module ADNChannels
+  class GetChannels
+    def initialize(token)
+      @base_url = 'http://api.app.net'
+      @token = token
+    end
+    def get_channels
+      args = {:count => 200, :before_id => nil}
+      channels = []
+      loop do
+        @url = "#{@base_url}/users/me/channels?access_token=#{@token}&include_machine=1&include_message_annotations=1&include_deleted=0&include_html=0&count=#{args[:count]}&before_id=#{args[:before_id]}"
+        resp = JSON.parse(RestClient.get(@url))
+        resp['data'].each { |m| channels << m }
+        break unless resp['meta']['more']
+        args = {:count => 200, :before_id => resp['meta']['min_id']}
+      end
+      channels
+    end
+  end
+end

data/lib/TokiCLI/get_messages.rb

@@ -0,0 +1,32 @@
+require 'json'
+require 'rest_client'
+module ADNChannels
+  class GetMessages
+    def initialize(token)
+      @base_url = 'http://api.app.net'
+      @token = token
+    end
+    def get_messages(channel_id)
+      args = {:count => 200, :before_id => nil}
+      messages = []
+      @index = 1
+      puts "Downloading synced objects (~1MB/page):\n\n"
+      loop do
+        puts "--Downloading from page #{@index}...\n"
+        @url = "#{@base_url}/channels/#{channel_id}/messages?access_token=#{@token}&include_machine=1&include_message_annotations=1&include_deleted=0&include_html=0&count=#{args[:count]}&before_id=#{args[:before_id]}"
+        begin
+          data = RestClient.get(@url)
+        rescue Interrupt
+          puts TokiCLI::Status.canceled
+          exit
+        end
+        resp = JSON.parse(data)
+        resp['data'].each { |m| messages << m }
+        break unless resp['meta']['more']
+        @index += 1
+        args = {:count => 200, :before_id => resp['meta']['min_id']}
+      end
+      messages
+    end
+  end
+end

data/lib/TokiCLI/module.rb

@@ -0,0 +1,142 @@
+# encoding: utf-8
+module TokiCLI
+  class Toki
+    def initialize(token, channel_id)
+      @token = token
+      @channel_id = channel_id
+    end
+    def get_content(options)
+      synced = []
+      if options[:adn]
+        messages = get_adn_data
+        messages.reverse.each do |obj|
+          synced << make_obj_from_adn(obj)
+        end
+      else
+        content = get_db_data
+        content.each do |obj|
+          synced << make_obj_from_db(obj)
+        end
+      end
+      synced
+    end
+    def get_app(app, synced)
+      name_regex = /#{app.downcase}/
+      app_data = {}
+      totals = {}
+      synced.each do |obj|
+        next unless obj.app[:name].downcase =~ name_regex
+        unless totals[:app].nil?
+          totals[:app] = totals[:app] + obj.app[:total]
+        else
+          totals[:app] = obj.app[:total]
+        end
+        human = epoch_to_human(obj.app[:total])
+        app_data[obj.msg[:id]] = {
+          from: Time.at(obj.app[:active_from]).strftime("%Y/%m/%d %Hh:%Mm:%Ss"),
+          to: Time.at(obj.app[:active_to]).strftime("%Y/%m/%d %Hh:%Mm:%Ss"),
+          duration: "#{human[:hours]}h #{human[:minutes]}m #{human[:seconds]}s"
+        }
+      end
+      human = epoch_to_human(totals[:app])
+      app_data['total'] = "#{human[:hours]}h #{human[:minutes]}m #{human[:seconds]}s"
+      app_data
+    end
+    def all_data(synced)
+      totals = {}
+      synced.each do |obj|
+        unless totals[obj.app[:name]].nil?
+          totals[obj.app[:name]] = totals[obj.app[:name]] + obj.app[:total]
+        else
+          totals[obj.app[:name]] = obj.app[:total]
+        end
+      end
+      bundles = {}
+      sorted = totals.sort_by{ |k,v| v }
+      sorted.each do |name, total|
+        human = epoch_to_human(total)
+        if human[:hours] < 1
+          bundles[name] = "#{human[:minutes]}m #{human[:seconds]}s"
+        else
+          bundles[name] = "#{human[:hours]}h #{human[:minutes]}m #{human[:seconds]}s"
+        end
+      end
+      bundles
+    end
+
+    private
+
+    def epoch_to_human(epoch)
+      hours = epoch / 3600
+      minutes = (epoch / 60 - hours * 60)
+      seconds = (epoch - (minutes * 60 + hours * 3600))
+      {hours: hours, minutes: minutes, seconds: seconds}
+    end
+
+    def get_adn_data
+      channels = ADNChannels::GetMessages.new(@token) #ericd
+      channels.get_messages(@channel_id)
+    end
+
+    def get_db_data
+      db = open_db
+      data = db.execute("SELECT * FROM KKAppActivity")
+      data
+    end
+
+
+
+    def open_db
+      tmp_path = "#{Dir.home}/temp/toki/"
+      db_path = "#{Dir.home}/Library/Containers/us.kkob.Toki/Data/Documents/toki_data.sqlite3"
+      FileUtils.mkdir_p(tmp_path)
+      FileUtils.copy(db_path, "#{tmp_path}/toki_data.bak")
+      Amalgalite::Database.new(db_path)
+    end
+
+    def make_obj_from_adn(obj)
+      msg = {}
+      usr = {}
+      app = {}
+      d = obj['annotations'][0]['value']['d']
+      msg[:id] = obj['id']
+      msg[:thread] = obj['thread_id']
+      msg[:date] = parsed_date obj['created_at']
+      usr[:username] = obj['user']['username']
+      usr[:name] = obj['user']['name']
+      usr[:id] = obj['user']['id']
+      app[:id] = d['id']
+      app[:uuid] = d['UUID']
+      app[:name] = d['bundleIdentifier']
+      app[:active_to] = d['activeTo']
+      app[:active_from] = d['activeFrom']
+      app[:total] = d['totalSeconds']
+      OpenStruct.new(app: app, msg: msg, usr: usr)
+    end
+
+    def make_obj_from_db(obj)
+      # content is an array:
+      # id (INTEGER) 0, bundleIdentifier (VARCHAR) 1, activeFrom (INTEGER) 2, activeTo (INTEGER) 3, totalSeconds (INTEGER) 4, UUID (VARCHAR) 5, synced (INTEGER) 6, availableToSync (INTEGER) 7
+      msg = {}
+      usr = {}
+      app = {}
+      msg[:id] = rand(100000000)
+      msg[:thread] = msg[:id]
+      msg[:date] = Time.now
+      usr[:username] = ENV['USERNAME']
+      usr[:name] = usr[:username]
+      usr[:id] = obj[5]
+      app[:id] = obj[0]
+      app[:uuid] = obj[5]
+      app[:name] = obj[1]
+      app[:active_to] = obj[3]
+      app[:active_from] = obj[2]
+      app[:total] = obj[4]
+      OpenStruct.new(app: app, msg: msg, usr: usr)
+    end
+
+    def parsed_date(string)
+      "#{string[0...10]} #{string[11...19]}"
+    end
+  end
+end

data/lib/TokiCLI/search_messages.rb

@@ -0,0 +1,23 @@
+require 'json'
+require 'rest_client'
+module ADNChannels
+  class SearchMessages
+    def initialize(token)
+      @base_url = 'http://api.app.net'
+      @token = token
+    end
+    def search_annotation(channel_id, query)
+      args = {:count => 200, :before_id => nil}
+      messages = []
+      loop do
+        @url = "#{@base_url}/channels/messages/search?query=#{query}&channel_ids=#{channel_id}&access_token=#{@token}&include_machine=1&include_message_annotations=1&include_deleted=0&count=#{args[:count]}&before_id=#{args[:before_id]}"
+        resp = JSON.parse(RestClient.get(@url))
+        resp['data'].each { |m| messages << m }
+        break unless resp['meta']['more']
+        args = {:count => 200, :before_id => resp['meta']['min_id']}
+      end
+      messages
+    end
+  end
+end
+

data/lib/TokiCLI/status.rb

@@ -0,0 +1,32 @@
+# encoding: utf-8
+module TokiCLI
+  class Status
+    def self.get_all
+      "Toki_CLI is crawling your Toki App.net channel to get your data.\nIt could take a while, please be patient.\n\n"
+    end
+    def self.canceled
+      "\n\nCanceled.\n\n"
+    end
+    def self.error(e)
+      "\n\nStopped. Error: \n#{e}\n\n"
+    end
+    def self.no_channel
+      "\n\nCan't find your Toki App.net channel, sorry.\n\n"
+    end
+    def self.paste_token
+      "\nPlease paste your App.net token:\n\n"
+    end
+    def self.no_token
+      "\nOops, I didn't get the token. Please try again.\n"
+    end
+    def self.done
+      "\nDone.\n"
+    end
+    def self.specify_name
+      "\nPlease specify the name of the app (or part of it).\n\n"
+    end
+    def self.run_auth
+      "\nPlease run `toki auth` first.\n\n"
+    end
+  end
+end

data/lib/TokiCLI/version.rb

@@ -0,0 +1,3 @@
+module TokiCLI
+  VERSION = "0.0.1"
+end

data/lib/TokiCLI/view.rb

@@ -0,0 +1,40 @@
+# encoding: utf-8
+module TokiCLI
+  class View
+    def total_table(list)
+      table = init_table
+      table.title = "Your apps monitored by Toki"
+      list.each do |k,v|
+        table << ["#{k}", "#{v}"]
+      end
+      table
+    end
+    def app_table(asked, app_data)
+      table = init_table
+      table.title = "Toki time tracking for '#{asked}'"
+      table.headings = ['From', 'To', 'Duration']
+      @index = 0
+      @length = app_data.length - 1
+      app_data.each do |id, log|
+        if id == 'total'
+          @total = log
+          next
+        end
+        @index += 1
+        table << ["#{log[:from]}", "#{log[:to]}", "#{log[:duration]}"]
+        table << :separator unless @index == @length
+      end
+      table << :separator
+      table << [{ :value => "Total: #{@total}", :colspan => 3, :alignment => :center }]
+      table
+    end
+
+    private
+
+    def init_table
+      Terminal::Table.new do |t|
+        t.style = { :width => 75 }
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,147 @@
+--- !ruby/object:Gem::Specification
+name: TokiCLI
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Eric Dejonckheere
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-04-18 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0.18'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0.18'
+- !ruby/object:Gem::Dependency
+  name: rest-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.6'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.6'
+- !ruby/object:Gem::Dependency
+  name: amalgalite
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: terminal-table
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.4'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.4'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: 'Toki command-line client: read your Toki data from the local database
+  or from the App.net backup channel'
+email:
+- eric@aya.io
+executables:
+- toki
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- TokiCLI.gemspec
+- bin/toki
+- lib/TokiCLI.rb
+- lib/TokiCLI/app.rb
+- lib/TokiCLI/get_channels.rb
+- lib/TokiCLI/get_messages.rb
+- lib/TokiCLI/module.rb
+- lib/TokiCLI/search_messages.rb
+- lib/TokiCLI/status.rb
+- lib/TokiCLI/version.rb
+- lib/TokiCLI/view.rb
+homepage: ''
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 1.9.3
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: Toki command-line client
+test_files: []
+has_rdoc: