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
|
package main
import (
"encoding/json"
"fmt"
"io/ioutil"
"log"
"net/http"
"os"
"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>TLE2JSON api</h1>")
fmt.Fprintf(w, "%s", "Usage: Make a get request to /{station}/{name}")
}
// dumpallHandler dumps all categories
func dumpallHandler(w http.ResponseWriter, r *http.Request) {
log.Println("[ ] Dumping all TLEs")
dumpall()
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
// define where the file containing the spoecified categories lies
categoryPath := fmt.Sprintf("data/%s.txt", vars["station"])
// find out if the file exists or not
if _, err := os.Stat(categoryPath); os.IsNotExist(err) {
// if it does not exist, download the category
downloadCategory(vars["station"])
}
// 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")
// 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")
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))
}
|
