57 lines
911 B
Go
57 lines
911 B
Go
|
package cmd
|
||
|
|
|
||
|
|
import (
|
||
|
|
"dehasher/internal/sqlite"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Table int64
|
||
|
|
|
||
|
|
const (
|
||
|
|
ResultsTable Table = iota
|
||
|
|
RunsTable
|
||
|
|
CredsTable
|
||
|
|
WhoIsTable
|
||
|
|
SubdomainsTable
|
||
|
|
HistoryTable
|
||
|
|
UnknownTable
|
||
|
|
)
|
||
|
|
|
||
|
|
func GetTable(userInput string) Table {
|
||
|
|
switch strings.ToLower(userInput) {
|
||
|
|
case "results":
|
||
|
|
return ResultsTable
|
||
|
|
case "runs":
|
||
|
|
return RunsTable
|
||
|
|
case "creds":
|
||
|
|
return CredsTable
|
||
|
|
case "whois":
|
||
|
|
return WhoIsTable
|
||
|
|
case "subdomains":
|
||
|
|
return SubdomainsTable
|
||
|
|
case "history":
|
||
|
|
return HistoryTable
|
||
|
|
default:
|
||
|
|
return UnknownTable
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t Table) Object() interface{} {
|
||
|
|
switch t {
|
||
|
|
case ResultsTable:
|
||
|
|
return sqlite.Result{}
|
||
|
|
case RunsTable:
|
||
|
|
return sqlite.QueryOptions{}
|
||
|
|
case CredsTable:
|
||
|
|
return sqlite.Creds{}
|
||
|
|
case WhoIsTable:
|
||
|
|
return sqlite.WhoisRecord{}
|
||
|
|
case SubdomainsTable:
|
||
|
|
return sqlite.SubdomainRecord{}
|
||
|
|
case HistoryTable:
|
||
|
|
return sqlite.HistoryRecord{}
|
||
|
|
default:
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
}
|