DEVELOPMENT ENVIRONMENT

~liljamo/felu

ref: d2a556d1cc463afad68f93a22c613261ba4c167c felu/internal/db/domains.go -rw-r--r-- 1.3 KiB
d2a556d1Jonni Liljamo refactor: directory layout 11 months ago
                                                                                
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
/*
 * Copyright (C) 2023 Jonni Liljamo <jonni@liljamo.com>
 *
 * This file is licensed under AGPL-3.0-or-later, see NOTICE and LICENSE for
 * more information.
 */
package db

import (
	"git.src.quest/~skye/felu-ddns/internal/api"
	"github.com/oklog/ulid/v2"
)

type Domain struct {
	Id      string
	ApiKey  string
	Domain  string
	A       string
}

func FetchDomainsForUser(userId string) ([]Domain, error) {
	rows, err := DBConn.Query(`SELECT id, apikey, ddns_domain, a_record
		FROM domains WHERE owner = $1`, userId)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	var domains []Domain
	for rows.Next() {
		var domain Domain
		err = rows.Scan(&domain.Id, &domain.ApiKey, &domain.Domain, &domain.A)
		if err != nil {
			return nil, err
		}
		domains = append(domains, domain)
	}
	err = rows.Err()
	if err != nil {
		return nil, err
	}

	return domains, nil
}

func CreateDomain(domain string, aRecord string, owner string) error {
	ulid := ulid.Make().String()
	apikey := api.GenKey()
	_, err := DBConn.Exec(`INSERT INTO domains(id, apikey, ddns_domain, a_record, owner)
		VALUES ($1, $2, $3, $4, $5)`, ulid, apikey, domain, aRecord, owner)
	if err != nil {
		return err
	}

	return nil
}

func DeleteDomain(id string) error {
	_, err := DBConn.Exec(`DELETE FROM domains WHERE id = $1`, id)
	if err != nil {
		return err
	}
	return nil
}