30 lines
681 B
Go
30 lines
681 B
Go
|
package controllers
|
||
|
|
|
||
|
|
import (
|
||
|
|
"reCoreD-UI/models"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"gorm.io/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
const dnsSep = ","
|
||
|
|
|
||
|
|
func (c *Controller) SetupDNS(dns ...string) error {
|
||
|
|
return c.DB.Transaction(func(tx *gorm.DB) error {
|
||
|
|
settings := &models.Settings{}
|
||
|
|
|
||
|
|
return tx.Where(&models.Settings{Key: models.SettingsKeyDNSServer}).
|
||
|
|
Attrs(&models.Settings{Value: strings.Join(dns, dnsSep)}).
|
||
|
|
FirstOrCreate(&settings).Error
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Controller) GetDNS() ([]string, error) {
|
||
|
|
settings := &models.Settings{}
|
||
|
|
if err := c.DB.Where(&models.Settings{Key: models.SettingsKeyDNSServer}).Find(&settings).Error; err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return strings.Split(settings.Value, dnsSep), nil
|
||
|
|
}
|