Added rules to handle specialCharacters

grrlopes · Aug 13, 2023 · 600fc55 · 600fc55
1 parent e79b179
commit 600fc55
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/helper/parsefilter.go b/helper/parsefilter.go
@@ -3,12 +3,19 @@ package helper
 import "strings"
 
 func ParseFilter(filter string) string {
+	var addQuotes []string
+
+	filter = strings.ReplaceAll(filter, "\"", "'")
 
 	trim := strings.TrimSpace(filter)
 
 	split := strings.Split(trim, " ")
 
-	join := strings.Join(split, "* AND ")
+	for _, v := range split {
+		addQuotes = append(addQuotes, `"`+v+`"`)
+	}
+
+	join := strings.Join(addQuotes, "* AND ")
 
 	join += "* AND"
 

diff --git a/repositories/sqlite/sqliterepository.go b/repositories/sqlite/sqliterepository.go
@@ -82,7 +82,7 @@ func (sql *SQLiteRepository) Search(filter string, limit int, skip int) ([]entit
 		result   *gorm.DB
 	)
 
-	if filter == "*" {
+	if filter == `""*` {
 		result = sql.database.Limit(limit).Offset(skip).Where("cmd LIKE ?", "%"+""+"%").Find(&commands)
 	} else {
 		result = sql.database.Raw(
@@ -105,7 +105,7 @@ func (sql *SQLiteRepository) SearchCount(filter string) (int, error) {
 		err         error
 	)
 
-	if filter == "*" {
+	if filter == `""*` {
 		err = sql.database.Model(&command).Where("cmd LIKE ?", "%"+""+"%").Count(&count).Error
 		countResult = int(count)
 	} else {