about summary refs log tree commit diff
path: root/http.go
blob: ece4fde9eef0ebe9c02dbe28267fe3a629d363e4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
package main

import (
	"encoding/json"
	"fmt"
	"io/ioutil"
	"log"
	"net/http"
	"strings"

	tle "git.darknebu.la/Satellite/tle"
	"github.com/gorilla/mux"
)

// indexHandler displays an index page
func indexHandler(w http.ResponseWriter, r *http.Request) {
	fmt.Fprintf(w, "%s", "<h1>Hello World</h1>")
}

// dumpallHandler dumps all categories
func dumpallHandler(w http.ResponseWriter, r *http.Request) {
	log.Println("[ ] Dumping all TLEs")

	// 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)
		}
	}

	// download all the categories to the data/<category name> directory
	for _, category := range categories {
		fmt.Fprintf(w, "downloading %s... ", category)
		log.Printf("downloading %s... ", category)
		downloadCategory(category)
		fmt.Fprintf(w, "done.\n")
	}

	log.Println("[+] Done dumping all TLEs")
}

// get Handler returns the specified TLE
func getHandler(w http.ResponseWriter, r *http.Request) {
	vars := mux.Vars(r)

	// read the TLEs from the categories file
	// TODO: Handle caching
	//			- dump all if not present
	//			- if old, fetch from celestrack

	// open the file containing the data for the individual station
	categoryPath := fmt.Sprintf("data/%s.txt", vars["station"])
	content, err := ioutil.ReadFile(categoryPath)
	if err != nil {
		log.Printf("Could not open %s", categoryPath)
	}

	lines := strings.Split(string(content), "\n")

	// iterate over all lines searching for the satellite
	var j int
	for i, line := range lines {
		if strings.Contains(line, vars["name"]) {
			j = i
			break
		}
	}

	RawTLE := strings.Join(lines[j:j+3], "\n")

	// handle the response according to the query parameter "format"
	// if ?format=json is given, return
	if r.URL.Query().Get("format") == "json" {
		w.Header().Set("Content-Type", "application/json")

		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)
		}
		fmt.Fprintf(w, string(b))
		//fmt.Fprintf(w, "{\"response\": \"Getting the TLE for %s/%s as json\"}", vars["station"], vars["name"])
	} else {
		fmt.Fprintf(w, "Getting the TLE for %s/%s raw", vars["station"], vars["name"])
	}

}