implemented the printAll endpoint

2 files changed, 63 insertions, 0 deletions

diff --git a/http.go b/http.go
index 4c77fcb..9808f07 100644
--- a/http.go
+++ b/http.go
@@ -68,5 +68,67 @@ func getHandler(w http.ResponseWriter, r *http.Request) {
 		fmt.Println("error: ", err)
 	}
 	fmt.Fprintf(w, string(b))
+}
+
+func printAllHandler(w http.ResponseWriter, r *http.Request) {
+	r.ParseForm()
+	dumpall()
+	w.Header().Set("Content-Type", "application/json")
+
+	// read the name of the categories from the categories.txt file and write
+	// the result to the categories slice
+	content, err := ioutil.ReadFile("categories.txt")
+	if err != nil {
+		log.Printf("%s\n", "could not read categories.txt file!")
+	}
+	lines := strings.Split(string(content), "\n")
+
+	for _, line := range lines {
+		if line != "" {
+			categories = append(categories, line)
+		}
+	}
+
+	fmt.Fprintf(w, "[")
 
+	// read the content of all files
+	for j, category := range categories {
+
+		// define where the file containing the spoecified categories lies
+		categoryPath := fmt.Sprintf("data/%s.txt", category)
+
+		// this will only download the category if the file does not exist or if it
+		// is to old
+		downloadCategory(category)
+
+		// open the file containing the data for the individual station
+		content, err := ioutil.ReadFile(categoryPath)
+		if err != nil {
+			log.Printf("Could not open %s", categoryPath)
+		}
+
+		lines := strings.Split(string(content), "\n")
+
+		for i := 0; i < len(lines)-3; i += 3 {
+			RawTLE := strings.Join(lines[i:i+3], "\n")
+
+			JSONTLE, err := tle.NewTLE(RawTLE)
+			if err != nil {
+				fmt.Println(err)
+			}
+
+			// convert the TLE to json
+			b, err := json.MarshalIndent(JSONTLE, "", " ")
+			if err != nil {
+				fmt.Println("error: ", err)
+			}
+
+			if i == len(lines)-4 && j == len(categories)-1 {
+				fmt.Fprintf(w, "%s", string(b))
+			} else {
+				fmt.Fprintf(w, "%s,", string(b))
+			}
+		}
+	}
+	fmt.Fprintf(w, "]")
 }
diff --git a/server.go b/server.go
index 928ca83..86df6a5 100644
--- a/server.go
+++ b/server.go
@@ -31,6 +31,7 @@ func main() {
 
 	r.HandleFunc("/", indexHandler)
 	r.HandleFunc("/dumpall", dumpallHandler)
+	r.HandleFunc("/all", printAllHandler)
 	r.HandleFunc("/{station}/{name}", getHandler)
 
 	// Start the http server using the port provided using command line arguments