sshpoke/pkg/convert/convert.go

package convert

import (
	"net"

	"github.com/Neur0toxine/sshpoke/pkg/dto"
	"github.com/Neur0toxine/sshpoke/pkg/plugin/pb"
)

func MessageToAppEvent(event *pb.EventMessage) dto.Event {
	return dto.Event{
		Type: MessageEventTypeToApp(event.GetType()),
		Container: dto.Container{
			ID:         event.GetContainer().GetId(),
			Names:      event.GetContainer().GetNames(),
			IP:         net.ParseIP(event.GetContainer().GetIp()),
			Port:       uint16(event.GetContainer().GetPort()),
			Server:     event.GetContainer().GetServer(),
			RemoteHost: event.GetContainer().GetRemoteHost(),
			Domain:     event.GetContainer().GetDomain(),
		},
	}
}

func AppEventToMessage(event dto.Event) *pb.EventMessage {
	return &pb.EventMessage{
		Type: AppEventTypeToMessage(event.Type),
		Container: &pb.Container{
			Id:         event.Container.ID,
			Names:      event.Container.Names,
			Ip:         event.Container.IP.String(),
			Port:       uint32(event.Container.Port),
			Server:     event.Container.Server,
			RemoteHost: event.Container.RemoteHost,
			Domain:     event.Container.Domain,
		},
	}
}

func MessageToAppEventStatus(val *pb.EventStatusMessage) dto.EventStatus {
	return dto.EventStatus{
		Type:   MessageEventTypeToApp(val.GetType()),
		ID:     val.GetId(),
		Error:  val.GetError(),
		Domain: val.GetDomain(),
	}
}

func AppEventStatusToMessage(status dto.EventStatus) *pb.EventStatusMessage {
	return &pb.EventStatusMessage{
		Type:   AppEventTypeToMessage(status.Type),
		Id:     status.ID,
		Error:  status.Error,
		Domain: status.Domain,
	}
}

func MessageEventTypeToApp(typ pb.EventType) dto.EventType {
	val := dto.EventType(typ.Number())
	if val < dto.EventStart || val > dto.EventUnknown {
		return dto.EventUnknown
	}
	return val
}

func AppEventTypeToMessage(typ dto.EventType) pb.EventType {
	switch typ {
	case 0:
		return pb.EventType_EVENT_START
	case 1:
		return pb.EventType_EVENT_STOP
	case 2:
		return pb.EventType_EVENT_SHUTDOWN
	case 3:
		return pb.EventType_EVENT_ERROR
	default:
		return pb.EventType_EVENT_UNKNOWN
	}
}