From 57160a279e07bc3b1739f999d42caebf004e30f6 Mon Sep 17 00:00:00 2001
From: Kenneth Shaw <kenneth.shaw@knq.io>
Date: Sat, 17 Jun 2017 07:56:52 +0700
Subject: [PATCH] Minor changes to chromedp-gen/build.sh

---
 cmd/chromedp-gen/.gitignore         |  1 +
 cmd/chromedp-gen/build.sh           | 24 ++++++++++++++++++++----
 cmd/chromedp-gen/internal/domain.go |  3 +++
 3 files changed, 24 insertions(+), 4 deletions(-)
 create mode 100644 cmd/chromedp-gen/.gitignore

diff --git a/cmd/chromedp-gen/.gitignore b/cmd/chromedp-gen/.gitignore
new file mode 100644
index 0000000..185f360
--- /dev/null
+++ b/cmd/chromedp-gen/.gitignore
@@ -0,0 +1 @@
+.last
diff --git a/cmd/chromedp-gen/build.sh b/cmd/chromedp-gen/build.sh
index 69c5a72..fc7179b 100755
--- a/cmd/chromedp-gen/build.sh
+++ b/cmd/chromedp-gen/build.sh
@@ -1,10 +1,26 @@
 #!/bin/bash
 
-set -ve
-go get -u \
-  github.com/mailru/easyjson/easyjson \
-  github.com/valyala/quicktemplate/qtc
+UPDATE=0
+LASTUPDATE=0
+if [ -f .last ]; then
+  LASTUPDATE=$(cat .last)
+fi
 
+NOW=$(date +%s)
+if (( "$NOW" >= $(($LASTUPDATE + 86400*5)) )); then
+  UPDATE=1
+fi
+
+if [[ "$UPDATE" == 1 ]]; then
+  go get -u \
+    golang.org/x/tools/cmd/goimports \
+    github.com/mailru/easyjson/easyjson \
+    github.com/valyala/quicktemplate/qtc
+
+  date +%s > .last
+fi
+
+set -ve
 go generate
 
 gofmt -w -s templates/*.go
diff --git a/cmd/chromedp-gen/internal/domain.go b/cmd/chromedp-gen/internal/domain.go
index 0b17b9a..0f39e42 100644
--- a/cmd/chromedp-gen/internal/domain.go
+++ b/cmd/chromedp-gen/internal/domain.go
@@ -19,6 +19,7 @@ const (
 	DomainConsole           DomainType = "Console"
 	DomainDOM               DomainType = "DOM"
 	DomainDOMDebugger       DomainType = "DOMDebugger"
+	DomainDOMSnapshot       DomainType = "DOMSnapshot"
 	DomainDOMStorage        DomainType = "DOMStorage"
 	DomainDatabase          DomainType = "Database"
 	DomainDebugger          DomainType = "Debugger"
@@ -83,6 +84,8 @@ func (dt *DomainType) UnmarshalJSON(buf []byte) error {
 		*dt = DomainDOM
 	case DomainDOMDebugger:
 		*dt = DomainDOMDebugger
+	case DomainDOMSnapshot:
+		*dt = DomainDOMSnapshot
 	case DomainDOMStorage:
 		*dt = DomainDOMStorage
 	case DomainDatabase: