From c2fe14c0d774f89a12ab593a1acebfd8803ce6e5 Mon Sep 17 00:00:00 2001
From: Pierre-Olivier Mercier <nemunaire@nemunai.re>
Date: Wed, 21 Jul 2021 10:42:33 +0200
Subject: [PATCH] db: Handle connection through unix socket

---
 libfic/db.go | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/libfic/db.go b/libfic/db.go
index 41ad70af..eb049ff6 100644
--- a/libfic/db.go
+++ b/libfic/db.go
@@ -5,6 +5,7 @@ import (
 	_ "github.com/go-sql-driver/mysql"
 	"log"
 	"os"
+	"strings"
 	"time"
 )
 
@@ -19,11 +20,15 @@ func DSNGenerator() string {
 	db_db := "fic"
 
 	if v, exists := os.LookupEnv("MYSQL_HOST"); exists {
-		db_host = "tcp(" + v + ":"
-		if p, exists := os.LookupEnv("MYSQL_PORT"); exists {
-			db_host += p + ")"
+		if strings.HasPrefix(v, "/") {
+			db_host = "unix(" + v + ")"
 		} else {
-			db_host += "3306)"
+			db_host = "tcp(" + v + ":"
+			if p, exists := os.LookupEnv("MYSQL_PORT"); exists {
+				db_host += p + ")"
+			} else {
+				db_host += "3306)"
+			}
 		}
 	}
 	if v, exists := os.LookupEnv("MYSQL_PASSWORD"); exists {