Refactoring.

writer/cmd/main.go

@@ -2,11 +2,11 @@ package main
 
 import (
 	"fmt"
-	logger "git.insit.tech/psa/rtsp_reader-writer/writer/internal/log"
 	"time"
 
 	"git.insit.tech/psa/rtsp_reader-writer/writer/internal/config"
 	"git.insit.tech/psa/rtsp_reader-writer/writer/internal/ingest/rtsp"
+	logger "git.insit.tech/psa/rtsp_reader-writer/writer/internal/log"
 	log2 "git.insit.tech/sas/rtsp_proxy/core/log"
 )
 

writer/go.mod

@@ -11,6 +11,7 @@ require (
 	github.com/gen2brain/aac-go v0.0.0-20230119102159-ef1e76509d21
 	github.com/golang/snappy v1.0.0
 	github.com/pion/rtp v1.8.13
+	go.uber.org/zap v1.27.0
 	gopkg.in/yaml.v3 v3.0.1
 )
 
@@ -23,7 +24,6 @@ require (
 	github.com/pion/sdp/v3 v3.0.11 // indirect
 	github.com/zaf/g711 v1.4.0 // indirect
 	go.uber.org/multierr v1.11.0 // indirect
-	go.uber.org/zap v1.27.0 // indirect
 	golang.org/x/net v0.37.0 // indirect
 	golang.org/x/sys v0.31.0 // indirect
 	gopkg.in/natefinch/lumberjack.v2 v2.2.1 // indirect

writer/internal/ingest/rtsp/operator.go

@@ -8,8 +8,8 @@ import (
 	"go.uber.org/zap"
 )
 
-// lastPartURI returns the last part of the URI after "/".
+// LastPartURI returns the last part of the URI after "/".
-func lastPartURI(URI string) (CutURI string) {
+func LastPartURI(URI string) (CutURI string) {
 	split := strings.Split(URI, "/")
 	return split[len(split)-1]
 }

writer/internal/ingest/rtsp/rtsp.go

@@ -52,7 +52,7 @@ func rtsp(dir string, period int, link string, number int) error {
 	dirData := log2.DirCreator(dir, "data")
 
 	// Create logger.
-	cutURI := lastPartURI(link)
+	cutURI := LastPartURI(link)
 	cam := log2.CamLogging(fmt.Sprintf("%s/%s/Cam_%s.log", dirData, cutURI, time.Now().Format("15-04-05_02-01-2006")))
 
 	// Connect to the server.

writer/pkg/storage/file_manager.go

@@ -2,6 +2,7 @@ package storage
 
 import (
 	"git.insit.tech/sas/rtsp_proxy/proto/common"
+	"os"
 	"time"
 )
 
@@ -17,3 +18,18 @@ func CreateFileName(dirData string, resolutions []string, cutURI string, period
 
 	return &fn
 }
+
+// ReadDir reads directory and returns the slice of files in th directory.
+func ReadDir(dirData string) ([]os.DirEntry, error) {
+	homeDir, err := os.UserHomeDir()
+	if err != nil {
+		return nil, err
+	}
+
+	files, err := os.ReadDir(homeDir + dirData)
+	if err != nil {
+		return nil, err
+	}
+
+	return files, nil
+}