From 26a3552b7b495b3abdb7c2ad404e713edf5cd044 Mon Sep 17 00:00:00 2001
From: simon987 <me@simon987.net>
Date: Sun, 28 Jun 2020 11:58:34 -0400
Subject: [PATCH] Add log function in go client

---
 client/client.go | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)

diff --git a/client/client.go b/client/client.go
index 8a394a9..6655295 100644
--- a/client/client.go
+++ b/client/client.go
@@ -4,11 +4,13 @@ import (
 	"bytes"
 	"encoding/base64"
 	"encoding/json"
+	"errors"
 	"github.com/simon987/task_tracker/api"
 	"github.com/simon987/task_tracker/storage"
 	"io/ioutil"
 	"net/http"
 	"strconv"
+	"time"
 )
 
 type TaskTrackerClient struct {
@@ -159,6 +161,42 @@ func (c TaskTrackerClient) RequestAccess(req api.CreateWorkerAccessRequest) (api
 	return jsonResp, err
 }
 
+func (c TaskTrackerClient) Log(level storage.LogLevel, message string) error {
+
+	var levelString string
+	req := api.LogRequest{
+		Scope:     "task_tracker go client",
+		Message:   message,
+		TimeStamp: time.Now().Unix(),
+	}
+
+	switch level {
+	case storage.ERROR:
+		levelString = "error"
+	case storage.WARN:
+		levelString = "warn"
+	case storage.INFO:
+		levelString = "panic"
+	case storage.TRACE:
+		levelString = "trace"
+	default:
+		return errors.New("this log level is not implemented")
+	}
+
+	httpResp := c.post("/log/"+levelString, req)
+	var jsonResp api.JsonResponse
+	err := unmarshalResponse(httpResp, &jsonResp)
+
+	if err != nil {
+		return err
+	}
+
+	if !jsonResp.Ok {
+		return errors.New(jsonResp.Message)
+	}
+	return nil
+}
+
 func handleErr(err error) {
 	if err != nil {
 		panic(err)