From 22dfd5a5ad1eb9a534d28335da8ae65475ba7a64 Mon Sep 17 00:00:00 2001
From: SoXX <soxx@fenpa.ws>
Date: Sat, 24 Aug 2024 23:42:38 +0200
Subject: [PATCH] fix(error): Refactor error handling to use errors.Is for more
 robust database error checking in Listen method.

---
 pkg/plug/server.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/pkg/plug/server.go b/pkg/plug/server.go
index d7f8890..bd3295d 100644
--- a/pkg/plug/server.go
+++ b/pkg/plug/server.go
@@ -3,6 +3,7 @@ package plug
 import (
 	"context"
 	"fmt"
+	"github.com/pkg/errors"
 	"log"
 	"net"
 
@@ -51,9 +52,10 @@ func (p *Plug) Listen() error {
 
 	source, err = database.GetSourceByDomain(p.ctx, p.source.Domain)
 	if err != nil {
-		if err.Error() == otterError.NoDataFound {
+		if errors.Is(err, &otterError.Database{Reason: otterError.NoDataFound}) {
 			log.Printf("Initalizing source!")
-			if source, err = database.CreateSource(p.ctx, p.source); err != nil {
+			source, err = database.CreateSource(p.ctx, p.source)
+			if err != nil {
 				panic(err)
 			}
 		} else {