extern/hishtory
1
0
Fork 0
mirror of https://github.com/ddworken/hishtory.git synced 2025-08-18 18:58:54 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add unit tests + rename method

Browse Source
This commit is contained in:
David Dworken
2023-02-13 22:26:02 -08:00
parent b6eb4da4f3
commit 162dd86893
2 changed files with 74 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
client/lib/lib.go
View File

@@ -1150,14 +1150,14 @@ func Search(ctx *context.Context, db *gorm.DB, query string, limit int) ([]*data
}
func parseNonAtomizedToken(token string) (string, interface{}, interface{}, interface{}, error) {
wildcardedToken := "%" + unescape(token) + "%"
wildcardedToken := "%" + stripBackslash(token) + "%"
return "(command LIKE ? OR hostname LIKE ? OR current_working_directory LIKE ?)", wildcardedToken, wildcardedToken, wildcardedToken, nil
}
func parseAtomizedToken(ctx *context.Context, token string) (string, interface{}, interface{}, error) {
splitToken := splitEscaped(token, ':', 2)
field := unescape(splitToken[0])
val := unescape(splitToken[1])
field := stripBackslash(splitToken[0])
val := stripBackslash(splitToken[1])
switch field {
case "user":
return "(local_username = ?)", val, nil, nil
@@ -1273,7 +1273,7 @@ func containsUnescaped(query string, token string) bool {
return false
}
func unescape(query string) string {
func stripBackslash(query string) string {
runeQuery := []rune(query)
var newQuery []rune
for i := 0; i < len(runeQuery); i++ {