add self update option (projectdiscovery#235)

* Add self update option (projectdiscovery#234) * Add update flags * Add go setup step in lint-test workflow
fopina · Apr 16, 2023 · 46fa75e · 46fa75e
1 parent f413054
commit 46fa75e
Show file tree

Hide file tree

Showing 8 changed files with 162 additions and 47 deletions.
diff --git a/.github/workflows/lint-test.yml b/.github/workflows/lint-test.yml
@@ -8,8 +8,14 @@ jobs:
     name: Lint Test
     runs-on: ubuntu-latest
     steps:
+      - name: Set up Go
+        uses: actions/setup-go@v3
+        with:
+          go-version: 1.19
+
       - name: Checkout code
         uses: actions/checkout@v3
+
       - name: Run golangci-lint
         uses: golangci/golangci-lint-action@v3.4.0
         with:

diff --git a/README.md b/README.md
@@ -52,24 +52,25 @@ notify -h
 
 This will display help for the tool. Here are all the switches it supports.
 
-| Flag               | Description                                        | Example                               |
-|--------------------|----------------------------------------------------|---------------------------------------|
-| `-bulk`            | enable bulk processing                             | `notify -bulk`                        |
-| `-char-limit`      | max character limit per message (default 4000)     | `notify -cl 2000`                     |
-| `-config`          | notify configuration file                          | `notify -config config.yaml`          |
-| `-data`            | input file to send for notify                      | `notify -i test.txt`                  |
-| `-delay`           | delay in seconds between each notification         | `notify -d 2`                         |
-| `-id`              | id to send the notification to (optional)          | `notify -id recon,scans`              |
-| `-msg-format`      | add custom formatting to message                   | `notify -mf Hey {{data}}`             |
-| `-no-color`        | disable colors in output                           | `notify -nc`                          |
-| `-provider-config` | provider config path                               | `notify -pc provider.yaml`            |
-| `-provider`        | provider to send the notification to (optional)    | `notify -p slack,telegram`            |
-| `-proxy`           | http proxy to use with notify                      | `notify -proxy http://127.0.0.1:8080` |
-| `-rate-limit`      | maximum number of HTTP requests to send per second | `notify -rl 1`                        |
-| `-silent`          | enable silent mode                                 | `notify -silent`                      |
-| `-verbose`         | enable verbose mode                                | `notify -verbose`                     |
-| `-version`         | display version                                    | `notify -version`                     |
-
+| Flag                    | Description                                        | Example                               |
+|-------------------------|----------------------------------------------------|---------------------------------------|
+| `-bulk`                 | enable bulk processing                             | `notify -bulk`                        |
+| `-char-limit`           | max character limit per message (default 4000)     | `notify -cl 2000`                     |
+| `-config`               | notify configuration file                          | `notify -config config.yaml`          |
+| `-data`                 | input file to send for notify                      | `notify -i test.txt`                  |
+| `-delay`                | delay in seconds between each notification         | `notify -d 2`                         |
+| `-id`                   | id to send the notification to (optional)          | `notify -id recon,scans`              |
+| `-msg-format`           | add custom formatting to message                   | `notify -mf Hey {{data}}`             |
+| `-no-color`             | disable colors in output                           | `notify -nc`                          |
+| `-provider-config`      | provider config path                               | `notify -pc provider.yaml`            |
+| `-provider`             | provider to send the notification to (optional)    | `notify -p slack,telegram`            |
+| `-proxy`                | http proxy to use with notify                      | `notify -proxy http://127.0.0.1:8080` |
+| `-rate-limit`           | maximum number of HTTP requests to send per second | `notify -rl 1`                        |
+| `-silent`               | enable silent mode                                 | `notify -silent`                      |
+| `-verbose`              | enable verbose mode                                | `notify -verbose`                     |
+| `-version`              | display version                                    | `notify -version`                     |
+| `-update`               | updates to latest version                          | `notify -update`                      |
+| `-disable-update-check` | disables automatic update check                    | `notify -duc`                         |
 
 # Notify Installation
 

diff --git a/cmd/notify/notify.go b/cmd/notify/notify.go
@@ -64,6 +64,8 @@ func readConfig() {
 	set.BoolVar(&options.Version, "version", false, "display version")
 	set.BoolVarP(&options.NoColor, "no-color", "nc", false, "disable colors in output")
 	set.StringVar(&options.Proxy, "proxy", "", "HTTP Proxy to use with notify")
+	set.CallbackVarP(runner.GetUpdateCallback(), "update", "up", "update notify to latest version")
+	set.BoolVarP(&options.DisableUpdateCheck, "disable-update-check", "duc", false, "disable automatic notify update check")
 
 	_ = set.Parse()
 

diff --git a/go.mod b/go.mod
@@ -12,46 +12,69 @@ require (
 	github.com/pkg/errors v0.9.1
 	github.com/projectdiscovery/goflags v0.1.8
 	github.com/projectdiscovery/gologger v1.1.8
-	github.com/projectdiscovery/utils v0.0.14
+	github.com/projectdiscovery/utils v0.0.15-0.20230312150239-2528d5ff811d
 	go.uber.org/multierr v1.10.0
 	go.uber.org/ratelimit v0.2.0
 	gopkg.in/yaml.v2 v2.4.0
 )
 
 require (
+	aead.dev/minisign v0.2.0 // indirect
 	github.com/Masterminds/goutils v1.1.1 // indirect
 	github.com/Masterminds/semver v1.5.0 // indirect
+	github.com/Masterminds/semver/v3 v3.2.0 // indirect
+	github.com/VividCortex/ewma v1.2.0 // indirect
+	github.com/alecthomas/chroma v0.10.0 // indirect
 	github.com/andres-erbsen/clock v0.0.0-20160526145045-9e14626cd129 // indirect
 	github.com/asaskevich/govalidator v0.0.0-20210307081110-f21760c49a8d // indirect
+	github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
 	github.com/aymerick/douceur v0.2.0 // indirect
+	github.com/charmbracelet/glamour v0.6.0 // indirect
+	github.com/cheggaaa/pb/v3 v3.1.2 // indirect
 	github.com/cnf/structhash v0.0.0-20201127153200-e1b16c1ebc08 // indirect
+	github.com/dlclark/regexp2 v1.8.1 // indirect
 	github.com/dsnet/compress v0.0.1 // indirect
-	github.com/fatih/color v1.13.0 // indirect
+	github.com/fatih/color v1.14.1 // indirect
+	github.com/golang/protobuf v1.5.2 // indirect
 	github.com/golang/snappy v0.0.3 // indirect
+	github.com/google/go-github/v30 v30.1.0 // indirect
+	github.com/google/go-querystring v1.1.0 // indirect
 	github.com/google/uuid v1.3.0 // indirect
 	github.com/gorilla/css v1.0.0 // indirect
 	github.com/huandu/xstrings v1.4.0 // indirect
 	github.com/imdario/mergo v0.3.13 // indirect
+	github.com/lucasb-eyer/go-colorful v1.2.0 // indirect
 	github.com/mattn/go-colorable v0.1.13 // indirect
-	github.com/mattn/go-isatty v0.0.16 // indirect
+	github.com/mattn/go-isatty v0.0.17 // indirect
+	github.com/mattn/go-runewidth v0.0.14 // indirect
 	github.com/mholt/archiver v3.1.1+incompatible // indirect
-	github.com/microcosm-cc/bluemonday v1.0.21 // indirect
+	github.com/microcosm-cc/bluemonday v1.0.23 // indirect
 	github.com/miekg/dns v1.1.50 // indirect
+	github.com/minio/selfupdate v0.6.0 // indirect
 	github.com/mitchellh/copystructure v1.2.0 // indirect
 	github.com/mitchellh/reflectwalk v1.0.2 // indirect
 	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
 	github.com/modern-go/reflect2 v1.0.2 // indirect
+	github.com/muesli/reflow v0.3.0 // indirect
+	github.com/muesli/termenv v0.15.1 // indirect
 	github.com/nwaples/rardecode v1.1.0 // indirect
+	github.com/olekukonko/tablewriter v0.0.5 // indirect
 	github.com/pierrec/lz4 v2.6.0+incompatible // indirect
+	github.com/rivo/uniseg v0.4.4 // indirect
 	github.com/saintfish/chardet v0.0.0-20120816061221-3af4cd4741ca // indirect
 	github.com/ulikunitz/xz v0.5.7 // indirect
 	github.com/xi2/xz v0.0.0-20171230120015-48954b6210f8 // indirect
-	golang.org/x/crypto v0.1.0 // indirect
+	github.com/yuin/goldmark v1.5.4 // indirect
+	github.com/yuin/goldmark-emoji v1.0.1 // indirect
+	golang.org/x/crypto v0.7.0 // indirect
 	golang.org/x/exp v0.0.0-20221019170559-20944726eadf // indirect
-	golang.org/x/mod v0.6.0 // indirect
-	golang.org/x/net v0.7.0 // indirect
-	golang.org/x/sys v0.5.0 // indirect
-	golang.org/x/tools v0.2.0 // indirect
+	golang.org/x/mod v0.8.0 // indirect
+	golang.org/x/net v0.8.0 // indirect
+	golang.org/x/oauth2 v0.5.0 // indirect
+	golang.org/x/sys v0.6.0 // indirect
+	golang.org/x/tools v0.6.0 // indirect
+	google.golang.org/appengine v1.6.7 // indirect
+	google.golang.org/protobuf v1.28.1 // indirect
 	gopkg.in/djherbis/times.v1 v1.3.0 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 )