From bea8e5aba8fc84dcb0c980c3948ed6d78719dded Mon Sep 17 00:00:00 2001
From: Julien Dessaux
Date: Thu, 30 Jan 2020 17:44:42 +0100
Subject: Big rafactoring : code split in several modules and some other best
 practices

---
 config/statefile.go | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 config/statefile.go

(limited to 'config/statefile.go')

diff --git a/config/statefile.go b/config/statefile.go
new file mode 100644
index 0000000..8b9f0a9
--- /dev/null
+++ b/config/statefile.go
@@ -0,0 +1,39 @@
+package config
+
+import (
+	"fmt"
+	"log"
+	"os"
+	"path"
+)
+
+const (
+	bareosStateFile = "bareos-fd.9102.state"
+	baculaStateFile = "bacula-fd.9102.state"
+)
+
+func (c *Config) checkStateFile() {
+	// Finds the state file to parse
+	if c.stateFile != "" {
+		c.stateFile = path.Join(c.workDir, c.stateFile)
+		info, err := os.Stat(c.stateFile)
+		if os.IsNotExist(err) || info.IsDir() {
+			fmt.Printf("INFO The state file %s does not exist.\n", c.stateFile)
+			os.Exit(0)
+		}
+	} else {
+		c.stateFile = path.Join(c.workDir, bareosStateFile)
+		info, err := os.Stat(c.stateFile)
+		if os.IsNotExist(err) || info.IsDir() {
+			c.stateFile = path.Join(c.workDir, baculaStateFile)
+			info, err = os.Stat(c.stateFile)
+			if os.IsNotExist(err) || info.IsDir() {
+				fmt.Println("INFO Could not find a suitable state file. Has a job ever run?")
+				os.Exit(0)
+			}
+		}
+	}
+	if c.verbose {
+		log.Println("Using state file ", c.stateFile)
+	}
+}
-- 
cgit v1.2.3