time_buffer 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: fb5a16f7fd58ccfb39b6d84410ec6c4e1326f119024189f652674dacf590170f
+  data.tar.gz: 5cc84eaaff62ae9741a9008e4ed5b8193d8f40f8d8e110c1d2b40c8f87152e08
+SHA512:
+  metadata.gz: 560ef5753cb3eb51bd1464df41dfa3d230a2f4676919af9d3ebb737d69c267bcd60adaeecf0b1477771fd5c34dc3f984c69592e8958bed8f42b51fc09ecb3dd8
+  data.tar.gz: '003745089e2980687ca331cebe91a3447c28d4e2262535a959c3fbc5a11d383ecac988e249a4da1634d7a0279a2088cf815ec270e18ffc70e902fec0a4fc47b6'

data/.rspec

@@ -0,0 +1,3 @@
+--format documentation
+--color
+--require spec_helper

data/.solargraph.yml

@@ -0,0 +1 @@
+reporters: []

data/.standard.yml

@@ -0,0 +1,3 @@
+# For available configuration options, see:
+#   https://github.com/standardrb/standard
+ruby_version: 3.0

data/CHANGELOG.md

@@ -0,0 +1,5 @@
+## [Unreleased]
+
+## [0.1.0] - 2024-08-10
+
+- Initial release

data/README.md

@@ -0,0 +1,35 @@
+# TimeBuffer
+
+A gem that tracks your time.
+
+## Installation
+
+Install the gem and add to the application's Gemfile by executing:
+
+    $ bundle add time_buffer
+
+If bundler is not being used to manage dependencies, install the gem by executing:
+
+    $ gem install time_buffer
+
+## Usage
+
+In IRB:
+
+```rb
+require "time_buffer"
+
+TimeBuffer::Tracker.start
+```
+
+This will initialize the SQLite database and start the tracker.
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/StevenElberger/time_buffer.

data/Rakefile

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+require "standard/rake"
+
+task default: %i[spec standard]

data/lib/time_buffer/application_handler.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module TimeBuffer
+  class ApplicationHandler
+    def initialize(app_data)
+      @app_data = app_data
+    end
+
+    def insert_application
+      TimeBuffer::DatabaseConnector.new.execute(
+        "INSERT OR IGNORE INTO applications (bundle_id, app_name) VALUES (?, ?)",
+        [app_data.bundle_id, app_data.name]
+      )
+    end
+
+    private
+
+    attr_reader :app_data
+  end
+end

data/lib/time_buffer/database_connector.rb

@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+module TimeBuffer
+  class DatabaseConnector
+    FILE_LOCATION = "usage_data.db"
+
+    def initialize
+      return if File.exist?(FILE_LOCATION)
+
+      # 1. Applications Table
+      connection.execute <<-SQL
+        CREATE TABLE IF NOT EXISTS applications (
+          id INTEGER PRIMARY KEY AUTOINCREMENT,
+          bundle_id TEXT UNIQUE NOT NULL,
+          app_name TEXT NOT NULL
+        );
+      SQL
+
+      # 2. Time Sessions Table
+      connection.execute <<-SQL
+        CREATE TABLE IF NOT EXISTS time_sessions (
+          id INTEGER PRIMARY KEY AUTOINCREMENT,
+          application_id INTEGER NOT NULL,
+          start_time DATETIME NOT NULL,
+          end_time DATETIME,
+          metadata JSONB,
+          FOREIGN KEY (application_id) REFERENCES applications(id)
+        );
+      SQL
+
+      # 3. Daily Summaries Table
+      connection.execute <<-SQL
+        CREATE TABLE IF NOT EXISTS daily_summaries (
+          id INTEGER PRIMARY KEY AUTOINCREMENT,
+          application_id INTEGER NOT NULL,
+          date DATE NOT NULL,
+          total_duration INTEGER NOT NULL,
+          FOREIGN KEY (application_id) REFERENCES applications(id),
+          UNIQUE(application_id, date)
+        );
+      SQL
+
+      puts "Database initialized successfully."
+    end
+
+    def connect
+      @db = SQLite3::Database.new(FILE_LOCATION)
+    end
+
+    def connection
+      @db || connect
+    end
+
+    def execute(sql, *params)
+      # puts "Writing to db: #{params}"
+      connection.execute(sql, *params)
+    end
+  end
+end

data/lib/time_buffer/osa_script.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module TimeBuffer
+  class OsaScript
+    class << self
+      def app_data
+        current_app_bundle_id = `osascript -e 'id of application (path to frontmost application as text)'`.strip
+        app = OsaScriptAppBuilder.build(current_app_bundle_id)
+        puts "app name: #{app.name} | bundle id: #{app.bundle_id} | class: #{app.class}"
+        app
+      end
+    end
+  end
+end

data/lib/time_buffer/osa_script_app.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module TimeBuffer
+  class OsaScriptApp
+    attr_reader :bundle_id
+
+    def initialize(app_bundle_id)
+      @bundle_id = app_bundle_id
+    end
+
+    def name
+      @name ||= `osascript -e 'tell application "System Events" to get name of first application process whose frontmost is true'`.strip
+    end
+
+    def metadata = {}
+  end
+end

data/lib/time_buffer/osa_script_app_browser.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require_relative "osa_script_app_builder"
+
+module TimeBuffer
+  class OsaScriptAppBrowser < OsaScriptApp
+    KNOWN_BROWSERS = %w[com.google.Chrome]
+
+    def self.register_known_browsers
+      KNOWN_BROWSERS.each do |bundle_id|
+        OsaScriptAppBuilder.register(app_bundle_id: bundle_id, app_class: self)
+      end
+    end
+
+    register_known_browsers
+
+    def initialize(app_bundle_id)
+      @bundle_id = app_bundle_id
+    end
+
+    def metadata
+      {
+        tab_title: tab_title
+      }
+    end
+
+    def tab_title
+      `osascript -e 'tell application "Google Chrome" to get title of active tab of front window'`.strip
+    end
+  end
+end

data/lib/time_buffer/osa_script_app_builder.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module TimeBuffer
+  class OsaScriptAppBuilder
+    class << self
+      REGISTRY = {}
+
+      def register(app_bundle_id:, app_class:)
+        REGISTRY[app_bundle_id] = app_class
+      end
+
+      def build(app_bundle_id)
+        app_class = REGISTRY[app_bundle_id]
+
+        if app_class.nil?
+          OsaScriptApp.new(app_bundle_id)
+        else
+          app_class.new(app_bundle_id)
+        end
+      end
+    end
+  end
+end

data/lib/time_buffer/session_handler.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+require "json"
+
+module TimeBuffer
+  class SessionHandler
+    def initialize(start_time:, end_time:)
+      @start_time = start_time
+      @end_time = end_time
+    end
+
+    def insert_time_session(app_data)
+      application_id = TimeBuffer::DatabaseConnector.new.execute(
+        "SELECT id FROM applications WHERE bundle_id = ?",
+        [app_data.bundle_id]
+      ).first&.first
+
+      puts "application ID: #{application_id}"
+
+      TimeBuffer::DatabaseConnector.new.execute(
+        "INSERT INTO time_sessions (application_id, start_time, end_time, metadata) VALUES (?, ?, ?, ?);",
+        [application_id, start_time.strftime("%Y-%m-%d %H:%M:%S"), end_time.strftime("%Y-%m-%d %H:%M:%S"), app_data.metadata.to_json]
+      )
+    end
+
+    private
+
+    attr_reader :start_time, :end_time
+  end
+end

data/lib/time_buffer/state_detector.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+module TimeBuffer
+  class StateDetector
+    attr_reader :current_app_data, :last_interaction_at, :previous_app, :previous_metadata
+
+    def initialize
+      @previous_app = nil
+      @previous_metadata = {}
+      @last_interaction_at = Time.now
+      @current_app_data = nil
+    end
+
+    def app_changed?
+      @current_app_data = OsaScript.app_data
+      @current_app_data.name != @previous_app
+    end
+
+    def metadata_changed?
+      metadata = current_app_data.metadata
+      metadata && metadata != previous_metadata && current_app_data.name == previous_app
+    end
+
+    def update_last_interaction_time(now)
+      @last_interaction_at = now
+      update_state
+    end
+
+    private
+
+    def update_state
+      @previous_app = current_app_data.name
+      @previous_metadata = current_app_data.metadata
+    end
+  end
+end

data/lib/time_buffer/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module TimeBuffer
+  VERSION = "0.1.0"
+end

data/lib/time_buffer.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+require_relative "time_buffer/application_handler"
+require_relative "time_buffer/database_connector"
+require_relative "time_buffer/osa_script"
+require_relative "time_buffer/osa_script_app"
+require_relative "time_buffer/osa_script_app_browser"
+require_relative "time_buffer/osa_script_app_builder"
+require_relative "time_buffer/session_handler"
+require_relative "time_buffer/state_detector"
+require_relative "time_buffer/version"
+require "sqlite3"
+module TimeBuffer
+  class Tracker
+    def initialize
+      @state_detector = StateDetector.new
+    end
+
+    def start
+      loop do
+        now = Time.now
+
+        if state_detector.app_changed?
+          handle_app_change(state_detector.current_app_data, now)
+        elsif state_detector.metadata_changed?
+          handle_metadata_change(state_detector.current_app_data, now)
+        end
+
+        sleep 1
+      end
+    end
+
+    private
+
+    attr_reader :state_detector
+
+    def handle_app_change(app_data, now)
+      ApplicationHandler.new(app_data).insert_application
+      SessionHandler.new(start_time: state_detector.last_interaction_at, end_time: now).insert_time_session(app_data)
+      state_detector.update_last_interaction_time(now)
+    end
+
+    def handle_metadata_change(app_data, now)
+      SessionHandler.new(start_time: state_detector.last_interaction_at, end_time: now).insert_time_session(app_data)
+      state_detector.update_last_interaction_time(now)
+    end
+  end
+end

data/sig/time_buffer.rbs

@@ -0,0 +1,4 @@
+module TimeBuffer
+  VERSION: String
+  # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+end

metadata

@@ -0,0 +1,77 @@
+--- !ruby/object:Gem::Specification
+name: time_buffer
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Steven Elberger
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2024-08-21 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+description:
+email:
+- stevenelberger@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".rspec"
+- ".solargraph.yml"
+- ".standard.yml"
+- CHANGELOG.md
+- README.md
+- Rakefile
+- lib/time_buffer.rb
+- lib/time_buffer/application_handler.rb
+- lib/time_buffer/database_connector.rb
+- lib/time_buffer/osa_script.rb
+- lib/time_buffer/osa_script_app.rb
+- lib/time_buffer/osa_script_app_browser.rb
+- lib/time_buffer/osa_script_app_builder.rb
+- lib/time_buffer/session_handler.rb
+- lib/time_buffer/state_detector.rb
+- lib/time_buffer/version.rb
+- sig/time_buffer.rbs
+homepage: https://github.com/StevenElberger/time_buffer
+licenses: []
+metadata:
+  allowed_push_host: https://rubygems.org
+  homepage_uri: https://github.com/StevenElberger/time_buffer
+  source_code_uri: https://github.com/StevenElberger/time_buffer
+  changelog_uri: https://github.com/StevenElberger/time_buffer/blob/main/CHANGELOG.md
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.0.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.5.17
+signing_key:
+specification_version: 4
+summary: Tracks time spent on all applications
+test_files: []