fixed crash of application when UserDefaults data does not exist at i…

…nitial startup
zaru · Dec 2, 2016 · 43b4350 · 43b4350
1 parent 90b71f7
commit 43b4350
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/LightningQR/AppDelegate.swift b/LightningQR/AppDelegate.swift
@@ -18,7 +18,7 @@ class AppDelegate: NSObject, NSApplicationDelegate, Validator {
     var changeCount: Int = 0
 
     func applicationDidFinishLaunching(_ aNotification: Notification) {
-
+        
         addingToLoginItems()
 
         if let button = statusItem.button {

diff --git a/LightningQR/QrcodeViewController.swift b/LightningQR/QrcodeViewController.swift
@@ -17,9 +17,12 @@ class QrcodeViewController: NSViewController {
 
     override func viewDidAppear() {
         let ud = UserDefaults.standard
-        let url:String = ud.object(forKey: "url") as! String!
-        let qrcode = Qrcode().generateQR(url: url)
-        imgQrcode.image = qrcode
+
+        let url = ud.object(forKey: "url")
+        if (url != nil) {
+            let qrcode = Qrcode().generateQR(url: url as! String)
+            imgQrcode.image = qrcode
+        }
     }
 
     @IBAction func terminate(_ sender: NSButton) {