From 408d1b2418fad5aac9ce30b53b3ea2d4527f8f20 Mon Sep 17 00:00:00 2001
From: Shawn Maddock <shawn@maddock.com>
Date: Wed, 25 Sep 2019 15:04:16 -0600
Subject: [PATCH] Added button titles to LaunchAgent for customization

---
 README.md                                     | 30 ++++++++++++++++++-
 .../Application Support/umad/Resources/umad   | 18 +++++++++++
 .../LaunchAgents/com.erikng.umad.plist        | 10 +++++++
 3 files changed, 57 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index f84852b8..c6dc6a90 100755
--- a/README.md
+++ b/README.md
@@ -39,7 +39,7 @@ The following operating system and versions have been tested.
 ## Getting started
 To start, you can use the default settings in `/Library/LaunchAgent/com.anothertoolappleshouldhaveprovided.umad.plist`
 
-Essentially every component of the UI is customizable, using the above LaunchAgent.  
+Essentially every component of the UI is customizable, using the above LaunchAgent.
 * Create your .pkg with munki-pkg and install on your target workstation.
 * Open terminal.
 <i>example</i>
@@ -351,6 +351,34 @@ You can customize the uamdm screenshot path. Option 2, just replace the included
 <string>/Some/Custom/Path/uamdm_ss.png</string>
 ```
 
+### Manual Enrollment button
+This is the label for the button that loads `manualenrollmenturl`.
+```xml
+<string>--manualenrollment</string>
+<string>Manual Enrollment</string>
+```
+
+### More Info button
+This is the label for the button that loads `moreinfourl`.
+```xml
+<string>--moreinfo</string>
+<string>More Info</string>
+```
+
+### Open System Preferences button
+This is the label for the button that launches the Profiles preference pane.
+```xml
+<string>--sysprefs</string>
+<string>Open System Preferences</string>
+```
+
+### I Understand button
+This is the label for the button that users will be required to click to close the UI after `cutoffdatewarning` days have passed.
+```xml
+<string>--understand</string>
+<string>I understand.</string>
+```
+
 ## Tips, Tricks, and Troubleshooting
 
 * <b><i>I made changes to the default LaunchAgent and now the UI isn't appearing?</b></i>
diff --git a/payload/Library/Application Support/umad/Resources/umad b/payload/Library/Application Support/umad/Resources/umad
index 173c6780..2825f5bc 100755
--- a/payload/Library/Application Support/umad/Resources/umad	
+++ b/payload/Library/Application Support/umad/Resources/umad	
@@ -329,6 +329,9 @@ def get_parsed_options():
     o.add_option('--logopath',
                  default='company_logo.png',
                  help=('Optional: Path to company logo.'))
+    o.add_option('--manualenrollment',
+                 default='Manual Enrollment',
+                 help=('Required: Manual enrollment button text'))
     o.add_option('--manualenrollmenturl',
                  default='https://apple.com',
                  help=('Required: Manual Enrollment URL.'))
@@ -341,6 +344,9 @@ def get_parsed_options():
     o.add_option('--manualenrollh2text',
                  default='Click on the Manual Enrollment button below.',
                  help=('Optional: Manual enrollment text.'))
+    o.add_option('--moreinfo',
+                 default='More Info',
+                 help=('Required: More info button text'))
     o.add_option('--moreinfourl',
                  default='https://google.com',
                  help=('Required: More info URL.'))
@@ -370,6 +376,9 @@ def get_parsed_options():
     o.add_option('--subtitletext',
                  default='A friendly reminder from your local IT team',
                  help=('Required: Sub-title text.'))
+    o.add_option('--sysprefs',
+                 default='Open System Preferences',
+                 help=('Required: System preferences button text'))
     o.add_option('--sysprefsh1text',
                  default='Want this box to go away?',
                  help=('Required: Sys Prefs header 1 text.'))
@@ -414,6 +423,9 @@ def get_parsed_options():
     o.add_option('--uasspath',
                  default='uamdm_ss.png',
                  help=('Optional: Path to User Accepted MDM screenshot.'))
+    o.add_option('--understand',
+                 default='I understand.',
+                 help=('Required: Understand button text'))
 
     return o.parse_args()
 
@@ -745,6 +757,12 @@ def main():
     umad.views['field.depfailuresubtext'].setStringValue_(
         opts.depfailuresubtext.decode('utf8'))
 
+    # Setup the UI buttons
+    umad.views['button.manualenrollment'].setTitle_(opts.manualenrollment.decode('utf8'))
+    umad.views['button.moreinfo'].setTitle_(opts.moreinfo.decode('utf8'))
+    umad.views['button.sysprefs'].setTitle_(opts.sysprefs.decode('utf8'))
+    umad.views['button.understand'].setTitle_(opts.understand.decode('utf8'))
+
     # Dynamically set username and serialnumber
     umad.views['field.username'].setStringValue_(str(user_name))
     umad.views['field.serialnumber'].setStringValue_(str(get_serial()))
diff --git a/payload/Library/LaunchAgents/com.erikng.umad.plist b/payload/Library/LaunchAgents/com.erikng.umad.plist
index cabdbf43..04a31e8e 100755
--- a/payload/Library/LaunchAgents/com.erikng.umad.plist
+++ b/payload/Library/LaunchAgents/com.erikng.umad.plist
@@ -60,6 +60,16 @@
 		<!--<string>--uamdmparagraph3</string> -->
 		<!--<string>Please go to System Preferences -> Profiles, click on the Device Enrollment profile and click on the approve button.</string> -->
 		<!-- -->
+		<!-- Buttons -->
+		<!--<string>--manualenrollment</string> -->
+		<!--<string>Manual Enrollment</string> -->
+		<!--<string>--moreinfo</string> -->
+		<!--<string>More Info</string> -->
+		<!--<string>--sysprefs</string> -->
+		<!--<string>Open System Preferences</string> -->
+		<!--<string>--understand</string> -->
+		<!--<string>I understand.</string> -->
+		<!-- -->
 		<!-- URLs -->
 		<!--<string>--manualenrollmenturl</string> -->
 		<!--<string>https://apple.com</string> -->