From 445b414b0a1a6074fe63497038b415c7f28ff46b Mon Sep 17 00:00:00 2001
From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Thu, 22 Jun 2017 08:39:52 +0100
Subject: [PATCH] [Feature] Allow to use HTTPS when connecting to Clickhouse

---
 src/plugins/lua/clickhouse.lua | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/src/plugins/lua/clickhouse.lua b/src/plugins/lua/clickhouse.lua
index dc6d7be66..2acf94580 100644
--- a/src/plugins/lua/clickhouse.lua
+++ b/src/plugins/lua/clickhouse.lua
@@ -30,6 +30,7 @@ local specific_rows = {}
 local asn_rows = {}
 local symbols_rows = {}
 local nrows = 0
+local connect_prefix = 'http://'
 
 local settings = {
   limit = 1000,
@@ -52,6 +53,7 @@ local settings = {
   full_urls = false,
   from_tables = nil,
   enable_symbols = false,
+  use_https = false,
 }
 
 --[[
@@ -248,7 +250,7 @@ local function clickhouse_send_data(task)
   local body = table.concat(rows, ' ')
   if not rspamd_http.request({
       task = task,
-      url = 'http://' .. settings['server'],
+      url = connect_prefix .. settings['server'],
       body = body,
       callback = http_cb,
       mime_type = 'text/plain',
@@ -262,7 +264,7 @@ local function clickhouse_send_data(task)
     body = table.concat(attachment_rows, ' ')
     if not rspamd_http.request({
       task = task,
-      url = 'http://' .. settings['server'],
+      url = connect_prefix .. settings['server'],
       body = body,
       callback = http_cb,
       mime_type = 'text/plain',
@@ -276,7 +278,7 @@ local function clickhouse_send_data(task)
     body = table.concat(urls_rows, ' ')
     if not rspamd_http.request({
       task = task,
-      url = 'http://' .. settings['server'],
+      url = connect_prefix .. settings['server'],
       body = body,
       callback = http_cb,
       mime_type = 'text/plain',
@@ -290,7 +292,7 @@ local function clickhouse_send_data(task)
     body = table.concat(asn_rows, ' ')
     if not rspamd_http.request({
       task = task,
-      url = 'http://' .. settings['server'],
+      url = connect_prefix .. settings['server'],
       body = body,
       callback = http_cb,
       mime_type = 'text/plain',
@@ -305,7 +307,7 @@ local function clickhouse_send_data(task)
     body = table.concat(symbols_rows, ' ')
     if not rspamd_http.request({
       task = task,
-      url = 'http://' .. settings['server'],
+      url = connect_prefix .. settings['server'],
       body = body,
       callback = http_cb,
       mime_type = 'text/plain',
@@ -321,7 +323,7 @@ local function clickhouse_send_data(task)
       body = table.concat(specific, ' ')
       if not rspamd_http.request({
         task = task,
-        url = 'http://' .. settings['server'],
+        url = connect_prefix .. settings['server'],
         body = body,
         callback = http_cb,
         mime_type = 'text/plain',
@@ -636,6 +638,10 @@ if opts then
     else
       settings['from_map'] = rspamd_map_add('clickhouse', 'from_tables',
         'regexp', 'clickhouse specific domains')
+      if settings.use_https then
+        connect_prefix = 'https://'
+      end
+
       clickhouse_first_row()
       rspamd_config:register_symbol({
         name = 'CLICKHOUSE_COLLECT',
-- 
2.39.5