szczanieckiej.git

commit 35b610943bef25561debf215eb6a1f97fdcf4819

Author: Adam Evyčędo <git@apiote.xyz>

save to TRAFFIC file lua scripts if they exist

 file/file.go | 12 ++++++++++--


diff --git a/file/file.go b/file/file.go
index e5865ba71e0e91762fbc42890a779cc8b4f4db53..020b851b21f27bcf81f7148f4a02aad27da60c71 100644
--- a/file/file.go
+++ b/file/file.go
@@ -89,11 +89,19 @@ 	defer xzWriter.Close()
 	tarWriter := tar.NewWriter(xzWriter)
 	defer tarWriter.Close()
 
-	files := []string{"calendar.bare", "ix_stop_codes.bare", "ix_stop_names.bare", "lines.bare", "stops.bare", "trips.bare", "vehicles.bare", "ix_lines.bare", "feed_info.bare", "agencies.bare", "ix_trips.bare", "updates.lua"}
+	files := []string{"calendar.bare", "ix_stop_codes.bare", "ix_stop_names.bare", "lines.bare", "stops.bare", "trips.bare", "vehicles.bare", "ix_lines.bare", "feed_info.bare", "agencies.bare", "ix_trips.bare", "updates.lua", "vehicles.lua", "alerts.lua"}
+	optionalFiles := map[string]struct{}{
+		"updates.lua":  {},
+		"vehicles.lua": {},
+		"alerts.lua":   {},
+	}
 	for _, file := range files {
 		stat, err := os.Stat(filepath.Join(path, file))
 		if err != nil {
-			// TODO if updates.lua does not exist, ignore
+			if _, ok := optionalFiles[file]; ok {
+				log.Printf("while stating %s: %v; ignoring\n", file, err)
+				continue
+			}
 			return err
 		}
 		bareFile, err := os.Open(filepath.Join(path, file))