core/dns-api/store.go

package main

import (
	"fmt"
	"io"
	"os"
)

type RecordStore interface {
	All() (string, error)
	Add(entry, txt string) error
	AddARecord(entry, ip string) error
	Delete(entry, txt string) error
	DeleteARecord(entry, ip string) error
}

type fsRecordStore struct {
	zone     string
	publicIP []string
	fs       FS
	db       string
}

func (s *fsRecordStore) All() (string, error) {
	r, err := s.fs.Reader(s.db)
	if err != nil {
		return "", err
	}
	defer r.Close()
	ret, err := io.ReadAll(r)
	if err != nil {
		return "", err
	}
	return string(ret), nil
}

func (s *fsRecordStore) read() (*RecordsFile, error) {
	r, err := s.fs.Reader(s.db)
	if err != nil {
		return nil, err
	}
	defer r.Close()
	return NewRecordsFile(io.TeeReader(r, os.Stdout))
}

func (s *fsRecordStore) write(z *RecordsFile) error {
	w, err := s.fs.Writer(s.db)
	if err != nil {
		return err
	}
	defer w.Close()
	return z.Write(io.MultiWriter(w, os.Stdout))
}

func (s *fsRecordStore) Add(entry, txt string) error {
	z, err := s.read()
	if err != nil {
		return err
	}
	fqdn := fmt.Sprintf("%s.%s.", entry, s.zone)
	z.CreateOrReplaceTxtRecord(fqdn, txt)
	// for _, ip := range s.publicIP {
	// 	z.CreateARecord(fqdn, ip)
	// }
	return s.write(z)
}

func (s *fsRecordStore) AddARecord(entry, ip string) error {
	z, err := s.read()
	if err != nil {
		return err
	}
	fqdn := fmt.Sprintf("%s.%s.", entry, s.zone)
	z.CreateARecord(fqdn, ip)
	// for _, ip := range s.publicIP {
	// 	z.CreateARecord(fqdn, ip)
	// }
	return s.write(z)
}

func (s *fsRecordStore) Delete(entry, txt string) error {
	z, err := s.read()
	if err != nil {
		return err
	}
	fqdn := fmt.Sprintf("%s.%s.", entry, s.zone)
	z.DeleteTxtRecord(fqdn, txt)
	// z.DeleteRecordsFor(fqdn)
	return s.write(z)
}

func (s *fsRecordStore) DeleteARecord(entry, ip string) error {
	z, err := s.read()
	if err != nil {
		return err
	}
	fqdn := fmt.Sprintf("%s.%s.", entry, s.zone)
	if err := z.DeleteARecord(fqdn, ip); err != nil {
		return err
	}
	// z.DeleteRecordsFor(fqdn)
	return s.write(z)
}