TokiCLI 0.2.1 → 0.3.0

data/lib/TokiCLI/fileops.rb

@@ -0,0 +1,187 @@
+# encoding: utf-8
+module TokiCLI
+  class FileOps
+
+    require 'fileutils'
+    require 'CFPropertyList'
+    require 'yaml'
+
+    attr_accessor :home_path, :toki_path, :db_path, :db_file, :bundles_file, :bundles, :config_file, :config_path, :config, :data_path, :files_path, :user_file, :log_path
+
+    def initialize
+      @home_path = Dir.home
+      @toki_path = "#{@home_path}/.TokiCLI"
+      @data_path = "#{@toki_path}/data"
+      @config_path = "#{@toki_path}/config"
+      @files_path = "#{@toki_path}/files"
+      @backup_path = "#{@toki_path}/backup"
+      @log_path = "#{@toki_path}/log"
+      @db_path = "#{@home_path}/Library/Containers/us.kkob.Toki/Data/Documents"
+      @db_file = "#{@db_path}/toki_data.sqlite3"
+      @bundles_file = "#{@files_path}/bundles.json"
+      @config_file = "#{@config_path}/config.yml"
+      @user_file = "#{@data_path}/user.json"
+      make_toki_dirs()
+      @bundles = load_bundles()
+      @config = create_config()
+    end
+
+    def backup_db
+      FileUtils.copy(@db_file, "#{@backup_path}/toki_data.sqlite3.bak")
+    end
+
+    def load_bundles
+      if File.exist?(@bundles_file)
+        JSON.parse(File.read(@bundles_file))
+      else
+        nil
+      end
+    end
+
+    def save_bundles
+      @bundles = get_bundle_ids()
+      File.write(@bundles_file, @bundles.to_json)
+    end
+
+    def get_bundle_from_name(name)
+      app_name = name.map {|n| n.downcase}.join(' ')
+      candidates = []
+      @bundles.each do |bundle_id, bundle_name|
+        if bundle_name.downcase =~ /#{app_name}/
+          candidates << bundle_id
+        end
+      end
+      return candidates
+    end
+
+    def export(toki, options, title = nil)
+      response = JSON.parse(toki.response)
+      type = response['meta']['request']['type']
+      prefix = response['meta']['request']['processed_at'][0..9]
+      path = if title.nil?
+        "#{@data_path}/#{prefix}_#{type}"
+      else
+        "#{@data_path}/#{prefix}_#{type}_#{title.tr_s(' ', '-')}"
+      end
+      if options[:json]
+        file = "#{path}.json"
+        File.write(file, toki.response)
+      elsif options[:csv]
+        file = "#{path}.csv"
+        if type == 'apps'
+          export_apps_csv(response, file)
+        elsif type == 'log'
+          export_log_csv(response, file)
+        end
+      else
+        abort(Status.wtf)
+      end
+      puts Status.file_saved(file)
+    end
+
+    private
+
+    def export_apps_csv(response, file)
+      CSV.open(file, "wb") do |csv|
+        csv << ['Bundle', 'Name', 'Total', 'Hours', 'Minutes', 'Seconds']
+        response['data'].each do |line|
+          csv << [line['bundle'], line['name'], line['total']['seconds'], line['total']['time']['hours'], line['total']['time']['minutes'], line['total']['time']['seconds']]
+        end
+      end
+    end
+
+    def export_log_csv(response, file)
+      CSV.open(file, "wb") do |csv|
+        csv << ['Start', 'Duration (seconds)', 'Minutes', 'Seconds', 'Sync ID']
+        response['data'].each {|line| csv << [line[1]['start'], line[1]['duration']['seconds'], line[1]['duration']['time']['minutes'], line[1]['duration']['time']['seconds'], line[0]]}
+      end
+    end
+
+    def make_toki_dirs
+      %w{backup data files config log}.each do |dir|
+        path = "#{@toki_path}/#{dir}"
+        FileUtils.mkdir_p(path) unless Dir.exist?(path)
+      end
+    end
+
+    def create_config
+      return YAML.load(File.read(@config_file)) if File.exist?(@config_file)
+      settings = {
+        'table' => {
+          'width' => 90
+        }
+      }
+      File.write(@config_file, settings.to_yaml)
+      return YAML.load(File.read(@config_file))
+    end
+
+    # Scan for names from bundle ids
+    def get_bundle_ids
+      @names = {}
+      get_bundles(get_plists("/Applications/*/Contents/*"))
+      get_bundles(get_plists("/Applications/Utilities/*/Contents/*"))
+      get_bundles(get_plists("#{@home_path}/Applications/*/Contents/*"))
+      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',
+        'com.apple.AirPlayUIAgent' => 'AirPlayUIAgent',
+        'com.apple.ScreenSaver.Engine' => 'ScreenSaver',
+        'com.adobe.PDApp' => 'Adobe Application Manager',
+        'com.screentime.flash.builder' => 'Screentime for Flash',
+        'com.apple.systemuiserver' => 'SystemUIServer',
+        'org.mozilla.plugincontainer' => 'Mozilla Plugin-container',
+        'com.adobe.ACCC.Uninstaller' => 'Adobe Creative Cloud Uninstaller',
+        'com.smileonmymac.textexpander' => 'TextExpander',
+        'org.pqrs.Karabiner-AXNotifier' => 'Karabiner Notifier',
+        'com.apple.DiskImageMounter' => 'Apple Disk Image Mounter'
+      }
+      specials.each { |k, v| @names[k] = v }
+      return @names
+    end
+
+    def get_plists path
+      Dir.glob(path).select {|f| (File.split f).last == 'Info.plist'}
+    end
+
+    def get_bundles plists
+      plists.each do |obj|
+        puts Status.analysing(obj)
+        begin
+          pl = CFPropertyList::List.new(:file => obj)
+        rescue CFFormatError, NoMethodError
+          puts Status.no_plist
+          next
+        end
+        data = CFPropertyList.native_types(pl.value)
+        name = data['CFBundleName'] || data['CFBundleExecutable']
+        next if name.nil?
+        bundle_id = data['CFBundleIdentifier']
+        next if bundle_id.nil? || bundle_id.empty?
+        @names[bundle_id] = name
+      end
+    end
+
+  end
+end

data/lib/TokiCLI/status.rb

@@ -1,41 +1,53 @@
 # 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.no_db
-      "\nOops, I can't find the Toki database. Please install Toki.app by http://app.net/keita.\n\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.specify_day
-      "\nPlease specify the date with this format: 2014-04-19\n\n"
-    end
-    def self.specify_range
-      "\nPlease specify the range with this format: 2014-04-17 2014-04-19\n\n"
-    end
-    def self.run_auth
-      "\nPlease run `toki auth` first.\n\n"
-    end
+
+  module Status
+
+    module ClassMethods
+
+      def canceled
+        "\nCanceled.\n\n"
+      end
+
+      def version(version)
+        "\n-- TokiCLI --\n\nVersion:\t#{version}\nUrl:\t\thttp://github.com/ericdke/TokiCLI\n\n"
+      end
+
+      def wtf
+        "\nAn error occurred! Goodbye, and thanks for all the fish.\n\n"
+      end
+
+      def no_plist
+        "Unable to read the file, skipping...\n"
+      end
+
+      def no_data
+        "\nNo data for this request.\n\n"
+      end
+
+      def analysing(obj)
+        "Analyzing #{obj} ...\n"
+      end
+
+      def scanning
+        "\nScanning applications bundles to find their names.\n\n"
+      end
+
+      def file_saved(path)
+        "\nFile saved in #{path}\n\n"
+      end
+
+      def next_launch_with_names
+        "Starting with next launch, TokiCLI will display apps names. Run `toki scan` again to update the apps list.\n\n"
+      end
+
+      def please_scan
+        "\nPlease run `toki scan` to populate/refresh the applications names database.\n\n"
+      end
+
+    end
+
+    extend ClassMethods # This is a way to avoid having to declare self.xxx for each def
   end
+
 end

data/lib/TokiCLI/version.rb

@@ -1,3 +1,3 @@
 module TokiCLI
-  VERSION = "0.2.1"
+  VERSION = "0.3.0"
 end

data/lib/TokiCLI/view.rb

@@ -1,132 +1,179 @@
 # encoding: utf-8
+
 module TokiCLI
+
   class View
 
-    def total_table(data)
-      table = init_table
-      clear
-      if data['data']['apps']
-        if data['data']['date']
-          table.title = "Your apps monitored by Toki: #{data['data']['date']}"
-        else
-          table.title = "Your apps monitored by Toki"
-        end
-        puts apps_3(data, table)
+    require 'terminal-table'
+
+    def initialize settings = {}
+      @settings = if settings.empty?
+        {"table"=>{"width"=>90}} # force default if no initialization
       else
-        table.title = "Your app monitored by Toki"
-        puts total_3(data, table)
+        settings
       end
-      puts "\n"
-    end
-
-    def app_table(asked, app_data)
-      table = init_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, total)
-      puts "\n"
     end
 
-    def max_30 text
-      text.length >= 30 ? "#{text[0..27]}..." : text
+    def version
+      table = init_table("TokiCLI for Toki.app")
+      table << ['Version', VERSION]
+      table << :separator
+      table << ['Infos', 'http://github.com/ericdke/TokiCLI']
+      puts "\n#{table}\n"
     end
 
-    def force_name name
-      name.nil? ? ' ' : name
+    def apps_total(api_response, title = "Your apps monitored by Toki")
+      table, apps = make_apps_table(api_response, title)
+      puts "\n"
+      display, total = populate_apps_table(table, apps)
+      display << :separator
+      display << insert_total_line(total)
+      puts "\e[H\e[2J"
+      puts display
     end
 
-    def humanized_date epoch
-      human = seconds_to_time epoch
-      "#{human[:hours]}h #{human[:minutes]}m #{human[:seconds]}s"
+    def apps(api_response, title = "Your apps monitored by Toki")
+      table, apps = make_apps_table(api_response, title)
+      puts "\n"
+      display, _ = populate_apps_table(table, apps)
+      puts "\e[H\e[2J"
+      puts display
     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}
+    def log(api_response, title = "Your app monitored by Toki")
+      display, _ = make_log_table(api_response, title)
+      puts "\nRendering the view, please wait.\n\n"
+      lines = display.render
+      puts "\e[H\e[2J"
+      puts lines
     end
 
-    def readable_time time
-      "#{'%.2d' % time['hours']}h #{'%.2d' % time['minutes']}m #{'%.2d' % time['seconds']}s"
+    def log_total(api_response, title = "Your app monitored by Toki")
+      display, total = make_log_table(api_response, title)
+      display << :separator
+      display << insert_total_line(total)
+      puts "\nRendering the view, please wait.\n\n"
+      lines = display.render
+      puts "\e[H\e[2J"
+      puts lines
     end
 
-    def clear
-      puts "\e[H\e[2J"
+    def log_activity(api_response)
+      log = JSON.parse(api_response)['data']
+      table = init_table()
+      table.headings = ['Bundle', 'Start', 'Duration', 'Sync ID']
+      lines = make_log_lines(log)
+      display, _ = populate_log_table(lines, table, true)
+      puts display
     end
 
     private
 
-    def create_table lines, table, total = 0
-      index = 0
-      length = lines.length - 1
-      lines.each do |obj|
-         table << obj
-         table << :separator unless index == length
-         index += 1
+    def init_table(title = 'TokiCLI')
+      Terminal::Table.new do |t|
+        t.style = { :width => @settings['table']['width'] }
+        t.title = title
       end
-      table << :separator
-      table << [{ :value => "Total: #{humanized_date(total)}", :colspan => 3, :alignment => :center }]
-      table
     end
 
-    def sort_lines lines
-      lines.sort_by! {|obj| obj[0]}
+    def make_log_table(api_response, title)
+      log = JSON.parse(api_response)['data']
+      table = init_table(title)
+      table.headings = ['Start', 'Duration', 'Sync ID']
+      lines = make_log_lines(log)
+      return populate_log_table(lines, table)
     end
 
-    def format_lines lines
-      lines.map! {|obj| [obj[0], obj[1], obj[2]]}
+    def make_apps_table(api_response, title)
+      apps = JSON.parse(api_response)['data']
+      table = init_table(title)
+      table.headings = ['Bundle ID', 'Name', 'Total']
+      return table, apps
     end
 
-    def calc_total lines
-      total = 0
-      lines.each {|v| total += v[3]}
-      total
+    def insert_total_line(total, text = 'Total')
+      [{ :value => "#{text}: #{readable_time(sec_to_time(total))}", :colspan => 3, :alignment => :center }]
     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
+    def make_log_lines(log)
+      log.map { |k, v| [v['start'], readable_time_log(v['duration']['time']), k, v['duration']['seconds'], v['bundle']] }
     end
 
-    def name_of app_data
-      if app_data['data']['name']
-        app_data['data']['name']
+    def populate_log_table(lines, table, with_bundle = false)
+      day = lines[0][0][0..9]
+      if with_bundle == false
+        table << [{ :value => "#{day}", :colspan => 3, :alignment => :center }]
       else
-        app_data['data']['bundle']
+        table << [{ :value => "#{day}", :colspan => 4, :alignment => :center }]
+        table << :separator
       end
+      total = 0
+      lines.each.with_index(1) do |line, index|
+        new_day = line[0][0..9]
+        unless day == new_day
+          table << :separator
+          if with_bundle == false
+            table << [{ :value => "#{new_day}", :colspan => 3, :alignment => :center }]
+          else
+            table << [{ :value => "#{new_day}", :colspan => 4, :alignment => :center }]
+          end
+          table << :separator
+        end
+        if with_bundle == false
+          table << [line[0][10..18], line[1], line[2]]
+        else
+          table << [line[4], line[0][10..18], line[1], line[2]]
+          # table << :separator unless index == lines.size
+        end
+        day = new_day
+        total += line[3]
+      end
+      return table, total
     end
 
-    def init_table
-      Terminal::Table.new do |t|
-        t.style = { :width => 100 }
+    def populate_apps_table(table, apps)
+      total = 0
+      apps.each do |app|
+        total += app['total']['seconds']
+        if app['name']
+          table << app_row_with_name(app)
+        else
+          table << app_row(app)
+        end
       end
+      return table, total
     end
 
-    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
+    def app_row_with_name(obj)
+      max = @settings['table']['width'] / 3
+      [width(max + 5, obj['bundle']), width(max - 5, obj['name']), readable_time(obj['total']['time'])]
+    end
+
+    def app_row(obj)
+      max = @settings['table']['width'] / 2
+      [width(max, obj['bundle']), '(unknown)', readable_time(obj['total']['time'])]
+    end
+
+    def width(width, text)
+      boundary = width - 3
+      text.length >= width ? "#{text[0..boundary]}..." : text
     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
+    def readable_time(obj)
+      "#{obj['hours']}h #{'%.2d' % obj['minutes']}m #{'%.2d' % obj['seconds']}s"
+    end
+
+    def readable_time_log(obj)
+      "#{'%.2d' % obj['minutes']}m #{'%.2d' % obj['seconds']}s"
+    end
+
+    def sec_to_time(secs)
+      hours = secs / 3600
+      minutes = (secs / 60 - hours * 60)
+      seconds = (secs - (minutes * 60 + hours * 3600))
+      {'hours' => hours, 'minutes' => minutes, 'seconds' => seconds}
     end
 
   end
+
 end