Add children

1 files changed, 165 insertions, 0 deletions

diff --git a/QSThingsPlugin/Handlers/ThingsTaskHandler.m b/QSThingsPlugin/Handlers/ThingsTaskHandler.m
new file mode 100644
index 0000000..5875feb
--- /dev/null
+++ b/QSThingsPlugin/Handlers/ThingsTaskHandler.m
@@ -0,0 +1,165 @@
+#import "ThingsTaskHandler.h"
+
+NS_ASSUME_NONNULL_BEGIN
+
+@implementation ThingsTaskHandler
+
+- (BOOL)objectHasChildren:(QSObject *)object {
+  return YES;
+}
+
+- (void)setQuickIconForObject:(QSObject *)object {
+  NSImage *baseImage = [QSResourceManager imageNamed:@"com.culturedcode.ThingsMac"];
+  NSString *badge = @"checkmark.circle";
+  NSColor *color = [NSColor colorWithRed:0.310 green:0.616 blue:0.976 alpha:1.0];
+  
+  if (object.primaryType == kProjectType) {
+    badge = @"circle";
+    color = [NSColor colorWithRed:0.157 green:0.376 blue:0.733 alpha:1.0];
+  }
+  NSImage *imageWithBadge = [baseImage imageOfSize:NSMakeSize(512, 512) withSystemBadge:badge ofColor:color];
+  
+  if (imageWithBadge) {
+    [object setIcon:imageWithBadge];
+  } else {
+    [object setIcon:baseImage];
+  }
+}
+
+- (BOOL)loadChildrenForObject:(QSObject *)object {
+  NSDictionary *contents = [object objectForType:object.primaryType];
+  
+  if (!contents) {
+    NSLog(@"Malformed Things Task: No content.");
+    return NO;
+  }
+
+  NSString *path = [contents objectForKey:@"path"];
+  if (!path) {
+    NSLog(@"Malformed Things Task: No path.");
+    return NO;
+  }
+  
+  NSString *uuid = [contents objectForKey:@"uuid"];
+  if (!uuid) {
+    NSLog(@"Malformed Things Task: No uuid.");
+    return NO;
+  }
+  
+  // Tasks and Projects are basically the same thing in the Things database. So we use the same handler.
+  // The big difference is that projects can't have projects, and tasks can't have tasks
+
+  NSMutableArray *children = [NSMutableArray array];
+  [children addObjectsFromArray:[self tagsForTaskIdentifiedBy:uuid usingDatabaseAt:path ]];
+  [children addObjectsFromArray:[self areasForTaskIdentifiedBy:uuid usingDatabaseAt:path ]];
+
+  if (object.primaryType == kProjectType) {
+    [children addObjectsFromArray:[self tasksForTaskIdentifiedBy:uuid usingDatabaseAt:path ]];
+  } else {
+    [children addObjectsFromArray:[self projectsForTaskIdentifiedBy:uuid usingDatabaseAt:path ]];
+  }
+
+  [object setChildren:children];
+  return YES;
+}
+
+- (NSArray *)tagsForTaskIdentifiedBy:(NSString *)uuid usingDatabaseAt:(NSString *)path {
+  NSString *query = [NSString stringWithFormat: @"SELECT "
+            "tag.uuid as uuid, "
+            "tag.title as title "
+            "FROM TMTask AS task "
+            "LEFT JOIN TMTaskTag as junction "
+            "ON junction.tasks = task.uuid "
+            "LEFT JOIN TMTag as tag "
+            "ON junction.tags = tag.uuid "
+            "WHERE task.uuid = '%@' "
+            "AND tag.uuid IS NOT NULL "
+            "ORDER BY tag.title;", uuid];
+  
+  NSArray *results = [SQLHelper executeSql:query onFile:path];
+  NSMutableArray *objects = [NSMutableArray array];
+  [results enumerateObjectsUsingBlock:^(NSDictionary *dictionary, NSUInteger i, BOOL *stop) {
+    [objects addObject:[ObjectHelper tagFromSQLResult:dictionary usingPath:path ]];
+  }];
+  
+  return objects;
+}
+
+- (NSArray *)projectsForTaskIdentifiedBy:(NSString *)uuid usingDatabaseAt:(NSString *)path {
+  NSString *query = [NSString stringWithFormat: @"SELECT "
+            "project.uuid as uuid, "
+            "project.creationDate as creationDate, "
+            "project.status as status, "
+            "project.title as title, "
+            "project.notes as notes, "
+            "project.area as area "
+            "FROM TMTask AS task "
+            "LEFT JOIN TMTask as project "
+            "ON task.project = project.uuid "
+            "AND project.type = 1 "
+            "WHERE task.uuid = '%@' "
+            "AND project.uuid IS NOT NULL "
+            "AND project.trashed = 0 "
+            "AND project.status = 0 "
+            "ORDER BY project.title;", uuid];
+  
+  NSArray *results = [SQLHelper executeSql:query onFile:path];
+  NSMutableArray *objects = [NSMutableArray array];
+  [results enumerateObjectsUsingBlock:^(NSDictionary *dictionary, NSUInteger i, BOOL *stop) {
+    [objects addObject:[ObjectHelper projectFromSQLResult:dictionary usingPath:path ]];
+  }];
+  
+  return objects;
+}
+
+- (NSArray *)areasForTaskIdentifiedBy:(NSString *)uuid usingDatabaseAt:(NSString *)path {
+  NSString *query = [NSString stringWithFormat: @"SELECT "
+            "area.uuid as uuid, "
+            "area.title as title "
+            "FROM TMTask AS task "
+            "LEFT JOIN TMTask as project "
+            "ON task.project = project.uuid "
+            "LEFT JOIN TMArea as area "
+            "ON task.area = area.uuid OR project.area = area.uuid "
+            "WHERE task.uuid = '%@' "
+            "AND area.uuid IS NOT NULL "
+            "ORDER BY area.title;", uuid];
+
+  NSArray *results = [SQLHelper executeSql:query onFile:path];
+  NSMutableArray *objects = [NSMutableArray array];
+  if (results) {
+    [results enumerateObjectsUsingBlock:^(NSDictionary *dictionary, NSUInteger i, BOOL *stop) {
+      [objects addObject:[ObjectHelper areaFromSQLResult:dictionary usingPath:path ]];
+    }];
+  }
+  
+  return objects;
+}
+
+- (NSArray *)tasksForTaskIdentifiedBy:(NSString *)uuid usingDatabaseAt:(NSString *)path {
+  NSString *query = [NSString stringWithFormat: @"SELECT "
+            "uuid, "
+            "creationDate, "
+            "status, "
+            "title, "
+            "notes, "
+            "area "
+            "FROM TMTask "
+            "WHERE project = '%@' "
+            "AND type = 0 "
+            "AND trashed = 0 "
+            "AND status = 0 "
+            "ORDER BY title;", uuid];
+  
+  NSArray *results = [SQLHelper executeSql:query onFile:path];
+  NSMutableArray *objects = [NSMutableArray array];
+  [results enumerateObjectsUsingBlock:^(NSDictionary *dictionary, NSUInteger i, BOOL *stop) {
+    [objects addObject:[ObjectHelper taskFromSQLResult:dictionary usingPath:path ]];
+  }];
+  
+  return objects;
+}
+
+@end
+
+NS_ASSUME_NONNULL_END