aboutsummaryrefslogtreecommitdiff
path: root/db.go
blob: 876e055e7c2e0c260ed03525a7eae3a400ca6610 (plain) (blame)
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
package main

import (
	"encoding/csv"
	"errors"
	"log"
	"os"
	"time"
)

func readAll() [][]string {
	f, err := os.Open("bowl.csv")
	defer f.Close()
	if err != nil {
		if errors.Is(err, os.ErrNotExist) {
			f, _ = os.Create("bowl.csv")
			_, err := f.Write([]byte("#type,notes,datetime,ip\n"))
			if err != nil {
				log.Fatalf("error writing to file: %s\n", err)
			}
		} else {
			log.Fatalf("failed to open file: %s\n", err)
		}
	}

	r := csv.NewReader(f)
	records, err := r.ReadAll()
	if err != nil {
		log.Fatalf("failed to read file: %s\n", err)
	}

	return records
}

func write(formType string, formNotes string, ip string) {
	records := readAll()

	f, err := os.Create("bowl.csv")
	defer f.Close()
	if err != nil {
		if errors.Is(err, os.ErrNotExist) {
			f, _ = os.Create("bowl.csv")
			_, err := f.Write([]byte("#type,notes,datetime,ip\n"))
			if err != nil {
				log.Fatalf("error writing to file: %s\n", err)
			}
		} else {
			log.Fatalf("failed to open file: %s\n", err)
		}
	}

	records = append(records, []string{
		formType, formNotes, time.Now().String(), ip,
	})

	w := csv.NewWriter(f)
	defer w.Flush()

	err = w.WriteAll(records)
	if err != nil {
		log.Fatalf("error writing entry to file: %s\n", err)
	}
}