duet/api: add some convenience methods

gertrude-app · Nov 7, 2023 · 3e3d352 · 3e3d352
1 parent 651c374
commit 3e3d352
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/api/Sources/Api/Models/Models+Duet.swift b/api/Sources/Api/Models/Models+Duet.swift
@@ -13,6 +13,15 @@ extension Model {
     try await Current.db.create(self)
   }
 
+  @discardableResult
+  func upsert() async throws -> Self {
+    if (try? await Current.db.query(Self.self).byId(id).first()) != nil {
+      return try await create()
+    } else {
+      return try await save()
+    }
+  }
+
   func delete() async throws {
     try await Current.db.query(Self.self).byId(id).delete()
   }

diff --git a/duet/Sources/Duet/Relation.swift b/duet/Sources/Duet/Relation.swift
@@ -131,6 +131,15 @@ public enum OptionalChild<C: Duet.Identifiable> {
       return loaded
     }
   }
+
+  public mutating func useLoaded(or load: () async throws -> C?) async throws -> C? {
+    guard case .loaded(let loaded) = self else {
+      let optionalChild = try await load()
+      self = .loaded(optionalChild)
+      return optionalChild
+    }
+    return loaded
+  }
 }
 
 public func connect<P: Duet.Identifiable, C: Duet.Identifiable>(