From 14621e7a1566edbe6f85e9fd580da9cd1c584a34 Mon Sep 17 00:00:00 2001
From: Hillel Coren <hillelcoren@gmail.com>
Date: Tue, 3 Mar 2020 18:44:01 +0200
Subject: [PATCH] Designs

---
 lib/constants.dart                |  2 ++
 lib/data/models/design_model.dart |  3 +++
 lib/ui/app/main_screen.dart       | 12 ++++++++++++
 stubs/data/models/stub_model      |  3 +++
 4 files changed, 20 insertions(+)

diff --git a/lib/constants.dart b/lib/constants.dart
index e8b69bd27..e0ffad6fe 100644
--- a/lib/constants.dart
+++ b/lib/constants.dart
@@ -204,6 +204,8 @@ const String kSettingsGroupSettingsView = 'group_settings_view';
 const String kSettingsGroupSettingsEdit = 'group_settings_edit';
 const String kSettingsCustomFields = 'custom_fields';
 const String kSettingsCustomDesigns = 'custom_designs';
+const String kSettingsCustomDesignsView = 'custom_designs_view';
+const String kSettingsCustomDesignsEdit = 'custom_designs_edit';
 const String kSettingsGeneratedNumbers = 'generated_numbers';
 const String kSettingsWorkflowSettings = 'workflow_settings';
 const String kSettingsInvoiceDesign = 'invoice_design';
diff --git a/lib/data/models/design_model.dart b/lib/data/models/design_model.dart
index 7d2814bd3..f27dcc2e4 100644
--- a/lib/data/models/design_model.dart
+++ b/lib/data/models/design_model.dart
@@ -51,6 +51,9 @@ abstract class DesignEntity extends Object
 
   DesignEntity._();
 
+  @override
+  EntityType get entityType => EntityType.design;
+
   String get name;
 
   String get displayName => name;
diff --git a/lib/ui/app/main_screen.dart b/lib/ui/app/main_screen.dart
index b7716bde6..10ab8b786 100644
--- a/lib/ui/app/main_screen.dart
+++ b/lib/ui/app/main_screen.dart
@@ -16,6 +16,9 @@ import 'package:invoiceninja_flutter/ui/credit/credit_screen.dart';
 import 'package:invoiceninja_flutter/ui/credit/credit_screen_vm.dart';
 import 'package:invoiceninja_flutter/ui/credit/edit/credit_edit_vm.dart';
 import 'package:invoiceninja_flutter/ui/credit/view/credit_view_vm.dart';
+import 'package:invoiceninja_flutter/ui/design/design_screen_vm.dart';
+import 'package:invoiceninja_flutter/ui/design/edit/design_edit_vm.dart';
+import 'package:invoiceninja_flutter/ui/design/view/design_view_vm.dart';
 import 'package:invoiceninja_flutter/ui/payment/refund/payment_refund_vm.dart';
 import 'package:invoiceninja_flutter/ui/reports/reports_screen.dart';
 import 'package:invoiceninja_flutter/ui/reports/reports_screen_vm.dart';
@@ -331,6 +334,15 @@ class SettingsScreens extends StatelessWidget {
       case kSettingsUserManagementEdit:
         screen = UserEditScreen();
         break;
+      case kSettingsCustomDesigns:
+        screen = DesignScreenBuilder();
+        break;
+      case kSettingsCustomDesignsView:
+        screen = DesignViewScreen();
+        break;
+      case kSettingsCustomDesignsEdit:
+        screen = DesignEditScreen();
+        break;
       case kSettingsAccountManagement:
         screen = AccountManagementScreen();
         break;
diff --git a/stubs/data/models/stub_model b/stubs/data/models/stub_model
index 0793c1ffd..22fb19fab 100644
--- a/stubs/data/models/stub_model
+++ b/stubs/data/models/stub_model
@@ -51,6 +51,9 @@ abstract class StubEntity extends Object with BaseEntity implements Built<StubEn
 
   StubEntity._();
 
+  @override
+  EntityType get entityType => EntityType.stub;
+
   String get displayName {
     // STARTER: display name - do not remove comment
   }