pictures-grpc-mapper.go

package goautowp

import (
	"time"

	"github.com/autowp/goautowp/pictures"
	"github.com/autowp/goautowp/query"
	"github.com/autowp/goautowp/schema"
	"github.com/autowp/goautowp/util"
)

func extractPictureModerVoteTemplate(tpl *schema.PictureModerVoteTemplateRow) *ModerVoteTemplate {
	return &ModerVoteTemplate{
		Id:      tpl.ID,
		UserId:  tpl.UserID,
		Message: tpl.Message,
		Vote:    int32(tpl.Vote),
	}
}

func extractPictureItemType(pictureItemType schema.PictureItemType) PictureItemType {
	switch pictureItemType {
	case 0:
		return PictureItemType_PICTURE_ITEM_UNKNOWN
	case schema.PictureItemContent:
		return PictureItemType_PICTURE_ITEM_CONTENT
	case schema.PictureItemAuthor:
		return PictureItemType_PICTURE_ITEM_AUTHOR
	case schema.PictureItemCopyrights:
		return PictureItemType_PICTURE_ITEM_COPYRIGHTS
	}

	return PictureItemType_PICTURE_ITEM_UNKNOWN
}

func convertPictureItemType(pictureItemType PictureItemType) schema.PictureItemType {
	switch pictureItemType {
	case PictureItemType_PICTURE_ITEM_UNKNOWN:
		return 0
	case PictureItemType_PICTURE_ITEM_CONTENT:
		return schema.PictureItemContent
	case PictureItemType_PICTURE_ITEM_AUTHOR:
		return schema.PictureItemAuthor
	case PictureItemType_PICTURE_ITEM_COPYRIGHTS:
		return schema.PictureItemCopyrights
	}

	return 0
}

func convertPictureStatus(status PictureStatus) schema.PictureStatus {
	switch status {
	case PictureStatus_PICTURE_STATUS_UNKNOWN:
		return ""
	case PictureStatus_PICTURE_STATUS_ACCEPTED:
		return schema.PictureStatusAccepted
	case PictureStatus_PICTURE_STATUS_REMOVING:
		return schema.PictureStatusRemoving
	case PictureStatus_PICTURE_STATUS_REMOVED:
		return schema.PictureStatusRemoved
	case PictureStatus_PICTURE_STATUS_INBOX:
		return schema.PictureStatusInbox
	}

	return ""
}

func convertPictureItemOptions(in *PictureItemListOptions) (*query.PictureItemListOptions, error) {
	if in == nil {
		return nil, nil //nolint: nilnil
	}

	result := query.PictureItemListOptions{
		PictureID:               in.GetPictureId(),
		ItemID:                  in.GetItemId(),
		TypeID:                  convertPictureItemType(in.GetTypeId()),
		PerspectiveID:           in.GetPerspectiveId(),
		ExcludePerspectiveID:    in.GetExcludePerspectiveId(),
		ExcludeAncestorOrSelfID: in.GetExcludeAncestorOrSelfId(),
		HasNoPerspectiveID:      in.GetHasNoPerspectiveId(),
		ItemVehicleType:         convertItemVehicleTypeListOptions(in.GetItemVehicleType()),
	}

	var err error

	result.Item, err = convertItemListOptions(in.GetItem())
	if err != nil {
		return nil, err
	}

	result.Pictures, err = convertPictureListOptions(in.GetPictures())
	if err != nil {
		return nil, err
	}

	result.ItemParentCacheAncestor, err = convertItemParentCacheListOptions(in.GetItemParentCacheAncestor())
	if err != nil {
		return nil, err
	}

	return &result, nil
}

func convertItemVehicleTypeListOptions(in *ItemVehicleTypeListOptions) *query.ItemVehicleTypeListOptions {
	if in == nil {
		return nil
	}

	result := query.ItemVehicleTypeListOptions{
		VehicleTypeID: in.GetVehicleTypeId(),
	}

	return &result
}

func convertPictureListOptions(in *PictureListOptions) (*query.PictureListOptions, error) {
	if in == nil {
		return nil, nil //nolint: nilnil
	}

	result := query.PictureListOptions{
		ID:                    in.GetId(),
		Status:                convertPictureStatus(in.GetStatus()),
		AcceptedInDays:        in.GetAcceptedInDays(),
		OwnerID:               in.GetOwnerId(),
		Identity:              in.GetIdentity(),
		HasNoComments:         in.GetHasNoComments(),
		HasPoint:              in.GetHasPoint(),
		HasNoPoint:            in.GetHasNoPoint(),
		HasNoPictureItem:      in.GetHasNoPictureItem(),
		HasNoReplacePicture:   in.GetHasNoReplacePicture(),
		HasNoPictureModerVote: in.GetHasNoPictureModerVote(),
		CommentTopic:          convertCommentTopicListOptions(in.GetCommentTopic()),
		PictureModerVote:      convertPictureModerVoteListOptions(in.GetPictureModerVote()),
		HasSpecialName:        in.GetHasSpecialName(),
	}

	var err error

	result.DfDistance, err = convertDfDistanceListOptions(in.GetDfDistance())
	if err != nil {
		return nil, err
	}

	inStatuses := in.GetStatuses()
	if len(inStatuses) > 0 {
		statuses := make([]schema.PictureStatus, 0, len(inStatuses))
		for _, status := range inStatuses {
			statuses = append(statuses, convertPictureStatus(status))
		}

		result.Statuses = statuses
	}

	addDate := in.GetAddDate()
	if addDate != nil {
		result.AddDate = &util.Date{
			Year:  int(addDate.GetYear()),
			Month: time.Month(addDate.GetMonth()),
			Day:   int(addDate.GetDay()),
		}
	}

	acceptDate := in.GetAcceptDate()
	if acceptDate != nil {
		result.AcceptDate = &util.Date{
			Year:  int(acceptDate.GetYear()),
			Month: time.Month(acceptDate.GetMonth()),
			Day:   int(acceptDate.GetDay()),
		}
	}

	addedFrom := in.GetAddedFrom()
	if addedFrom != nil {
		result.AddedFrom = &util.Date{
			Year:  int(addedFrom.GetYear()),
			Month: time.Month(addedFrom.GetMonth()),
			Day:   int(addedFrom.GetDay()),
		}
	}

	result.PictureItem, err = convertPictureItemOptions(in.GetPictureItem())
	if err != nil {
		return nil, err
	}

	result.ReplacePicture, err = convertPictureListOptions(in.GetReplacePicture())
	if err != nil {
		return nil, err
	}

	return &result, nil
}

func convertPictureModerVoteListOptions(in *PictureModerVoteListOptions) *query.PictureModerVoteListOptions {
	if in == nil {
		return nil
	}

	return &query.PictureModerVoteListOptions{
		VoteGtZero:  in.GetVoteGtZero(),
		VoteLteZero: in.GetVoteLteZero(),
	}
}

func convertDfDistanceListOptions(in *DfDistanceListOptions) (*query.DfDistanceListOptions, error) {
	if in == nil {
		return nil, nil //nolint: nilnil
	}

	var err error

	result := query.DfDistanceListOptions{}

	result.DstPicture, err = convertPictureListOptions(in.GetDstPicture())
	if err != nil {
		return nil, err
	}

	return &result, nil
}

func convertCommentTopicListOptions(in *CommentTopicListOptions) *query.CommentTopicListOptions {
	if in == nil {
		return nil
	}

	result := query.CommentTopicListOptions{
		MessagesGtZero: in.GetMessagesGtZero(),
	}

	return &result
}

func convertPictureFields(fields *PictureFields) pictures.PictureFields {
	return pictures.PictureFields{
		NameText: fields.GetNameText(),
	}
}

func convertPicturesOrder(order PicturesRequest_Order) pictures.OrderBy {
	switch order {
	case PicturesRequest_ORDER_NONE:
		return pictures.OrderByNone
	case PicturesRequest_ORDER_ADD_DATE_DESC:
		return pictures.OrderByAddDateDesc
	case PicturesRequest_ORDER_ADD_DATE_ASC:
		return pictures.OrderByAddDateAsc
	case PicturesRequest_ORDER_RESOLUTION_DESC:
		return pictures.OrderByResolutionDesc
	case PicturesRequest_ORDER_RESOLUTION_ASC:
		return pictures.OrderByResolutionAsc
	case PicturesRequest_ORDER_FILESIZE_DESC:
		return pictures.OrderByFilesizeDesc
	case PicturesRequest_ORDER_FILESIZE_ASC:
		return pictures.OrderByFilesizeAsc
	case PicturesRequest_ORDER_COMMENTS:
		return pictures.OrderByComments
	case PicturesRequest_ORDER_VIEWS:
		return pictures.OrderByViews
	case PicturesRequest_ORDER_MODER_VOTES:
		return pictures.OrderByModerVotes
	case PicturesRequest_ORDER_REMOVING_DATE:
		return pictures.OrderByRemovingDate
	case PicturesRequest_ORDER_LIKES:
		return pictures.OrderByLikes
	case PicturesRequest_ORDER_DISLIKES:
		return pictures.OrderByDislikes
	case PicturesRequest_ORDER_ACCEPT_DATETIME_DESC:
		return pictures.OrderByAcceptDatetimeDesc
	case PicturesRequest_ORDER_PERSPECTIVES:
		return pictures.OrderByPerspectives
	case PicturesRequest_ORDER_DF_DISTANCE_SIMILARITY:
		return pictures.OrderByDfDistanceSimilarity
	}

	return pictures.OrderByNone
}