plugin

var (
	GetPluginConfig = promauto.NewCounter(prometheus.CounterOpts{
		Namespace: metrics.Namespace,
		Name:      "get_plugin_config_total",
		Help:      "The total number of calls to the getPluginConfig method",
	})
	OnTrafficFromClient = promauto.NewCounter(prometheus.CounterOpts{
		Namespace: metrics.Namespace,
		Name:      "on_traffic_from_client_total",
		Help:      "The total number of calls to the onTrafficFromClient method",
	})
	Detections = promauto.NewCounterVec(prometheus.CounterOpts{
		Namespace: metrics.Namespace,
		Name:      "detections_total",
		Help:      "The total number of malicious requests detected",
	}, []string{"detector"})
	Preventions = promauto.NewCounter(prometheus.CounterOpts{
		Namespace: metrics.Namespace,
		Name:      "preventions_total",
		Help:      "The total number of malicious requests prevented",
	})
)

var (
	PluginID = v1.PluginID{
		Name:      "gatewayd-plugin-sql-ids-ips",
		Version:   "0.0.5",
		RemoteUrl: "github.com/gatewayd-io/gatewayd-plugin-sql-ids-ips",
	}
	PluginMap = map[string]goplugin.Plugin{
		"gatewayd-plugin-sql-ids-ips": &InjectionDetectionPlugin{},
	}
	// TODO: Handle this in a better way
	// https://github.com/gatewayd-io/gatewayd-plugin-sdk/issues/3
	PluginConfig = map[string]interface{}{
		"id": map[string]interface{}{
			"name":      PluginID.Name,
			"version":   PluginID.Version,
			"remoteUrl": PluginID.RemoteUrl,
		},
		"description": "GatewayD plugin for detection and prevention of SQL injection attacks",
		"authors": []interface{}{
			"Mostafa Moradian <mostafa@gatewayd.io>",
		},
		"license":    "AGPL-3.0",
		"projectUrl": "https://github.com/gatewayd-io/gatewayd-plugin-sql-ids-ips",

		"config": map[string]interface{}{
			"metricsEnabled": sdkConfig.GetEnv("METRICS_ENABLED", "true"),
			"metricsUnixDomainSocket": sdkConfig.GetEnv(
				"METRICS_UNIX_DOMAIN_SOCKET", "/tmp/gatewayd-plugin-sql-ids-ips.sock"),
			"metricsEndpoint": sdkConfig.GetEnv("METRICS_ENDPOINT", "/metrics"),
			"tokenizerAPIAddress": sdkConfig.GetEnv(
				"TOKENIZER_API_ADDRESS", "http://localhost:8000"),
			"servingAPIAddress": sdkConfig.GetEnv(
				"SERVING_API_ADDRESS", "http://localhost:8501"),
			"modelName":                  sdkConfig.GetEnv("MODEL_NAME", "sqli_model"),
			"modelVersion":               sdkConfig.GetEnv("MODEL_VERSION", "1"),
			"threshold":                  sdkConfig.GetEnv("THRESHOLD", "0.8"),
			"enableLibinjection":         sdkConfig.GetEnv("ENABLE_LIBINJECTION", "true"),
			"libinjectionPermissiveMode": sdkConfig.GetEnv("LIBINJECTION_MODE", "true"),
		},
		"hooks": []interface{}{

			int32(v1.HookName_HOOK_NAME_ON_TRAFFIC_FROM_CLIENT),
		},
		"tags":       []interface{}{"plugin", "sql", "ids", "ips", "security", "waf"},
		"categories": []interface{}{"plugin", "enterprise"},
	}
)