make Project.Category optional and safely unwrap it

Author: scottkicks

KsApi/models/Project.swift

@@ -5,7 +5,7 @@ import ReactiveSwift
 public struct Project {
   public var availableCardTypes: [String]?
   public var blurb: String
-  public var category: Category
+  public var category: Category?
   public var country: Country
   public var creator: User
   public var extendedProjectProperties: ExtendedProjectProperties?

KsApi/models/graphql/adapters/Project+ProjectFragment.swift

@@ -25,7 +25,6 @@ extension Project {
         currency: projectFragment.currency.value
       ),
       let categoryFragment = projectFragment.category?.fragments.categoryFragment,
-      let category = Project.Category.category(from: categoryFragment),
       let dates = projectDates(from: projectFragment),
       let memberData = projectMemberData(from: projectFragment),
       let photo = projectPhoto(from: projectFragment),
@@ -34,6 +33,11 @@ extension Project {
       let creator = User.user(from: userFragment)
     else { return nil }
 
+    var category: Category?
+    if let categoryFragment = projectFragment.category?.fragments.categoryFragment {
+      category = Project.Category.category(from: categoryFragment)
+    }
+
     var location: Location?
     if let locationFragment = projectFragment.location?.fragments.locationFragment {
       location = Location.location(from: locationFragment)

KsApi/models/lenses/ProjectLenses.swift

@@ -104,7 +104,7 @@ extension Project {
       ) }
     )
 
-    public static let category = Lens<Project, Project.Category>(
+    public static let category = Lens<Project, Project.Category?>(
       view: { $0.category },
       set: { Project(
         availableCardTypes: $1.availableCardTypes, blurb: $1.blurb, category: $0, country: $1.country,
@@ -133,6 +133,47 @@ extension Project {
       ) }
     )
 
+    public static let categoryName = Lens<Project, String>(
+      view: { $0.category?.name ?? "" },
+      set: { newName, project in
+        let updatedCategory: Project.Category? = project.category.map { category in
+          Project.Category(id: category.id, name: newName)
+        }
+
+        return Project(
+          availableCardTypes: project.availableCardTypes,
+          blurb: project.blurb,
+          category: updatedCategory,
+          country: project.country,
+          creator: project.creator,
+          extendedProjectProperties: project.extendedProjectProperties,
+          memberData: project.memberData, dates: project.dates,
+          displayPrelaunch: project.displayPrelaunch, flagging: project.flagging, id: project.id,
+          lastWave: project.lastWave,
+          location: project.location,
+          name: project.name,
+          pledgeManager: project.pledgeManager,
+          pledgeOverTimeCollectionPlanChargeExplanation: project
+            .pledgeOverTimeCollectionPlanChargeExplanation,
+          pledgeOverTimeCollectionPlanChargedAsNPayments: project
+            .pledgeOverTimeCollectionPlanChargedAsNPayments,
+          pledgeOverTimeCollectionPlanShortPitch: project.pledgeOverTimeCollectionPlanShortPitch,
+          pledgeOverTimeMinimumExplanation: project.pledgeOverTimeMinimumExplanation,
+          personalization: project.personalization, photo: project.photo,
+          isInPostCampaignPledgingPhase: project.isInPostCampaignPledgingPhase,
+          postCampaignPledgingEnabled: project.postCampaignPledgingEnabled,
+          prelaunchActivated: project.prelaunchActivated,
+          redemptionPageUrl: project.redemptionPageUrl,
+          rewardData: project.rewardData,
+          sendMetaCapiEvents: project.sendMetaCapiEvents, slug: project.slug,
+          staffPick: project.staffPick, state: project.state, stats: project.stats, tags: project.tags,
+          urls: project.urls,
+          video: project.video, watchesCount: project.watchesCount,
+          isPledgeOverTimeAllowed: project.isPledgeOverTimeAllowed
+        )
+      }
+    )
+
     public static let country = Lens<Project, Country>(
       view: { $0.country },
       set: { Project(

KsApi/models/templates/ProjectTemplates.swift

@@ -76,7 +76,7 @@ extension Project {
     |> Project.lens.photo.small .~ "https://i.kickstarter.com/assets/012/224/660/847bc4da31e6863e9351bee4e55b8005_original.jpg?fit=pad&height=90&origin=ugc&q=92&width=160&sig=rOHQ6Fif6TxwI%2BL8F9RQY0wUgN%2F4yusD%2FTGXhYW8w%2FQ%3D"
     |> Project.lens.name .~ "Today"
     |> Project.lens.blurb .~ "A 24-hour timepiece beautifully designed to change the way you see your day."
-    |> \.category.name .~ "Product Design"
+    |> Project.lens.categoryName .~ "Product Design"
     |> Project.lens.stats.backersCount .~ 1_090
     |> Project.lens.stats.pledged .~ 212_870
     |> Project.lens.stats.goal .~ 24_000
@@ -87,7 +87,7 @@ extension Project {
     |> Project.lens.photo.small .~ "https://i.kickstarter.com/assets/012/347/230/2eddca8c4a06ecb69b8787b985201b92_original.jpg?fit=contain&origin=ugc&q=92&width=460&sig=ewWbTA9q%2BTNYpB9KQnwXKCfjCJum57sWhpZkp%2FiwHKY%3D"
     |> Project.lens.name .~ "Cosmic Surgery"
     |> Project.lens.blurb .~ "Cosmic Surgery is a photo book, set in the not too distant future where the world of cosmetic surgery is about to be transformed."
-    |> \.category.name .~ "Photo Books"
+    |> Project.lens.categoryName .~ "Photo Books"
     |> Project.lens.stats.backersCount .~ 329
     |> Project.lens.stats.pledged .~ 22_318
     |> Project.lens.stats.goal .~ 22_000
@@ -114,7 +114,7 @@ extension Project {
     |> Project.lens.photo.med .~ "https://i.kickstarter.com/assets/005/055/025/6e0d27710c9ae20d661e2974e99fe239_original.jpg?fit=contain&origin=ugc&q=92&width=460&sig=C05wZhm%2Fm7cw9lbn9H05zOhA8ApoQ%2Bu%2FCAO%2FuGJDMo0%3D"
     |> Project.lens.name .~ "Charlie Kaufman's Anomalisa"
     |> Project.lens.blurb .~ "From writer Charlie Kaufman (Being John Malkovich, Eternal Sunshine of the Spotless Mind) and Duke Johnson (Moral Orel, Frankenhole) comes Anomalisa."
-    |> \.category.name .~ "Animation"
+    |> Project.lens.categoryName .~ "Animation"
     |> Project.lens.stats.backersCount .~ 5_770
     |> Project.lens.stats.pledged .~ 406_237
     |> Project.lens.stats.goal .~ 200_000

Library/Tracking/ProjectAnalyticsProperties.swift

@@ -54,15 +54,15 @@ extension Project: HasProjectAnalyticsProperties {
 
 extension Project: ProjectAnalyticsProperties {
   public var categoryAnalyticsName: String? {
-    self.category.analyticsName
+    self.category?.analyticsName
   }
 
   public var categoryParentAnalyticsName: String? {
-    self.category.parentAnalyticsName
+    self.category?.parentAnalyticsName
   }
 
   public var categoryParentId: String? {
-    self.category.parentId.flatMap(String.init)
+    self.category?.parentId.flatMap(String.init)
   }
 
   public var countryCode: String {

Library/ViewModels/ActivityFriendBackingViewModel.swift

@@ -52,7 +52,7 @@ public final class ActivityFriendBackingViewModel: ActivityFriendBackingViewMode
     self.friendTitle = activity
       .map { activity in
         let stringCategoryId = (
-          activity.project?.category.parentId ?? activity.project?.category.id
+          activity.project?.category?.parentId ?? activity.project?.category?.id
         )
         .map(String.init)
 

Library/ViewModels/DiscoveryPostcardViewModel.swift

@@ -29,7 +29,7 @@ private enum PostcardMetadataType {
         iconAndTextColor: LegacyColors.ksr_create_700.uiColor()
       )
     case .featured:
-      guard let rootCategory = project.category.parentName else { return nil }
+      guard let rootCategory = project.category?.parentName else { return nil }
       return PostcardMetadataData(
         iconImage: image(named: "metadata-featured"),
         labelText: Strings.discovery_baseball_card_metadata_featured_project(
@@ -224,7 +224,7 @@ public final class DiscoveryPostcardViewModel: DiscoveryPostcardViewModelType,
       .map(fundingStatusText(forProject:))
 
     self.projectCategoryName = configuredProject
-      .map { $0.category.name }
+      .map { $0.category?.name ?? "" }
 
     self.projectCategoryViewHidden = Signal.combineLatest(
       configuredProject,
@@ -235,9 +235,11 @@ public final class DiscoveryPostcardViewModel: DiscoveryPostcardViewModelType,
         return false
       }
 
-      // if we are in a subcategory, compare categories
-      if !category.isRoot {
-        return Int(project.category.id) == category.intID
+      if let projectCategory = project.category {
+        // if we are in a subcategory, compare categories
+        if !category.isRoot {
+          return Int(projectCategory.id) == category.intID
+        }
       }
 
       // otherwise, always show category

Library/ViewModels/DiscoveryProjectCardViewModel.swift

@@ -171,7 +171,7 @@ private func projectCategoryTagShouldHide(for project: Project, in category: KsA
     return false
   }
 
-  return project.category.id == category.intID
+  return project.category?.id == category.intID
 }
 
 private func projectPWLTagShouldHide(project: Project) -> Bool {
@@ -194,10 +194,10 @@ private func projectTags(project: Project, category: KsApi.Category?) -> [Discov
     tags.append(pwlTag)
   }
 
-  if shouldShowCategoryTag {
+  if shouldShowCategoryTag, let category = project.category {
     let categoryTag = DiscoveryPillData(
       imageName: "icon--compass",
-      text: project.category.name,
+      text: category.name,
       type: .grey
     )
 

Library/ViewModels/ProjectPamphletMainCellProperties.swift

@@ -28,7 +28,7 @@ public struct ProjectPamphletMainCellProperties {
   public let displayPrelaunch: Bool?
   public let isBacking: Bool?
   public let backersCount: Int
-  public let categoryName: String
+  public let categoryName: String?
   public let locationName: String
   public let deadline: TimeInterval?
   public let fxRate: Float
@@ -58,7 +58,7 @@ public struct ProjectPamphletMainCellProperties {
     displayPrelaunch: Bool?,
     isBacking: Bool?,
     backersCount: Int,
-    categoryName: String,
+    categoryName: String?,
     locationName: String,
     deadline: TimeInterval?,
     fxRate: Float,
@@ -127,7 +127,7 @@ extension Project: HasProjectPamphletMainCellProperties {
       displayPrelaunch: self.displayPrelaunch,
       isBacking: self.personalization.isBacking,
       backersCount: self.stats.backersCount,
-      categoryName: self.category.name,
+      categoryName: self.category?.name,
       locationName: self.location?.displayableName ?? "",
       deadline: self.dates.deadline,
       fxRate: self.stats.userCurrencyRate ?? self.stats.staticUsdRate,

Library/ViewModels/ProjectPamphletMainCellViewModel.swift

@@ -219,7 +219,7 @@ public final class ProjectPamphletMainCellViewModel: ProjectPamphletMainCellView
     self.backersTitleLabelText = backersTitleAndSubtitleText.map { title, _ in title ?? "" }
     self.backersSubtitleLabelText = backersTitleAndSubtitleText.map { _, subtitle in subtitle ?? "" }
 
-    self.categoryNameLabelText = properties.map { $0.categoryName }
+    self.categoryNameLabelText = properties.map { $0.categoryName ?? "" }
 
     let deadlineTitleAndSubtitle = properties.map { properties -> (String, String) in
       var durationValue = ("", "")