Autodetect datetime format

index.go

@@ -62,7 +62,8 @@ func (index Index) Save(fileName string) error {
 
 	enc, err := zstd.NewWriter(f, zstd.WithEncoderLevel(zstd.SpeedBestCompression))
 	if err != nil {
-		f.Close() // TODO: удалить частичный файл?
+		f.Close()
+		os.Remove(fileName)
 		return err
 	}
 
@@ -82,7 +83,8 @@ func (index Index) Save(fileName string) error {
 			err := csvWriter.Write([]string{fileName, historyItem.ArchiveFileName, strconv.Itoa(int(historyItem.ModificationTime.Unix()))})
 			if err != nil {
 				enc.Close()
-				f.Close() // TODO: удалить частичный файл?
+				f.Close()
+				os.Remove(fileName)
 				return err
 			}
 		}
@@ -91,13 +93,15 @@ func (index Index) Save(fileName string) error {
 	csvWriter.Flush()
 	if err := csvWriter.Error(); err != nil {
 		enc.Close()
-		f.Close() // TODO: удалить частичный файл?
+		f.Close()
+		os.Remove(fileName)
 		return err
 	}
 
 	err = enc.Close()
 	if err != nil {
-		f.Close() // TODO: удалить частичный файл?
+		f.Close()
+		os.Remove(fileName)
 		return err
 	}
 

main.go

@@ -4,7 +4,6 @@ import (
 	"log"
 	"os"
 	"path/filepath"
-	"time"
 )
 
 func init() {
@@ -56,25 +55,9 @@ func main() {
 			log.Fatalln(err)
 		}
 
-		var t time.Time
+		t, err := parseTime(os.Args[4])
-		switch len(os.Args[4]) {
+		if err != nil {
-		case len("02.01.2006"):
+			config.fatalln(err)
-			t, err = time.Parse("02.01.2006 15:04", os.Args[4])
-			if err != nil {
-				config.fatalln("time parse error:", err)
-			}
-		case len("02.01.2006 15:04"):
-			t, err = time.Parse("02.01.2006 15:04", os.Args[4])
-			if err != nil {
-				config.fatalln("time parse error:", err)
-			}
-		case len("02.01.2006 15:04:05"):
-			t, err = time.Parse("02.01.2006 15:04:05", os.Args[4])
-			if err != nil {
-				config.fatalln("time parse error:", err)
-			}
-		default:
-			config.fatalln(`wrong time format, must be ["DD.MM.YYYY", "DD.MM.YYYY hh:mm", "DD.MM.YYYY hh:mm:ss"]`)
 		}
 
 		plan, err := config.extractionPlan(os.Args[3], t)

utils.go

@@ -1,9 +1,11 @@
 package main
 
 import (
+	"errors"
 	"fmt"
 	"path/filepath"
 	"strings"
+	"time"
 
 	"github.com/tidwall/match"
 )
@@ -32,7 +34,7 @@ func sizeToApproxHuman(s int64) string {
 	return fmt.Sprintf("%d B", s)
 }
 
-// clean убирает невозможнын комбинации символов из пути
+// clean убирает невозможные комбинации символов из пути
 func clean(s string) string {
 	s = strings.ReplaceAll(s, ":", "")
 	s = strings.ReplaceAll(s, `\\`, `\`)
@@ -71,3 +73,16 @@ func isFilePathMatchPatterns(patterns []string, fileName string) bool {
 
 	return false
 }
+
+func parseTime(s string) (time.Time, error) {
+	switch len(s) {
+	case len("02.01.2006"):
+		return time.ParseInLocation("02.01.2006", s, time.Local)
+	case len("02.01.2006 15:04"):
+		return time.ParseInLocation("02.01.2006 15:04", s, time.Local)
+	case len("02.01.2006 15:04:05"):
+		return time.ParseInLocation("02.01.2006 15:04:05", s, time.Local)
+	}
+
+	return time.Time{}, errors.New("unknown time format")
+}

utils_test.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"testing"
+	"time"
 
 	"github.com/stretchr/testify/assert"
 )
@@ -10,3 +11,21 @@ func TestSizeToApproxHuman(t *testing.T) {
 	assert.Equal(t, "1.0 KiB", sizeToApproxHuman(1024))
 	assert.Equal(t, "1.1 KiB", sizeToApproxHuman(1126))
 }
+
+func TestParseTime(t *testing.T) {
+	tests := []struct {
+		input    string
+		expected time.Time
+	}{
+		{"02.01.2006", time.Date(2006, 01, 02, 0, 0, 0, 0, time.Local)},
+		{"02.01.2006 15:04", time.Date(2006, 01, 02, 15, 4, 0, 0, time.Local)},
+		{"02.01.2006 15:04:05", time.Date(2006, 01, 02, 15, 4, 5, 0, time.Local)},
+	}
+
+	for _, test := range tests {
+		got, err := parseTime(test.input)
+		assert.NoError(t, err)
+
+		assert.Equal(t, test.expected, got)
+	}
+}