release v1.2.0

server/service/asset.go

@@ -0,0 +1,60 @@
+package service
+
+import (
+	"next-terminal/server/config"
+	"next-terminal/server/repository"
+	"next-terminal/server/utils"
+)
+
+type AssetService struct {
+	assetRepository *repository.AssetRepository
+}
+
+func NewAssetService(assetRepository *repository.AssetRepository) *AssetService {
+	return &AssetService{assetRepository: assetRepository}
+}
+
+func (r AssetService) Encrypt() error {
+	items, err := r.assetRepository.FindAll()
+	if err != nil {
+		return err
+	}
+	for i := range items {
+		item := items[i]
+		if item.Encrypted {
+			continue
+		}
+		if err := r.assetRepository.Encrypt(&item, config.GlobalCfg.EncryptionPassword); err != nil {
+			return err
+		}
+		if err := r.assetRepository.UpdateById(&item, item.ID); err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func (r AssetService) CheckStatus(accessGatewayId string, ip string, port int) (active bool, err error) {
+	if accessGatewayId != "" && accessGatewayId != "-" {
+		g, e1 := accessGatewayService.GetGatewayAndReconnectById(accessGatewayId)
+		if err != nil {
+			return false, e1
+		}
+
+		uuid := utils.UUID()
+		exposedIP, exposedPort, e2 := g.OpenSshTunnel(uuid, ip, port)
+		if e2 != nil {
+			return false, e2
+		}
+		defer g.CloseSshTunnel(uuid)
+
+		if g.Connected {
+			active, err = utils.Tcping(exposedIP, exposedPort)
+		} else {
+			active = false
+		}
+	} else {
+		active, err = utils.Tcping(ip, port)
+	}
+	return active, err
+}