Refactor error handling for cleaner code

afterlock/afterlock.go

@@ -84,23 +84,12 @@ func (af *AfterLock) detectKeypresses() {
 
 func (af *AfterLock) hybernateDisplayLoop() {
 	for {
-		screenLocked, err := af.LockScreen.IsActive()
-		if err != nil {
-			panic(err)
-		}
-		if !screenLocked {
+		if !af.screenLocked() {
 			return
 		}
 
-		displayOn, err := af.Display.IsOn()
-		if err != nil {
-			panic(err)
-		}
-		if displayOn {
-			err := af.Display.Suspend()
-			if err != nil {
-				panic(err)
-			}
+		if af.displayIsOn() {
+			af.suspendDisplay()
 		}
 
 		for {
@@ -112,4 +101,27 @@ func (af *AfterLock) hybernateDisplayLoop() {
 			break
 		}
 	}
+}
+
+func (af *AfterLock) screenLocked() bool {
+	screenLocked, err := af.LockScreen.IsActive()
+	if err != nil {
+		panic(err)
+	}
+	return screenLocked
+}
+
+func (af *AfterLock) displayIsOn() bool {
+	displayOn, err := af.Display.IsOn()
+	if err != nil {
+		panic(err)
+	}
+	return displayOn
+}
+
+func (af *AfterLock) suspendDisplay() {
+	err := af.Display.Suspend()
+	if err != nil {
+		panic(err)
+	}
 }