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 !af.screenLocked() {
-		if err != nil {
-			panic(err)
-		}
-		if !screenLocked {
 			return
 		}
 
-		displayOn, err := af.Display.IsOn()
+		if af.displayIsOn() {
-		if err != nil {
+			af.suspendDisplay()
-			panic(err)
-		}
-		if displayOn {
-			err := af.Display.Suspend()
-			if err != nil {
-				panic(err)
-			}
 		}
 
 		for {
@@ -113,3 +102,26 @@ func (af *AfterLock) hybernateDisplayLoop() {
 		}
 	}
 }
+
+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)
+	}
+}