From ece59ad8fb169e28e09109440e82720d9aa61b7c Mon Sep 17 00:00:00 2001
From: Ian Pye <ianpye@gmail.com>
Date: Thu, 29 Feb 2024 22:14:47 -0800
Subject: [PATCH] Allowing provider to be changed for kentik flow

---
 .DS_Store       | Bin 6148 -> 0 bytes
 .gitignore      |   2 ++
 pkg/cat/jchf.go |   8 +++++---
 pkg/cat/kkc.go  |   6 ++++++
 4 files changed, 13 insertions(+), 3 deletions(-)
 delete mode 100644 .DS_Store

diff --git a/.DS_Store b/.DS_Store
deleted file mode 100644
index 5008ddfcf53c02e82d7eee2e57c38e5672ef89f6..0000000000000000000000000000000000000000
GIT binary patch
literal 0
HcmV?d00001

literal 6148
zcmeH~Jr2S!425mzP>H1@V-^m;4Wg<&0T*E43hX&L&p$$qDprKhvt+--jT7}7np#A3
zem<@ulZcFPQ@L2!n>{z**<q8>++&mCkOWA81W14cNZ<zv;LbK1Poaz?KmsK2CSc!(
z0ynLxE!0092;Krf2c+FF_Fe*7ECH>lEfg7;MkzE(HCqgga^y>{tEnwC%0;vJ&^%eQ
zLs35+`xjp>T0<F0fCPF1$Cyrb|F7^5{eNG?83~ZUUlGt@xh*qZDeu<Z%US-OSsOPv
j)R!Z4KLME7ReXlK;d!wEw5GODWMKRea10D2@KpjYNUI8I

diff --git a/.gitignore b/.gitignore
index dd2efc17..2fd7e779 100644
--- a/.gitignore
+++ b/.gitignore
@@ -28,3 +28,5 @@ cmd/version/generated_version.go
 
 ## local configuration
 config.yaml
+
+.DS_Store
diff --git a/pkg/cat/jchf.go b/pkg/cat/jchf.go
index a81ca486..d4e3b387 100644
--- a/pkg/cat/jchf.go
+++ b/pkg/cat/jchf.go
@@ -18,6 +18,8 @@ import (
 var (
 	DEFAULT_GEO        = []byte("--")
 	DEFAULT_GEO_PACKED = patricia.PackGeo(DEFAULT_GEO)
+
+	defaultProvider = kt.ProviderFlowDevice
 )
 
 func (kc *KTranslate) getEventType(dst *kt.JCHF) string {
@@ -47,12 +49,12 @@ func (kc *KTranslate) getProviderType(dst *kt.JCHF) kt.Provider {
 
 	udr, ok := dst.CustomStr[UDR_TYPE]
 	if !ok { // Return this right away.
-		return kt.ProviderFlowDevice
+		return defaultProvider
 	}
 
 	// Or maybe its a host.
 	if udr == "kprobe" || udr == "kappa" {
-		return kt.ProviderFlowDevice
+		return defaultProvider
 	}
 
 	// Else, if its synth, return this.
@@ -66,7 +68,7 @@ func (kc *KTranslate) getProviderType(dst *kt.JCHF) kt.Provider {
 	}
 
 	// Default to provider.
-	return kt.ProviderFlowDevice
+	return defaultProvider
 }
 
 func (kc *KTranslate) flowToJCHF(ctx context.Context, dst *kt.JCHF, src *Flow, currentTS int64, tagcache map[uint64]string) error {
diff --git a/pkg/cat/kkc.go b/pkg/cat/kkc.go
index b640a8ff..7d95f946 100644
--- a/pkg/cat/kkc.go
+++ b/pkg/cat/kkc.go
@@ -216,6 +216,12 @@ func NewKTranslate(config *ktranslate.Config, log logger.ContextL, registry go_m
 		kc.objmgr = s3mgr
 	}
 
+	// If the default provider env var is set, pass this into the system.
+	dp := kt.LookupEnvString("KENTIK_DEFAULT_PROVIDER_TYPE", "")
+	if dp != "" {
+		defaultProvider = kt.Provider(dp)
+	}
+
 	return kc, nil
 }