chore: updates

This commit is contained in:
Simon Ding
2024-07-17 21:55:25 +08:00
parent 1fb6abc299
commit 135e225df2
4 changed files with 22 additions and 14 deletions

View File

@@ -42,11 +42,22 @@ func Open() (*Client, error) {
ent: client,
}
c.generateJwtSerectIfNotExist()
c.generateDefaultLocalStorage()
return c, nil
}
func (c *Client) init() {
c.generateJwtSerectIfNotExist()
if err := c.generateDefaultLocalStorage(); err != nil {
log.Errorf("generate default storage: %v", err)
}
downloadDir := c.GetSetting(SettingDownloadDir)
if downloadDir == "" {
log.Infof("set default download dir")
c.SetSetting(downloadDir, "/downloads")
}
}
func (c *Client) generateJwtSerectIfNotExist() {
v := c.GetSetting(JwtSerectKey)
if v == "" {
@@ -56,12 +67,13 @@ func (c *Client) generateJwtSerectIfNotExist() {
}
}
func (c *Client) generateDefaultLocalStorage() {
func (c *Client) generateDefaultLocalStorage() error {
n, _ := c.ent.Storage.Query().Count(context.TODO())
if n != 0 {
return
return nil
}
c.AddStorage(&StorageInfo{
log.Infof("add default storage")
return c.AddStorage(&StorageInfo{
Name: "local",
Implementation: "local",
Default: true,