DWS/hadrian
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update cli flags to load ruleset

Browse Source
This commit is contained in:
Kevin Pham
2023-12-05 21:28:53 -06:00
parent 5a7fe8a70a
commit 6192373587
4 changed files with 28 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
proxychain/ruleset/ruleset.go
View File

@@ -18,9 +18,10 @@ import (
)
type IRuleset interface {
HasRule(url url.URL) bool
GetRule(url url.URL) (rule Rule, exists bool)
YAML() (string, error)
JSON() (string, error)
HasRule(url *url.URL) bool
GetRule(url *url.URL) (rule *Rule, exists bool)
}
type Ruleset struct {
@@ -318,7 +319,7 @@ func (rs *Ruleset) loadRulesFromRemoteFile(rulesURL string) error {
// ================= utility methods ==========================
// YAML returns the ruleset as a Yaml string
func (rs *Ruleset) YAML() (string, error) {
func (rs Ruleset) YAML() (string, error) {
y, err := yaml.Marshal(rs)
if err != nil {
return "", err
@@ -333,8 +334,8 @@ func (rs *Ruleset) YAML() (string, error) {
return x, nil
}
// YAML returns the ruleset as a JSON string
func (rs *Ruleset) JSON() (string, error) {
// JSON returns the ruleset as a JSON string
func (rs Ruleset) JSON() (string, error) {
j, err := json.Marshal(rs)
if err != nil {
return "", err