udeServer/internal/logic/helpers.go

package logic

import (
	"context"
	"database/sql"
	"errors"
	"strings"
	"time"

	"udeServer/internal/apierr"
	"udeServer/internal/model"
	"udeServer/internal/types"
)

func nullString(v string) sql.NullString {
	return sql.NullString{String: v, Valid: v != ""}
}

func nullInt64(v int64) sql.NullInt64 {
	return sql.NullInt64{Int64: v, Valid: v != 0}
}

func stringValue(v sql.NullString) string {
	if !v.Valid {
		return ""
	}

	return v.String
}

func int64Value(v sql.NullInt64) int64 {
	if !v.Valid {
		return 0
	}

	return v.Int64
}

func timeValue(t time.Time) string {
	if t.IsZero() {
		return ""
	}

	return t.Format(time.RFC3339)
}

func checkId(id int64) error {
	if id <= 0 {
		return apierr.BadRequest("id must be greater than 0")
	}

	return nil
}

func findDirectory(ctx context.Context, m model.TDirectoryModel, id int64) (*model.TDirectory, error) {
	data, err := m.FindOne(ctx, id)
	if errors.Is(err, model.ErrNotFound) {
		return nil, apierr.NotFound("directory", id)
	}
	if err != nil {
		return nil, err
	}

	return data, nil
}

func findKey(ctx context.Context, m model.TKeyModel, id int64) (*model.TKey, error) {
	data, err := m.FindOne(ctx, id)
	if errors.Is(err, model.ErrNotFound) {
		return nil, apierr.NotFound("key", id)
	}
	if err != nil {
		return nil, err
	}

	return data, nil
}

func findService(ctx context.Context, m model.TServiceModel, id int64) (*model.TService, error) {
	data, err := m.FindOne(ctx, id)
	if errors.Is(err, model.ErrNotFound) {
		return nil, apierr.NotFound("service", id)
	}
	if err != nil {
		return nil, err
	}

	return data, nil
}

func toDirectoryResp(data *model.TDirectory) *types.DirectoryResp {
	return &types.DirectoryResp{
		Id:        data.Id,
		ServiceId: int64Value(data.ServiceId),
		Directory: stringValue(data.Directory),
	}
}

func toKeyResp(data *model.TKey) *types.KeyResp {
	return &types.KeyResp{
		Id:       data.Id,
		Iden:     stringValue(data.Iden),
		Name:     stringValue(data.Name),
		Content:  stringValue(data.Content),
		Status:   stringValue(data.Status),
		CreateAt: timeValue(data.CreateAt),
	}
}

func toServiceResp(data *model.TService) *types.ServiceResp {
	return &types.ServiceResp{
		Id:          data.Id,
		Name:        stringValue(data.Name),
		Ip:          stringValue(data.Ip),
		Port:        data.Port,
		Iden:        stringValue(data.Iden),
		KeyIden:     stringValue(data.KeyIden),
		KeyAlgo:     stringValue(data.KeyAlgo),
		Status:      stringValue(data.Status),
		MountStatus: stringValue(data.MountStatus),
		Brush:       data.Brush,
		CreateAt:    timeValue(data.CreateAt),
	}
}

func toListServiceResp(data *model.TService) *types.ServiceResp {
	return &types.ServiceResp{
		Id:          data.Id,
		Name:        stringValue(data.Name),
		Ip:          stringValue(data.Ip),
		Port:        data.Port,
		Iden:        maskSensitive(stringValue(data.Iden)),
		KeyIden:     stringValue(data.KeyIden),
		KeyAlgo:     stringValue(data.KeyAlgo),
		Status:      stringValue(data.Status),
		MountStatus: stringValue(data.MountStatus),
		Brush:       data.Brush,
		CreateAt:    timeValue(data.CreateAt),
	}
}

// maskSensitive 掩码展示:前 4 后 4 明文,中间打星;长度不大于 8 时原样返回。
func maskSensitive(value string) string {
	if len(value) <= 8 {
		return value
	}
	return value[:4] + strings.Repeat("*", len(value)-8) + value[len(value)-4:]
}