feat: adding logn zip, used to automate zip your log every month

Author: ibnumardini

README.md

@@ -32,6 +32,14 @@ os.Setenv("logn_default_loc", "Asia/Jakarta") // timezone in log
 os.Setenv("tg_send", "true") // set true, if you need report log type warning & error to telegam
 os.Setenv("tg_token", "1416xxxxx:AAF3VOBjt7rIeO4tUL_dHxG0qxxxxxxxxx") // tg bot token
 os.Setenv("tg_chat_id", "-4121xxxx") //  tg grup / chat_id
+os.Setenv("is_zipped", "true") // if you need to zip your log every month
+os.Setenv("logn_dir_zip", "log_zip/") // set directory to save your log
+```
+
+Set this in main func.
+
+```go
+logn.CronZip() // this command used to run zip log with cronjob
 ```
 
 ## Usage
@@ -42,6 +50,13 @@ logn.WarningLog("this is log warning")
 logn.ErrorLog("this is log error")
 ```
 
+#### To run zip with cron
+
+```script
+./app logn_zip_run // add this args in first args
+
+```
+
 ## Contributing
 
 Thank you for considering contributing to the Logn!.

logn.go

@@ -1,12 +1,16 @@
 package logn
 
 import (
+	"archive/zip"
+	"errors"
 	"fmt"
+	"io/ioutil"
 	"log"
 	"net/http"
 	"net/url"
 	"os"
 	"runtime"
+	"strconv"
 	"strings"
 	"time"
 )
@@ -135,6 +139,9 @@ func makeLog(typeLogs int, title string, logMessage interface{}, loc string) err
 	if typeLog != "info" && isSendTg == "true" {
 		sendTg(url.QueryEscape(logMessageTg))
 	}
+
+	makeZip()
+
 	return nil
 }
 
@@ -204,3 +211,158 @@ func filePath(original string) string {
 		return original[i+1:]
 	}
 }
+
+// CronZip used for cron zipped log
+func CronZip() error {
+	if len(os.Args) == 2 {
+		if os.Args[1] == "logn_zip_run" {
+			err := makeZip()
+			if err != nil {
+				return err
+			}
+		}
+	}
+
+	return nil
+}
+
+func makeZip() error {
+	var dirZip, isZipped, dir string
+
+	if len(os.Getenv("logn_dir")) != 0 {
+		dir = os.Getenv("logn_dir")
+	} else {
+		dir = "log/"
+	}
+
+	if len(os.Getenv("logn_dir_zip")) != 0 {
+		dirZip = os.Getenv("logn_dir_zip")
+	} else {
+		dirZip = "log_zip/"
+	}
+
+	if len(os.Getenv("is_zipped")) != 0 {
+		isZipped = os.Getenv("is_zipped")
+	} else {
+		isZipped = "false"
+	}
+
+	year, err := timeNow("Y")
+	if err != nil {
+		return err
+	}
+
+	month, err := timeNow("M")
+	if err != nil {
+		return err
+	}
+
+	day, err := timeNow("D")
+	if err != nil {
+		return err
+	}
+
+	if day == "01" && isZipped == "true" {
+
+		monthInt, err := strconv.Atoi(month)
+		if err != nil {
+			return err
+		}
+
+		prevMonth := monthInt - 1
+
+		if month == "01" {
+			prevMonth = 12
+
+			yearInt, err := strconv.Atoi(year)
+			if err != nil {
+				return err
+			}
+
+			prevYear := yearInt - 1
+
+			year = strconv.Itoa(prevYear)
+		}
+
+		dirFile := dirZip + year
+
+		_, errYear := os.Stat(dirFile)
+		if os.IsNotExist(errYear) {
+			errDir := os.MkdirAll(dirFile, 0755)
+			if errDir != nil {
+				return errDir
+			}
+		}
+
+		filename := fmt.Sprintf("log_%s_%s.zip", year, strconv.Itoa(prevMonth))
+
+		baseDir := dir + year + "/" + strconv.Itoa(prevMonth) + "/"
+		_, errYear = os.Stat(baseDir)
+		if os.IsNotExist(errYear) {
+			return errors.New("base dir doesn't exists")
+		}
+
+		outDir := dirFile + "/" + filename
+
+		err = zipWriter(baseDir, outDir)
+		if err != nil {
+			fmt.Println(err.Error())
+		}
+	}
+
+	return nil
+}
+
+func zipWriter(baseDir, outDir string) error {
+	outFile, err := os.Create(outDir)
+	if err != nil {
+		return err
+	}
+	defer outFile.Close()
+
+	w := zip.NewWriter(outFile)
+
+	err = addFiles(w, baseDir, "")
+	if err != nil {
+		return err
+	}
+
+	err = w.Close()
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func addFiles(w *zip.Writer, basePath, baseInZip string) error {
+	files, err := ioutil.ReadDir(basePath)
+	if err != nil {
+		return err
+	}
+
+	for _, file := range files {
+		fmt.Println(basePath + file.Name())
+		if !file.IsDir() {
+			dat, err := ioutil.ReadFile(basePath + file.Name())
+			if err != nil {
+				return err
+			}
+
+			f, err := w.Create(baseInZip + file.Name())
+			if err != nil {
+				return err
+			}
+			_, err = f.Write(dat)
+			if err != nil {
+				return err
+			}
+		} else if file.IsDir() {
+			newBase := basePath + file.Name() + "/"
+
+			addFiles(w, newBase, baseInZip+file.Name()+"/")
+		}
+	}
+
+	return nil
+}