package custom

import (
	"net/url"
	"strings"

	. "openreplay/backend/pkg/messages"
)

func getURLExtention(URL string) string {
	u, err := url.Parse(URL)
	if err != nil {
		return ""
	}
	i := strings.LastIndex(u.Path, ".")
	return u.Path[i+1:]
}

func getResourceType(initiator string, URL string) string {
	switch initiator {
	case "xmlhttprequest", "fetch":
		return "fetch"
	case "img":
		return "img"
	default:
		switch getURLExtention(URL) {
		case "css":
			return "stylesheet"
		case "js":
			return "script"
		case "png", "gif", "jpg", "jpeg", "svg":
			return "img"
		case "mp4", "mkv", "ogg", "webm", "avi", "mp3":
			return "media"
		default:
			return "other"
		}
	}
}

type EventMapper struct{}

func (b *EventMapper) Build() Message {
	return nil
}

func (b *EventMapper) Handle(message Message, messageID uint64, timestamp uint64) Message {
	switch msg := message.(type) {
	case *MouseClick:
		if msg.Label != "" {
			return &ClickEvent{
				MessageID:      messageID,
				Label:          msg.Label,
				HesitationTime: msg.HesitationTime,
				Timestamp:      timestamp,
				Selector:       msg.Selector,
			}
		}
	case *JSException:
		return &ErrorEvent{
			MessageID: messageID,
			Timestamp: timestamp,
			Source:    "js_exception",
			Name:      msg.Name,
			Message:   msg.Message,
			Payload:   msg.Payload,
		}
	case *ResourceTiming:
		return &ResourceEvent{
			MessageID:       messageID,
			Timestamp:       msg.Timestamp,
			Duration:        msg.Duration,
			TTFB:            msg.TTFB,
			HeaderSize:      msg.HeaderSize,
			EncodedBodySize: msg.EncodedBodySize,
			DecodedBodySize: msg.DecodedBodySize,
			URL:             msg.URL,
			Type:            getResourceType(msg.Initiator, msg.URL),
			Success:         msg.Duration != 0,
		}
	case *RawCustomEvent:
		return &CustomEvent{
			MessageID: messageID,
			Timestamp: timestamp,
			Name:      msg.Name,
			Payload:   msg.Payload,
		}
	case *CustomIssue:
		return &IssueEvent{
			Type:          "custom",
			Timestamp:     timestamp,
			MessageID:     messageID,
			ContextString: msg.Name,
			Payload:       msg.Payload,
		}
	case *Fetch:
		return &FetchEvent{
			MessageID: messageID,
			Timestamp: msg.Timestamp,
			Method:    msg.Method,
			URL:       msg.URL,
			Request:   msg.Request,
			Response:  msg.Response,
			Status:    msg.Status,
			Duration:  msg.Duration,
		}
	case *GraphQL:
		return &GraphQLEvent{
			MessageID:     messageID,
			Timestamp:     timestamp,
			OperationKind: msg.OperationKind,
			OperationName: msg.OperationName,
			Variables:     msg.Variables,
			Response:      msg.Response,
		}
	case *StateAction:
		return &StateActionEvent{
			MessageID: messageID,
			Timestamp: timestamp,
			Type:      msg.Type,
		}
	}
	return nil
}