TokiCLI 0.0.9 → 0.1.1

data/lib/TokiCLI/authorize.rb

@@ -1,10 +1,12 @@
 require 'json'
 require 'rest_client'
-module ADNAuthorize
+module TokiCLI
+  CLIENT_ID = 'm6AccJFM56ENCn58Vde9cSg3uSpbvAAs'
+  CALLBACK_URL = 'http://aya.io/toki_cli/auth.html'
   class Authorize
-    def initialize
-      @auth_url = "https://account.app.net/oauth/authenticate?client_id=#{TokiCLI::Toki::CLIENT_ID}&response_type=token&redirect_uri=#{TokiCLI::Toki::CALLBACK_URL}&scope=basic,messages&include_marker=1"
-      @home = Dir.home + "/.TokiCLI"
+    def initialize home
+      @auth_url = "https://account.app.net/oauth/authenticate?client_id=#{TokiCLI::CLIENT_ID}&response_type=token&redirect_uri=#{TokiCLI::CALLBACK_URL}&scope=basic,messages&include_marker=1"
+      @home = home
     end
 
     def authorize

data/lib/TokiCLI/export.rb

@@ -0,0 +1,81 @@
+# encoding: utf-8
+module TokiCLI
+  class Export
+
+    def initialize toki, view
+      @toki = toki
+      @view = view
+    end
+
+    def apps_total options, apps, type
+      arr = []
+      apps['data']['apps'].each do |obj|
+        arr << {
+          bundle: obj['bundle'],
+          name: obj['name'],
+          total: {
+            seconds: obj['total']['seconds'],
+            time: @view.readable_time(obj['total']['time'])
+          }
+        }
+      end
+      save options, arr,  "toki-#{type}__#{Time.now.to_s[0..9]}"
+    end
+
+    def log options, log, type
+      arr = []
+      temp = log['data']['log'].map do |obj|
+        [obj[1]['start'], obj[1]['duration']['seconds'], @view.readable_time(obj[1]['duration']['time']), obj[0]]
+      end
+      temp.sort_by! {|obj| obj[0]}
+      temp.each do |obj|
+        arr << {
+          start: obj[0],
+          duration: {
+            seconds: obj[1],
+            time: obj[2]
+            },
+          sync_id: obj[3]
+        }
+      end
+      save_log options, arr,  "toki-#{type}__#{Time.now.to_s[0..9]}"
+    end
+
+    private
+
+    def save options, data, filename
+      case
+      when options[:json]
+        name = @toki.helpers.toki_path + "/#{filename}.json"
+        File.write(name, data.to_json)
+      when options[:csv]
+        name = @toki.helpers.toki_path + "/#{filename}.csv"
+        CSV.open(name, "wb") do |csv|
+          csv << ['Bundle Identifier', 'Name', 'Total seconds', 'Total time']
+          data.each {|line| csv << [line[:bundle], line[:name], line[:total][:seconds], line[:total][:time]]}
+        end
+      else
+        exit
+      end
+      puts "File converted and exported to #{name}\n\n"
+    end
+
+    def save_log options, data, filename
+      case
+      when options[:json]
+        name = @toki.helpers.toki_path + "/#{filename}.json"
+        File.write(name, data.to_json)
+      when options[:csv]
+        name = @toki.helpers.toki_path + "/#{filename}.csv"
+        CSV.open(name, "wb") do |csv|
+          csv << ['Start', 'Duration (seconds)', 'Duration (time)', 'Sync ID']
+          data.each {|line| csv << [line[:start], line[:duration][:seconds], line[:duration][:time], line[:sync_id]]}
+        end
+      else
+        exit
+      end
+      puts "File converted and exported to #{name}\n\n"
+    end
+
+  end
+end

data/lib/TokiCLI/import.rb

@@ -0,0 +1,122 @@
+# encoding: utf-8
+module TokiCLI
+  class Import
+
+    def initialize toki, view
+      @toki_api = toki
+      @view = view
+    end
+
+    def restore token, channel_id
+      please_quit
+      backup_db
+      adn_data = get_adn_data token, channel_id
+      db = create_db
+      create_table db
+      lines = decode adn_data
+      populate db, lines
+      replace
+    end
+
+
+    def replace
+      if File.exist? @toki_api.helpers.db_path
+        puts "\nAre you sure you want to replace the current Toki.app database?\n\nHit SPACE to continue (any other key to cancel).\n\n"
+        abort("Canceled.\n\n") unless STDIN.getch == ' '
+        FileUtils.mv @toki_api.helpers.db_path, "#{Dir.home}/.Trash/"
+      end
+      FileUtils.mv @new_db_path, @toki_api.helpers.db_path
+      puts "\n\nDone. You may relaunch Toki.app now.\n\n"
+    end
+
+    def populate db, lines
+      puts "Populating new database with App.net data...\n\n"
+      before = Time.now
+      idx = 0
+      db.transaction do |db_in_transaction|
+        lines.each do |obj|
+          insert_data = {}
+          insert_data[':id'] = obj[:id]
+          insert_data[':bundleIdentifier'] = obj[:bundleIdentifier]
+          insert_data[':activeTo'] = obj[:activeTo]
+          insert_data[':activeFrom'] = obj[:activeFrom]
+          insert_data[':totalSeconds'] = obj[:totalSeconds]
+          insert_data[':UUID'] = obj[:uuid]
+          insert_data[':synced'] = 1
+          insert_data[':availableToSync'] = 1
+
+          db_in_transaction.prepare("INSERT INTO plain(id, bundleIdentifier, activeTo, activeFrom, totalSeconds, UUID, synced, availableToSync) VALUES(:id, :bundleIdentifier, :activeTo, :activeFrom, :totalSeconds, :UUID, :synced, :availableToSync);") do |insert|
+            insert.execute insert_data
+          end
+
+          idx += 1
+          print "Processed #{idx} of #{lines.size}\r"
+          $stdout.flush
+        end
+      end
+      puts "\n\n\nFinished inserting #{idx} lines in #{Time.now - before}.\n\n"
+    end
+
+    def decode adn_data
+      puts "Decoding App.net data...\n\n"
+      lines = []
+      adn_data.each do |obj|
+        adn = obj['annotations'][0]
+        lines <<
+        {
+          type: adn['type'],
+          table: adn['value']['c'],
+          uuid: adn['value']['d']['UUID'],
+          bundleIdentifier: adn['value']['d']['bundleIdentifier'],
+          activeTo: adn['value']['d']['activeTo'],
+          activeFrom: adn['value']['d']['activeFrom'],
+          totalSeconds: adn['value']['d']['totalSeconds'],
+          id: adn['value']['d']['id']
+        }
+      end
+      lines
+    end
+
+    def create_table db
+      puts "Creating table...\n\n"
+      db.execute_batch <<-SQL
+        CREATE TABLE KKAppActivity (
+          id INTEGER,
+          bundleIdentifier VARCHAR(256),
+          activeTo INTEGER,
+          activeFrom INTEGER,
+          totalSeconds INTEGER,
+          UUID VARCHAR(256),
+          synced INTEGER,
+          availableToSync INTEGER
+        );
+      SQL
+      db.reload_schema!
+    end
+
+    def create_db
+      puts "\n\nCreating a new SQLite3 database...\n\n"
+      @new_db_path = "#{@toki_api.helpers.toki_path}/db_from_adn.sqlite3"
+      Amalgalite::Database.new @new_db_path
+    end
+
+    def get_adn_data token, channel_id
+      puts "Connecting to App.net...\n\n"
+      channels = ADNChannels::GetMessages.new(token)
+      channels.get_messages(channel_id)
+    end
+
+    def backup_db
+      if File.exist? @toki_api.helpers.db_path
+        FileUtils.copy(@toki_api.helpers.db_path, "#{@toki_api.helpers.toki_path}/toki_data.sqlite3.bak")
+        puts "Current database has been backed up in #{@toki_api.helpers.toki_path}\n\n"
+      end
+    end
+
+    def please_quit
+      puts "\n\nYou need to quit Toki.app before restoring data from App.net.\n\nPlease quit Toki.app then hit SPACE to continue (any other key to cancel).\n\n"
+      abort("Canceled.\n\n") unless STDIN.getch == ' '
+    end
+
+  end
+end

data/lib/TokiCLI/scan.rb

@@ -0,0 +1,19 @@
+# encoding: utf-8
+module TokiCLI
+  class Scan
+
+    def initialize toki
+      @toki_api = toki
+    end
+
+    def scan params
+      bundles = @toki_api.helpers.scan params
+      if bundles.empty? || bundles.nil?
+        puts "\nError.\n\n"
+      else
+        puts "\nDone.\n\n"
+      end
+    end
+
+  end
+end

data/lib/TokiCLI/version.rb

@@ -1,3 +1,3 @@
 module TokiCLI
-  VERSION = "0.0.9"
+  VERSION = "0.1.1"
 end

data/lib/TokiCLI/view.rb

@@ -1,70 +1,127 @@
 # encoding: utf-8
 module TokiCLI
   class View
-    def total_table(list)
+
+    def total_table(data)
       table = init_table
-      table.title = "Your apps monitored by Toki"
-      puts table_2(list, table)
+      clear
+      if data['data']['apps']
+        table.title = "Your apps monitored by Toki"
+        puts apps_3(data, table)
+      else
+        table.title = "Your app monitored by Toki"
+        puts total_3(data, table)
+      end
       puts "\n"
     end
-    def day_table(day, list)
+
+    def app_table(asked, app_data)
       table = init_table
-      table.title = "Your apps monitored by Toki on #{day}"
-      puts table_2(list, table)
+      table.style = { :width => 100 }
+      name = name_of app_data
+      table.title = "Toki time tracking for #{name}"
+      table.headings = ['Start', 'Duration', 'Sync ID']
+      lines = create_lines app_data
+      total = calc_total lines
+      format_lines lines
+
+      #TODO: change this
+      display_limit = Time.now - (3600*24*7) # 7 days
+      lines.select! {|obj| Time.parse(obj[0]) > display_limit}
+
+      sort_lines lines
+      puts create_table(lines, table)
       puts "\n"
     end
-    def range_table(day1, day2, list)
-      table = init_table
-      table.title = "Your apps monitored by Toki between #{day1} and #{day2}"
-      puts table_2(list, table)
-      puts "\n"
+
+    def max_30 text
+      text.length >= 30 ? "#{text[0..27]}..." : text
     end
-    def hits_table(list)
-      table = init_table
-      table.title = "Your top apps monitored by Toki"
-      puts table2_index(list, table)
-      puts "\n"
+
+    def force_name name
+      name.nil? ? ' ' : name
     end
-    def app_table(asked, app_data)
-      table = init_table
-      table.style = { :width => 100 }
-      table.title = "Toki time tracking for '#{asked}' => #{app_data.first[1][:name]}"
-      table.headings = ['From', 'To', 'Duration', 'Sync id']
-      @index = 0
-      @length = app_data.length - 1
-      @total = app_data.select {|k,v| k == 'total'}
-      app_data.delete('total')
-      data = app_data.to_a
-      data.sort_by! {|k,v| v[:raw_from] }
-      data.each do |id, log|
-        @index += 1
-        table << ["#{log[:from]}", "#{log[:to]}", "#{log[:duration]}", "#{id}"]
-        table << :separator unless @index == @length
+
+    def humanized_date epoch
+      human = seconds_to_time epoch
+      "#{human[:hours]}h #{human[:minutes]}m #{human[:seconds]}s"
+    end
+
+    def seconds_to_time epoch
+      hours = epoch / 3600
+      minutes = (epoch / 60 - hours * 60)
+      seconds = (epoch - (minutes * 60 + hours * 3600))
+      {hours: hours, minutes: minutes, seconds: seconds}
+    end
+
+    def readable_time time
+      "#{'%.2d' % time['hours']}h #{'%.2d' % time['minutes']}m #{'%.2d' % time['seconds']}s"
+    end
+
+    def clear
+      puts "\e[H\e[2J"
+    end
+
+    private
+
+    def create_table lines, table
+      index = 0
+      length = lines.length - 1
+      lines.each do |obj|
+         table << obj
+         table << :separator unless index == length
+         index += 1
       end
       table << :separator
-      table << [{ :value => "Total: #{@total['total']}", :colspan => 4, :alignment => :center }]
-      puts table
-      puts "\n"
+      table << [{ :value => "Total: #{humanized_date(total)}", :colspan => 3, :alignment => :center }]
+      table
     end
 
-    private
+    def sort_lines lines
+      lines.sort_by! {|obj| obj[0]}
+    end
+
+    def format_lines lines
+      lines.map! {|obj| [obj[0], obj[1], obj[2]]}
+    end
+
+    def calc_total lines
+      total = 0
+      lines.each {|v| total += v[3]}
+      total
+    end
+
+    def create_lines app_data
+      lines = []
+      app_data['data']['log'].each {|k,v| lines << [v['start'], readable_time(v['duration']['time']), k, v['duration']['seconds']]}
+      lines
+    end
+
+    def name_of app_data
+      if app_data['data']['name']
+        app_data['data']['name']
+      else
+        app_data['data']['bundle']
+      end
+    end
 
     def init_table
       Terminal::Table.new do |t|
-        t.style = { :width => 75 }
+        t.style = { :width => 100 }
       end
     end
-    def table_2(list, table)
-      list.each do |k,v|
-        table << ["#{k}", "#{v}"]
+
+    def apps_3 resp, table
+      resp['data']['apps'].each do |obj|
+        table << [max_30(obj['bundle']), max_30(force_name(obj['name'])), readable_time(obj['total']['time'])]
       end
       table
     end
-    def table2_index(list, table)
-      list.each.with_index(1) do |obj,index|
-        table << ["#{index}", "#{obj[0]}", "#{obj[1]}"]
-      end
+
+    def total_3 resp, table
+      table << [max_30(resp['data']['bundle']), max_30(force_name(resp['data']['name'])), readable_time(resp['data']['total']['time'])]
       table
     end
+
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: TokiCLI
 version: !ruby/object:Gem::Version
-  version: 0.0.9
+  version: 0.1.1
 platform: ruby
 authors:
 - Eric Dejonckheere
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-05-29 00:00:00.000000000 Z
+date: 2014-06-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor
@@ -72,14 +72,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 2.2.8
+        version: '2.2'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 2.2.8
+        version: '2.2'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -126,12 +126,16 @@ files:
 - Rakefile
 - TokiCLI.gemspec
 - bin/toki
+- lib/API/dbapi.rb
+- lib/API/helpers.rb
 - lib/TokiCLI.rb
 - lib/TokiCLI/app.rb
 - lib/TokiCLI/authorize.rb
+- lib/TokiCLI/export.rb
 - lib/TokiCLI/get_channels.rb
 - lib/TokiCLI/get_messages.rb
-- lib/TokiCLI/module.rb
+- lib/TokiCLI/import.rb
+- lib/TokiCLI/scan.rb
 - lib/TokiCLI/search_messages.rb
 - lib/TokiCLI/status.rb
 - lib/TokiCLI/version.rb
@@ -148,7 +152,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 1.9.3
+      version: 2.0.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
@@ -161,4 +165,3 @@ signing_key:
 specification_version: 4
 summary: Toki.app command-line client
 test_files: []
-has_rdoc: