app_conf.go

package main

import (
	"github.com/labstack/gommon/log"
	"github.com/twsnmp/twsnmpfk/datastore"
	"github.com/twsnmp/twsnmpfk/i18n"
	"github.com/twsnmp/twsnmpfk/notify"
	wails "github.com/wailsapp/wails/v2/pkg/runtime"
)

// GetMapConf returns map config
func (a *App) GetMapConf() datastore.MapConfEnt {
	return datastore.MapConf
}

// UpdateMapConf save map config
func (a *App) UpdateMapConf(m datastore.MapConfEnt) bool {
	datastore.RestartSnmpTrapd = datastore.MapConf.SnmpMode != m.SnmpMode ||
		datastore.MapConf.Community != m.Community ||
		datastore.MapConf.SnmpUser != m.SnmpUser ||
		datastore.MapConf.SnmpPassword != m.SnmpPassword
	if m.MapName != datastore.MapConf.MapName {
		a.setMenu()
	}
	datastore.MapConf = m

	return datastore.SaveMapConf() == nil
}

// GetMapName returns map name
func (a *App) GetMapName() string {
	return datastore.MapConf.MapName
}

// GetNotifyConf returns notify config
func (a *App) GetNotifyConf() datastore.NotifyConfEnt {
	return datastore.NotifyConf
}

// UpdateNotifyConf save notify config
func (a *App) UpdateNotifyConf(n datastore.NotifyConfEnt) bool {
	datastore.NotifyConf = n
	return datastore.SaveNotifyConf() == nil
}

// TestNotifyConf test notfiy
func (a *App) TestNotifyConf(n datastore.NotifyConfEnt) bool {
	return notify.SendTestMail(&n) == nil
}

// TestLine test line notify
func (a *App) TestLine(n datastore.NotifyConfEnt) bool {
	return notify.SendLine(&n, n.Subject+i18n.Trans("(LINE test)"), 8522, 16581266) == nil
}

// GetAIConf returns AI config
func (a *App) GetAIConf() datastore.AIConfEnt {
	return datastore.AIConf
}

// UpdateAIConf save AI config
func (a *App) UpdateAIConf(ai datastore.AIConfEnt) bool {
	datastore.AIConf = ai
	return datastore.SaveAIConf() == nil
}

// GetLocConf returns Loc config
func (a *App) GetLocConf() datastore.LocConfEnt {
	return datastore.LocConf
}

// UpdateLocConf save Loc config
func (a *App) UpdateLocConf(loc datastore.LocConfEnt) bool {
	datastore.LocConf = loc
	return datastore.SaveLocConf() == nil
}

// Backupは、データベースのバックアップを作成します。
func (a *App) Backup() bool {
	f, err := wails.SaveFileDialog(a.ctx, wails.SaveDialogOptions{
		Title:           i18n.Trans("Backup file"),
		DefaultFilename: "twsnmpfk.db",
		Filters: []wails.FileFilter{{
			DisplayName: "*.db",
			Pattern:     "*.db",
		}},
		CanCreateDirectories: true,
	})
	if err != nil || f == "" {
		return false
	}
	err = datastore.BackupDB(f)
	if err != nil {
		log.Printf("backup database err=%v", err)
		return false
	}
	datastore.AddEventLog(&datastore.EventLogEnt{
		Type:  "user",
		Level: "info",
		Event: i18n.Trans("Backup done"),
	})
	return true
}

func (a *App) GetMIBModules() []*datastore.MIBModuleEnt {
	return datastore.MIBModules
}

// GetIconsは、カスタムアイコンのリストを返します。
func (a *App) GetIcons() []*datastore.IconEnt {
	return datastore.GetIcons()
}

// UpdateIconは、カスタムアイコンを追加または更新します。
func (a *App) UpdateIcon(icon datastore.IconEnt) bool {
	if err := datastore.AddOrUpdateIcon(&icon); err != nil {
		log.Printf("UpdateIcon err=%v", err)
		return false
	}
	return true
}

// DeleteIconは、カスタムアイコンを削除します。
func (a *App) DeleteIcon(icon string) bool {
	if err := datastore.DeleteIcon(icon); err != nil {
		log.Printf("UpdateIcon err=%v", err)
		return false
	}
	return true
}

// GetSshdPublicKeysは、SSHサーバーにアクセスを許可するホストの公開鍵を取得
func (a *App) GetSshdPublicKeys() string {
	return datastore.GetSshdPublicKeys()
}

// SaveSshdPublicKeysは、SSHサーバーにアクセスを許可するホストの公開鍵を保存
func (a *App) SaveSshdPublicKeys(pk string) bool {
	if err := datastore.SaveSshdPublicKeys(pk); err != nil {
		log.Printf("SaveSshdPublicKeys err=%v", err)
	}
	return true
}

// GetMySSHPublicKeyは、自分のSSH公開鍵を取得
func (a *App) GetMySSHPublicKey() string {
	r, err := datastore.GetSSHPublicKey()
	if err != nil {
		log.Printf("GetMySSHPublicKeys err=%v", err)
	}
	return r
}

// InitMySSHKeyは、自分のSSH秘密鍵を再作成
func (a *App) InitMySSHKey() bool {
	result, err := wails.MessageDialog(a.ctx, wails.MessageDialogOptions{
		Type:          wails.QuestionDialog,
		Title:         i18n.Trans("Confirm init ssh key"),
		Message:       i18n.Trans("Do you want to init?"),
		Buttons:       []string{"Yes", "No"},
		DefaultButton: "No",
	})
	if err != nil || result == "No" {
		return false
	}
	datastore.AddEventLog(&datastore.EventLogEnt{
		Type:  "user",
		Level: "info",
		Event: i18n.Trans("Init ssh private key"),
	})
	return len(datastore.GenSSHPrivateKey()) > 0
}