From f1b7f703134236ab09ec51fafe46655bf52c90f4 Mon Sep 17 00:00:00 2001
From: Arsen Musayelyan <moussaelianarsen@gmail.com>
Date: Wed, 27 Oct 2021 07:27:12 -0700
Subject: [PATCH] Add whitelist support

---
 go.mod   | 2 +-
 go.sum   | 4 ++--
 itd.toml | 4 ++++
 main.go  | 2 ++
 4 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/go.mod b/go.mod
index a8cb8cd..ae3def2 100644
--- a/go.mod
+++ b/go.mod
@@ -25,7 +25,7 @@ require (
 	github.com/srwiley/oksvg v0.0.0-20210519022825-9fc0c575d5fe // indirect
 	github.com/srwiley/rasterx v0.0.0-20210519020934-456a8d69b780 // indirect
 	github.com/yuin/goldmark v1.4.1 // indirect
-	go.arsenm.dev/infinitime v0.0.0-20211023042633-53aa6f8a0c72
+	go.arsenm.dev/infinitime v0.0.0-20211027044241-75942bdb0cab
 	golang.org/x/image v0.0.0-20210628002857-a66eb6448b8d // indirect
 	golang.org/x/net v0.0.0-20211011170408-caeb26a5c8c0 // indirect
 	golang.org/x/sys v0.0.0-20211007075335-d3039528d8ac // indirect
diff --git a/go.sum b/go.sum
index 107892e..6be6231 100644
--- a/go.sum
+++ b/go.sum
@@ -367,8 +367,8 @@ github.com/yuin/goldmark v1.3.5/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1
 github.com/yuin/goldmark v1.3.8/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k=
 github.com/yuin/goldmark v1.4.1 h1:/vn0k+RBvwlxEmP5E7SZMqNxPhfMVFEJiykr15/0XKM=
 github.com/yuin/goldmark v1.4.1/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k=
-go.arsenm.dev/infinitime v0.0.0-20211023042633-53aa6f8a0c72 h1:e8kOuL6Jj8ZjJzkGwJ3xqpGG9EhUzfvZk9AlSsm3X1U=
-go.arsenm.dev/infinitime v0.0.0-20211023042633-53aa6f8a0c72/go.mod h1:gaepaueUz4J5FfxuV19B4w5pi+V3mD0LTef50ryxr/Q=
+go.arsenm.dev/infinitime v0.0.0-20211027044241-75942bdb0cab h1:IbvDG8y94zCzjAEx0UDsrV6pCJo2uoETH99DM6E3Z1A=
+go.arsenm.dev/infinitime v0.0.0-20211027044241-75942bdb0cab/go.mod h1:gaepaueUz4J5FfxuV19B4w5pi+V3mD0LTef50ryxr/Q=
 go.etcd.io/etcd/api/v3 v3.5.0/go.mod h1:cbVKeC6lCfl7j/8jBhAK6aIYO9XOjdptoxU/nLQcPvs=
 go.etcd.io/etcd/client/pkg/v3 v3.5.0/go.mod h1:IJHfcCEKxYu1Os13ZdwCwIUTUVGYTSAM3YSwc9/Ac1g=
 go.etcd.io/etcd/client/v2 v2.305.0/go.mod h1:h9puh54ZTgAKtEbut2oe9P4L/oqKCVB6xsXlzd7alYQ=
diff --git a/itd.toml b/itd.toml
index d5990c0..c78fdf1 100644
--- a/itd.toml
+++ b/itd.toml
@@ -8,6 +8,10 @@ cfg.version = 2
 [conn]
     reconnect = true
 
+[conn.whitelist]
+    enabled = false
+    devices = []
+
 [on.connect]
     notify = true
 
diff --git a/main.go b/main.go
index 34561fc..7b415ee 100644
--- a/main.go
+++ b/main.go
@@ -39,6 +39,8 @@ func main() {
 	// Connect to InfiniTime with default options
 	dev, err := infinitime.Connect(&infinitime.Options{
 		AttemptReconnect: viper.GetBool("conn.reconnect"),
+		WhitelistEnabled: viper.GetBool("conn.whitelist.enabled"),
+		Whitelist:        viper.GetStringSlice("conn.whitelist.devices"),
 	})
 	if err != nil {
 		log.Error().Err(err).Msg("Error connecting to InfiniTime")