join.go

// Copyright (C) 2020 Finogeeks Co., Ltd
//
// This program is free software: you can redistribute it and/or  modify
// it under the terms of the GNU Affero General Public License, version 3,
// as published by the Free Software Foundation.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

package syncconsumer

import (
	"context"

	"github.com/finogeeks/ligase/federation/client"
	"github.com/finogeeks/ligase/federation/model/backfilltypes"
	"github.com/finogeeks/ligase/model/service/roomserverapi"
	"github.com/finogeeks/ligase/plugins/message/external"
	"github.com/finogeeks/ligase/skunkworks/gomatrixserverlib"
	log "github.com/finogeeks/ligase/skunkworks/log"
)

func MakeJoin(
	ctx context.Context,
	fedClient *client.FedClientWrap,
	request *roomserverapi.FederationEvent,
	destination string,
) gomatrixserverlib.RespMakeJoin {
	var req external.GetMakeJoinRequest
	if err := json.Unmarshal(request.Extra, &req); err != nil {
		log.Errorf("federation make join unmarshal error: %v", err)
		return gomatrixserverlib.RespMakeJoin{}
	}

	redResp, err := fedClient.MakeJoin(ctx, gomatrixserverlib.ServerName(destination), req.RoomID, req.UserID, req.Ver)
	if err != nil {
		log.Errorf("federation make join error response: %v", err)
	}
	return redResp
}

func SendJoin(
	ctx context.Context,
	fedClient *client.FedClientWrap,
	request *roomserverapi.FederationEvent,
	destination string,
	proc backfilltypes.BackFillProcessor,
) gomatrixserverlib.RespSendJoin {
	var req external.PutSendJoinRequest
	if err := json.Unmarshal(request.Extra, &req); err != nil {
		log.Errorf("federation send join unmarshal error: %v", err)
		return gomatrixserverlib.RespSendJoin{}
	}

	redResp, err := fedClient.SendJoin(ctx, gomatrixserverlib.ServerName(destination), req.RoomID, req.EventID, req.Event)
	if err != nil {
		log.Errorf("federation send join error response: %v", err)
	}
	if len(redResp.StateEvents) > 0 {
		proc.AddRequest(ctx, redResp.StateEvents, false) // TODO: false是因为自动邀请时有可能需要历史消息，这是临时解决方案，看以后有没有更好的处理
	}
	return redResp
}