package logic

import (
	"context"
	"time"

	"muyu-apiserver/rpc/production/internal/svc"
	"muyu-apiserver/rpc/production/pb"

	"github.com/zeromicro/go-zero/core/logx"
)

type GetShareLinkLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewGetShareLinkLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetShareLinkLogic {
	return &GetShareLinkLogic{ctx: ctx, svcCtx: svcCtx, Logger: logx.WithContext(ctx)}
}

func (l *GetShareLinkLogic) GetShareLink(in *pb.GetShareLinkReq) (*pb.ShareLinkInfo, error) {
	link, err := l.svcCtx.ShareLinkModel.FindByShortCode(l.ctx, in.ShortCode)
	if err != nil {
		return nil, ErrLinkNotFound
	}

	// Check if expired
	if time.Now().After(link.ExpiresAt) {
		return nil, ErrLinkExpired
	}

	info := &pb.ShareLinkInfo{
		LinkId:        link.LinkId,
		TenantId:      link.TenantId,
		PlanId:        link.PlanId,
		ShortCode:     link.ShortCode,
		Token:         link.Token,
		FactoryType:   link.FactoryType,
		HideSensitive: link.HideSensitive,
		Status:        link.Status,
		ClickCount:    link.ClickCount,
		RejectReason:  link.RejectReason,
		ExpiresAt:     link.ExpiresAt.Format("2006-01-02 15:04:05"),
		CreatedBy:     link.CreatedBy,
		CreatedAt:     link.CreatedAt.Format("2006-01-02 15:04:05"),
		UpdatedAt:     link.UpdatedAt.Format("2006-01-02 15:04:05"),
	}
	if link.ClickedAt.Valid {
		info.ClickedAt = link.ClickedAt.Time.Format("2006-01-02 15:04:05")
	}
	if link.RespondedAt.Valid {
		info.RespondedAt = link.RespondedAt.Time.Format("2006-01-02 15:04:05")
	}

	return info, nil
}