124 lines
2.7 KiB
Go
124 lines
2.7 KiB
Go
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
|
|
}
|