Add support for Linear

wakatime · Mar 20, 2024 · c126c7d · c126c7d
1 parent a6ec69b
commit c126c7d
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 3 deletions.
diff --git a/WakaTime/Extensions/AXUIElementExtension.swift b/WakaTime/Extensions/AXUIElementExtension.swift
@@ -32,6 +32,20 @@ extension AXUIElement {
         // swiftlint:enable force_cast
     }
 
+    var nextSibling: AXUIElement? {
+        guard let parentChildren = self.parent?.children, let currentIndex = parentChildren.firstIndex(of: self) else { return nil }
+        let nextIndex = currentIndex + 1
+        guard parentChildren.indices.contains(nextIndex) else { return nil }
+        return parentChildren[nextIndex]
+    }
+
+    var previousSibling: AXUIElement? {
+        guard let parentChildren = self.parent?.children, let currentIndex = parentChildren.firstIndex(of: self) else { return nil }
+        let previousIndex = currentIndex - 1
+        guard parentChildren.indices.contains(previousIndex) else { return nil }
+        return parentChildren[previousIndex]
+    }
+
     var id: String? {
         guard let ref = getValue(for: kAXIdentifierAttribute) else { return nil }
         // swiftlint:disable force_cast
@@ -72,6 +86,10 @@ extension AXUIElement {
             case .firefox:
                 let addressField = findAddressField()
                 address = addressField?.value
+            case .linear:
+                let projectLabel = firstDescendantWhere { $0.value == "Project" }
+                let projectButton = projectLabel?.nextSibling?.firstDescendantWhere { $0.role == kAXButtonRole }
+                return projectButton?.rawTitle
             case .safari:
                 let addressField = elementById(identifier: "WEB_BROWSER_ADDRESS_AND_SEARCH_FIELD")
                 address = addressField?.value

diff --git a/WakaTime/Extensions/NSRunningApplicationExtension.swift b/WakaTime/Extensions/NSRunningApplicationExtension.swift
@@ -57,9 +57,7 @@ enum MonitoredApp: String, CaseIterable {
     ]
 
     // list apps which we aren't yet able to track, so they're hidden from the Monitored Apps menu
-    static let unsupportedAppIds = [
-        MonitoredApp.linear.rawValue,
-    ]
+    static let unsupportedAppIds = [String]()
 }
 
 extension NSRunningApplication {

diff --git a/WakaTime/Helpers/MonitoringManager.swift b/WakaTime/Helpers/MonitoringManager.swift
@@ -103,6 +103,7 @@ class MonitoringManager {
             case .linear:
                 return HeartbeatData(
                     entity: title,
+                    project: project,
                     category: .planning)
             case .notes:
                 if activeWindow.rawTitle == "Notes" {