TokiCLI 0.2.1 → 0.3.0

data/TokiCLI.gemspec

@@ -18,8 +18,6 @@ Gem::Specification.new do |spec|
   spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
   spec.require_paths = ["lib"]
 
-  spec.required_ruby_version = '>= 2.0.0'
-
   spec.add_dependency "thor", "~> 0.18"
   spec.add_dependency "rest-client", "~> 1.6"
   spec.add_dependency "amalgalite", "~> 1.3"
@@ -28,7 +26,10 @@ Gem::Specification.new do |spec|
   spec.add_dependency 'sinatra', '~> 1.4', '>= 1.4.5'
   spec.add_dependency "thin", "~> 1.6"
   spec.add_dependency 'sinatra-assetpack', "~> 0.3", '>= 0.3.2'
+  spec.add_dependency "sinatra-contrib", '~> 1.4', '>= 1.4.2'
 
-  spec.add_development_dependency "bundler", "~> 1.6"
-  spec.add_development_dependency "rake"
+  spec.add_development_dependency "bundler", "~> 1.7"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rspec", "~> 3.0"
+  spec.add_development_dependency "coveralls"
 end

data/bin/toki

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

data/lib/API/helpers.rb

@@ -1,195 +1,52 @@
 # encoding: utf-8
-module TokiCLI
-  class Helpers
-
-    attr_accessor :table, :bundles, :bundles_file, :home_path, :toki_path, :db_path
-    attr_reader :db
-
-    def initialize
-      @home_path = Dir.home
-      @toki_path = "#{@home_path}/.TokiCLI"
-      FileUtils.mkdir_p(@toki_path) unless Dir.exist?(@toki_path)
-      @db_path = "#{@home_path}/Library/Containers/us.kkob.Toki/Data/Documents/toki_data.sqlite3"
-      if File.exist? @db_path
-        FileUtils.copy @db_path, "#{@toki_path}/toki_data.sqlite3.bak"
-        @db = Amalgalite::Database.new @db_path
-      else
-        raise 'No DB'
-      end
-      @table = 'KKAppActivity'
-      @bundles_file = "#{@toki_path}/apps.json"
-      @bundles = if File.exist? @bundles_file
-        JSON.parse(File.read @bundles_file)
-      else
-        {}
-      end
-    end
-
-
-    # Returns the app name if exists
-    def find_app_name bundle_id
-      if @bundles[bundle_id]
-        @bundles[bundle_id]
-      else
-        nil
-      end
-    end
 
-    def humanized_date epoch
-      human = seconds_to_time epoch
-      "#{'%.2d' % human[:hours]}h #{'%.2d' % human[:minutes]}m #{'%.2d' % human[:seconds]}s"
-    end
+module TokiCLI
 
-    def humanized_minutes_seconds epoch
-      human = seconds_to_time epoch
-      "#{'%.2d' % human[:minutes]}m #{'%.2d' % human[:seconds]}s"
-    end
+  class Helpers
 
-    def seconds_to_time epoch
+    def sec_to_time(secs)
       begin
-        hours = epoch / 3600
-        minutes = (epoch / 60 - hours * 60)
-        seconds = (epoch - (minutes * 60 + hours * 3600))
-        {hours: hours, minutes: minutes, seconds: seconds}
+        hours = secs / 3600
+        minutes = (secs / 60 - hours * 60)
+        seconds = (secs - (minutes * 60 + hours * 3600))
+        {'hours' => hours, 'minutes' => minutes, 'seconds' => seconds}
       rescue Exception => e
-        oops e
+        raise e, Status.wtf
       end
     end
 
-    def oops error
-      puts "\nOOPS! A gremlin crashed the app.\n\n"
-      puts "STACK: #{caller}\n\n"
-      puts "GREMLIN: #{error}\n\n"
-    end
-
-    def epoch_to_date epoch
+    def epoch_to_date(epoch)
       Time.at(epoch).to_time
     end
 
-    def sec_to_time sec
-      seconds_to_time sec
-    end
-
-    # Scans for names from bundle ids, saves the file and returns its content
-    def scan params = {}
-      bundle_ids = get_bundle_ids params
-      f = File.new "#{@toki_path}/apps.json", "w"
-      f.write bundle_ids.to_json
-      f.close
-      bundle_ids
-    end
-
-    def scan_app_icons
-      @bundles.each do |k,v|
-        gs = v.gsub(' ', '\ ')
-        where = "/Applications/#{gs}.app/Contents/Resources/*.icns"
-        files = Dir.glob(where)
-        candidates = files.select {|f| File.basename(f) ==  "Icon.icns" || File.basename(f, ".icns").downcase =~ /#{v.downcase}/}
-        puts "#{v} => #{candidates}"
-      end
-    end
-
-    # Scan for names from bundle ids
-    def get_bundle_ids params = {}
-      @infos = {}
-      get_bundles(get_plists("/Applications/*/Contents/*"), params)
-      get_bundles(get_plists("/Applications/Utilities/*/Contents/*"), params)
-      get_bundles(get_plists("#{@home_path}/Applications/*/Contents/*"), params)
-      specials = {
-        'com.evernote.EvernoteHelper' => 'Evernote Helper',
-        'com.apple.finder' => 'Finder',
-        'com.apple.ReportPanic' => 'Kernel Panic',
-        'com.apple.coreservices.uiagent' => 'CoreServices UIAgent',
-        'com.apple.installer' => 'Apple Installer',
-        'com.apple.frameworks.diskimages.diuiagent' => 'Apple Disk Images Agent',
-        'com.mediaatelier.CheatSheet' => 'Cheat Sheet',
-        'com.vyprvpn.authorization' => 'VyprVPN',
-        'com.apple.WebKit.WebContent' => 'Safari module',
-        'com.apple.ProblemReporter' => 'Apple Problem Reporter',
-        'org.andymatuschak.sparkle.finish-installation' => 'Sparkle Install',
-        'com.noodlesoft.HazelHelper' => 'Hazel Helper',
-        'de.appsolute.MAMP' => 'MAMP',
-        'com.apple.KeyboardSetupAssistant' => 'Apple Keyboard Setup Assistant',
-        'com.apple.NetAuthAgent' => 'Apple Net Auth Agent',
-        'com.adobe.acc.AdobeCreativeCloud' => 'Adobe Creative Cloud',
-        'com.apple.iphonesimulator' => 'Apple iPhone Simulator',
-        'com.macromates.TextMate.preview' => 'TextMate',
-        'com.alfredapp.Alfred' => 'Alfred',
-        'com.runningwithcrayons.Alfred-Preferences' => 'Alfred Preferences',
-        'com.apple.WebKit.PluginProcess' => 'Safari Plugin',
-        'com.apple.ScreenSharing' => 'Apple Screen Sharing',
-        'org.virtualbox.app.VirtualBoxVM' => 'VirtualBox',
-        '2BUA8C4S2C.com.agilebits.onepassword-osx-helper' => '1Password Helper'
-      }
-      specials.each do |k, v|
-        @infos[k] = v
-      end
-      @infos
-    end
-
-    def get_bundle_from_name name
-      (@bundles).each do |k,v|
-        if v.downcase =~ /#{name.downcase}/
-          return [v, k]
-        end
+    def check_date_validity(day)
+      begin
+        DateTime.strptime(day, '%Y-%m-%d')
+      rescue ArgumentError, TypeError => e
+        false
       end
-      []
-    end
-
-    def meta_nodata
-      {
-        code: 404,
-        message: 'No data returned from the database'
-      }
-    end
-
-    def json_nodata
-      {
-        meta: meta_nodata,
-        data: {}
-      }.to_json
     end
 
-    def tracked_bundles apps
-      tracked = JSON.parse(apps)
-      list = []
-      tracked['data']['apps'].each {|v| list << v['bundle']}
-      list
+    def readable_time(obj)
+      "#{obj['hours']}h #{'%.2d' % obj['minutes']}m #{'%.2d' % obj['seconds']}s"
     end
 
-    private
-
-    def get_plists path
-      Dir.glob(path).select {|f| (File.split f).last == 'Info.plist'}
+    def readable_time_log(obj)
+      "#{'%.2d' % obj['minutes']}m #{'%.2d' % obj['seconds']}s"
     end
 
-    def get_bundles plists, params
-      plists.each do |obj|
-        puts "Analyzing #{obj} ...\n" if params[:verbose] == true
-        begin
-          pl = CFPropertyList::List.new(:file => obj)
-        rescue CFFormatError
-          bundles_error if params[:verbose] == true
-          next
-        rescue NoMethodError
-          bundles_error if params[:verbose] == true
-          next
-        end
-        data = CFPropertyList.native_types(pl.value)
-        name = data['CFBundleName']
-        bundle_id = data['CFBundleIdentifier']
-        if name.nil?
-          name = data['CFBundleExecutable']
-        end
-        next if name.nil?
-        next if bundle_id.nil? || bundle_id.empty?
-        @infos[bundle_id] = name
-      end
+    def calc_apps_total(data)
+      total = 0
+      data.each { |obj| total += obj['total']['seconds'] }
+      return total
     end
 
-    def bundles_error
-      puts "Unable to read the file, skipping...\n"
+    def calc_logs_total(data)
+      total = 0
+      data.each { |k, v| total += v['duration']['seconds'] }
+      return total
     end
 
   end
+
 end

data/lib/API/toki_api.rb

@@ -0,0 +1,227 @@
+# encoding: utf-8
+
+module TokiCLI
+
+  class TokiAPI
+
+    require 'json'
+    require 'amalgalite'
+
+    require_relative 'helpers'
+    require_relative 'toki_db'
+
+    attr_reader :bundles, :response, :helpers
+
+    def initialize(db_path, bundles = nil)
+      @db = TokiCLI::TokiDB.new(db_path)
+      @helpers = TokiCLI::Helpers.new
+      @bundles = bundles
+    end
+
+    def apps_total
+      request = {command: 'apps_total', type: 'apps', args: [], processed_at: Time.now}
+      resp = @db.apps_total
+      return response_wrapper(request, resp) { make_apps_list(resp) }
+    end
+
+    def apps_top(number = 5)
+      request = {command: 'apps_top', type: 'apps', args: [number], processed_at: Time.now}
+      resp = @db.apps_total
+      return invalid_response(request) if resp.empty?
+      index = -number
+      list = make_apps_list(resp)[index..-1]
+      @response = make_basic_response(request, list)
+    end
+
+    def apps_day(day)
+      request = {command: 'apps_day', type: 'apps', args: [day], processed_at: Time.now}
+      date = @helpers.check_date_validity(day)
+      return invalid_response(request) if date == false
+      resp = @db.apps_range(date.to_time.to_i, date.next_day.to_time.to_i)
+      return response_wrapper(request, resp) { make_apps_list(resp) }
+    end
+
+    def apps_range(day1, day2)
+      request = {command: 'apps_range', type: 'apps', args: [day1, day2], processed_at: Time.now}
+      starting, ending = @helpers.check_date_validity(day1), @helpers.check_date_validity(day2)
+      return invalid_response(request) if starting == false || ending == false || starting > ending
+      resp = @db.apps_range(starting.to_time.to_i, ending.to_time.to_i)
+      return response_wrapper(request, resp) { make_apps_list(resp) }
+    end
+
+    def apps_since(day)
+      request = {command: 'apps_since', type: 'apps', args: [day], processed_at: Time.now}
+      starting = @helpers.check_date_validity(day)
+      return invalid_response(request) if starting == false
+      resp = @db.apps_since(starting.to_time.to_i)
+      return response_wrapper(request, resp) { make_apps_list(resp) }
+    end
+
+    def apps_before(day)
+      request = {command: 'apps_before', type: 'apps', args: [day], processed_at: Time.now}
+      ending = @helpers.check_date_validity(day)
+      return invalid_response(request) if ending == false
+      resp = @db.apps_before(ending.to_time.to_i)
+      return response_wrapper(request, resp) { make_apps_list(resp) }
+    end
+
+    def bundle_log(bundle)
+      request = {command: 'bundle_log', type: 'log', args: [bundle], processed_at: Time.now}
+      resp = @db.bundle_log(bundle)
+      return response_wrapper(request, resp) { make_log_objects(resp) }
+    end
+
+    def bundle_log_since(bundle, date)
+      request = {command: 'bundle_log_since', type: 'log', args: [bundle, date], processed_at: Time.now}
+      starting = @helpers.check_date_validity(date)
+      return invalid_response(request) if starting == false
+      resp = @db.bundle_log_since(bundle, starting.to_time.to_i)
+      return response_wrapper(request, resp) { make_log_objects(resp) }
+    end
+
+    def bundle_log_before(bundle, date)
+      request = {command: 'bundle_log_before', type: 'log', args: [bundle, date], processed_at: Time.now}
+      ending = @helpers.check_date_validity(date)
+      return invalid_response(request) if ending == false
+      resp = @db.bundle_log_before(bundle, ending.to_time.to_i)
+      return response_wrapper(request, resp) { make_log_objects(resp) }
+    end
+
+    def bundle_log_range(bundle, day1, day2)
+      request = {command: 'bundle_log_range', type: 'log', args: [bundle, day1, day2], processed_at: Time.now}
+      starting, ending = @helpers.check_date_validity(day1), @helpers.check_date_validity(day2)
+      return invalid_response(request) if starting == false || ending == false || starting > ending
+      resp = @db.bundle_log_range(bundle, starting.to_time.to_i, ending.to_time.to_i)
+      return response_wrapper(request, resp) { make_log_objects(resp) }
+    end
+    def bundle_log_day(bundle, day)
+      starting = @helpers.check_date_validity(day)
+      return invalid_response(request) if starting == false
+      next_day = starting.next_day.strftime('%Y-%m-%d')
+      bundle_log_range(bundle, day, next_day)
+    end
+
+    def log_since(day = nil)
+      request = {command: 'log_since', type: 'log', args: [day], processed_at: Time.now}
+      starting = if day.nil?
+        @helpers.check_date_validity(Time.now.to_s[0..9])
+      else
+        @helpers.check_date_validity(day)
+      end
+      resp = @db.log_since(starting.to_time.to_i)
+      return response_wrapper(request, resp) { make_log_objects(resp) }
+    end
+
+    def log_day(day)
+      request = {command: 'log_range', type: 'log', args: [day], processed_at: Time.now}
+      starting = @helpers.check_date_validity(day)
+      return invalid_response(request) if starting == false
+      ending = @helpers.check_date_validity(starting.next_day.strftime('%Y-%m-%d'))
+      resp = @db.log_range(starting.to_time.to_i, ending.to_time.to_i)
+      return response_wrapper(request, resp) { make_log_objects(resp) }
+    end
+
+    # ---
+
+    def delete_bundle(bundle_id)
+      request = {command: 'delete_bundle', type: 'action', args: [bundle_id], processed_at: Time.now}
+      resp = @db.delete_bundle(bundle_id)
+      if resp.empty? # success
+        resp = [{'deleted' => bundle_id}]
+        @response = make_basic_response(request, resp)
+      else
+        @response = bad_request(request)
+      end
+      return @response
+    end
+
+    # ---
+
+    def invalid_response(request)
+      @response = bad_request(request)
+    end
+
+    private
+
+    def response_wrapper(request, resp)
+      return invalid_response(request) if resp.empty?
+      list = yield
+      @response = make_basic_response(request, list)
+    end
+
+    def make_apps_list(resp)
+      result = make_apps_objects(resp)
+      result.sort_by {|obj| obj[:total][:seconds]}
+    end
+
+    def make_basic_response(request, list)
+      {
+        meta: {
+          code: 200,
+          request: request
+        },
+        data: list
+      }.to_json
+    end
+
+    def make_apps_objects(db_resp)
+      if @bundles.nil?
+        db_resp.map do |id, sec|
+          {
+            bundle: id,
+            total: {
+              seconds: sec,
+              time: @helpers.sec_to_time(sec)
+            }
+          }
+        end
+      else
+        db_resp.map do |id, sec|
+          {
+            bundle: id,
+            name: @bundles[id],
+            total: {
+              seconds: sec,
+              time: @helpers.sec_to_time(sec)
+            }
+          }
+        end
+      end
+    end
+
+    def make_log_objects(db_resp)
+      result = {}
+      sorted = db_resp.sort_by { |arr| arr[2] }
+      sorted.each do |arr|
+        name = if @bundles.nil?
+          arr[1]
+        else
+          @bundles[arr[1]]
+        end
+        result[arr[0]] = {
+          bundle: arr[1],
+          name: name,
+          start: @helpers.epoch_to_date(arr[2]),
+          duration: {
+            seconds: arr[4],
+            time: @helpers.sec_to_time(arr[4])
+          }
+        }
+      end
+      return result
+    end
+
+    def bad_request(request)
+      {
+        meta: {
+          code: 403,
+          request: request,
+          message: 'Bad request.'
+        },
+        data: []
+      }.to_json
+    end
+
+  end
+
+end