package protocol import ( "context" "fmt" "strings" "testing" "github.com/stretchr/testify/suite" "github.com/status-im/status-go/eth-node/crypto" "github.com/status-im/status-go/eth-node/types" "github.com/status-im/status-go/protocol/common" "github.com/status-im/status-go/protocol/communities" "github.com/status-im/status-go/protocol/protobuf" "github.com/status-im/status-go/protocol/requests" "github.com/status-im/status-go/protocol/urls" ) func TestMessengerShareUrlsSuite(t *testing.T) { suite.Run(t, new(MessengerShareUrlsSuite)) } type MessengerShareUrlsSuite struct { MessengerBaseTestSuite } func (s *MessengerShareUrlsSuite) createCommunity() *communities.Community { description := &requests.CreateCommunity{ Membership: protobuf.CommunityPermissions_AUTO_ACCEPT, Name: "status", Color: "#ffffff", Description: "status community description", } // Create an community chat response, err := s.m.CreateCommunity(description, false) s.Require().NoError(err) s.Require().NotNil(response) return response.Communities()[0] } func (s *MessengerShareUrlsSuite) createContact() (*Messenger, *Contact) { theirMessenger := s.newMessenger() _, err := theirMessenger.Start() s.Require().NoError(err) contactID := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey)) ensName := "blah.stateofus.eth" s.Require().NoError(s.m.ENSVerified(contactID, ensName)) response, err := s.m.AddContact(context.Background(), &requests.AddContact{ID: contactID}) s.Require().NoError(err) s.Require().NotNil(response) s.Require().Len(response.Contacts, 1) contact := response.Contacts[0] s.Require().Equal(ensName, contact.EnsName) s.Require().True(contact.ENSVerified) return theirMessenger, contact } func (s *MessengerShareUrlsSuite) createCommunityWithChannel() (*communities.Community, *protobuf.CommunityChat, string) { community := s.createCommunity() chat := &protobuf.CommunityChat{ Permissions: &protobuf.CommunityPermissions{ Access: protobuf.CommunityPermissions_AUTO_ACCEPT, }, Identity: &protobuf.ChatIdentity{ DisplayName: "status-core", Emoji: "😎", Description: "status-core community chat", }, } response, err := s.m.CreateCommunityChat(community.ID(), chat) s.Require().NoError(err) s.Require().NotNil(response) s.Require().Len(response.Communities(), 1) s.Require().Len(response.Chats(), 1) community = response.Communities()[0] s.Require().Len(community.Chats(), 1) var channelID string var channel *protobuf.CommunityChat for key, value := range community.Chats() { channelID = key channel = value break } s.Require().NotNil(channel) return community, channel, channelID } func (s *MessengerShareUrlsSuite) TestDecodeEncodeDataURL() { ts := [][]byte{ []byte("test data 123"), []byte("test data 123test data 123test data 123test data 123test data 123"), } for i := range ts { encodedData, err := urls.EncodeDataURL(ts[i]) s.Require().NoError(err) decodedData, err := urls.DecodeDataURL(encodedData) s.Require().NoError(err) s.Require().Equal(ts[i], decodedData) } } func (s *MessengerShareUrlsSuite) TestSerializePublicKey() { key, err := crypto.GenerateKey() s.Require().NoError(err) serializedKey, err := s.m.SerializePublicKey(crypto.CompressPubkey(&key.PublicKey)) s.Require().NoError(err) s.Require().Len(serializedKey, 49) s.Require().True(strings.HasPrefix(serializedKey, "zQ3sh")) } func (s *MessengerShareUrlsSuite) TestDeserializePublicKey() { serializedKey := "zQ3shPyZJnxZK4Bwyx9QsaksNKDYTPmpwPvGSjMYVHoXHeEgB" shortKey, err := s.m.DeserializePublicKey(serializedKey) s.Require().NoError(err) s.Require().Len(shortKey, 33) s.Require().True(strings.HasPrefix(shortKey.String(), "0x")) } func (s *MessengerShareUrlsSuite) TestParseWrongUrls() { urls := map[string]string{ "https://status.appc/#zQ3shYSHp7GoiXaauJMnDcjwU2yNjdzpXLosAWapPS4CFxc11": "unhandled shared url", "https://status.app/cc#zQ3shYSHp7GoiXaauJMnDcjwU2yNjdzpXLosAWapPS4CFxc11": "unhandled shared url", "https://status.app/a#zQ3shYSHp7GoiXaauJMnDcjwU2yNjdzpXLosAWapPS4CFxc11": "unhandled shared url", "https://status.app/u/": "url should contain at least one `#` separator", "https://status.im/u#zQ3shYSHp7GoiXaauJMnDcjwU2yNjdzpXLosAWapPS4CFxc11": "url should start with 'https://status.app'", } for url, expectedError := range urls { urlData, err := s.m.ParseSharedURL(url) s.Require().Error(err) s.Require().True(strings.HasPrefix(err.Error(), expectedError)) s.Require().Nil(urlData) } } func (s *MessengerShareUrlsSuite) TestShareCommunityURLWithChatKey() { community := s.createCommunity() url, err := s.m.ShareCommunityURLWithChatKey(community.ID()) s.Require().NoError(err) shortKey, err := s.m.SerializePublicKey(community.ID()) s.Require().NoError(err) expectedURL := fmt.Sprintf("%s/c#%s", baseShareURL, shortKey) s.Require().Equal(expectedURL, url) } func (s *MessengerShareUrlsSuite) TestParseCommunityURLWithChatKey() { community := s.createCommunity() shortKey, err := s.m.SerializePublicKey(community.ID()) s.Require().NoError(err) url := fmt.Sprintf("%s/c#%s", baseShareURL, shortKey) urlData, err := s.m.ParseSharedURL(url) s.Require().NoError(err) s.Require().NotNil(urlData) s.Require().NotNil(urlData.Community) s.Require().Equal(community.Identity().DisplayName, urlData.Community.DisplayName) s.Require().Equal(community.DescriptionText(), urlData.Community.Description) s.Require().Equal(uint32(community.MembersCount()), urlData.Community.MembersCount) s.Require().Equal(community.Identity().GetColor(), urlData.Community.Color) s.Require().Equal(community.TagsIndices(), urlData.Community.TagIndices) } func (s *MessengerShareUrlsSuite) TestShareCommunityURLWithData() { community := s.createCommunity() url, err := s.m.ShareCommunityURLWithData(community.ID()) s.Require().NoError(err) communityData, chatKey, err := s.m.prepareEncodedCommunityData(community) s.Require().NoError(err) expectedURL := fmt.Sprintf("%s/c/%s#%s", baseShareURL, communityData, chatKey) s.Require().Equal(expectedURL, url) } func (s *MessengerShareUrlsSuite) TestParseCommunityURLWithData() { url := "https://status.app/c/iyKACkQKB0Rvb2RsZXMSJ0NvbG9yaW5nIHRoZSB3b3JsZCB3aXRoIGpveSDigKIg4bSXIOKAohiYohsiByMxMzFEMkYqAwEhMwM=#zQ3shYSHp7GoiXaauJMnDcjwU2yNjdzpXLosAWapPS4CFxc11" urlData, err := s.m.ParseSharedURL(url) s.Require().NoError(err) s.Require().NotNil(urlData) s.Require().NotNil(urlData.Community) s.Require().Equal("0x02a3d2fdb9ac335917bf9d46b38d7496c00bbfadbaf832e8aa61d13ac2b4452084", urlData.Community.CommunityID) s.Require().Equal("Doodles", urlData.Community.DisplayName) s.Require().Equal("Coloring the world with joy • ᴗ •", urlData.Community.Description) s.Require().Equal(uint32(446744), urlData.Community.MembersCount) s.Require().Equal("#131D2F", urlData.Community.Color) s.Require().Equal([]uint32{1, 33, 51}, urlData.Community.TagIndices) } func (s *MessengerShareUrlsSuite) TestShareAndParseCommunityURLWithData() { community := s.createCommunity() url, err := s.m.ShareCommunityURLWithData(community.ID()) s.Require().NoError(err) urlData, err := s.m.ParseSharedURL(url) s.Require().NoError(err) s.Require().Equal(community.Identity().DisplayName, urlData.Community.DisplayName) s.Require().Equal(community.DescriptionText(), urlData.Community.Description) s.Require().Equal(uint32(community.MembersCount()), urlData.Community.MembersCount) s.Require().Equal(community.Identity().GetColor(), urlData.Community.Color) s.Require().Equal(community.TagsIndices(), urlData.Community.TagIndices) } func (s *MessengerShareUrlsSuite) TestShareCommunityChannelURLWithChatKey() { community := s.createCommunity() channelID := "003cdcd5-e065-48f9-b166-b1a94ac75a11" request := &requests.CommunityChannelShareURL{ CommunityID: community.ID(), ChannelID: channelID, } url, err := s.m.ShareCommunityChannelURLWithChatKey(request) s.Require().NoError(err) shortKey, err := s.m.SerializePublicKey(community.ID()) s.Require().NoError(err) expectedURL := fmt.Sprintf("%s/cc/%s#%s", baseShareURL, channelID, shortKey) s.Require().Equal(expectedURL, url) } func (s *MessengerShareUrlsSuite) TestParseCommunityChannelURLWithChatKey() { community, channel, channelID := s.createCommunityWithChannel() shortKey, err := s.m.SerializePublicKey(community.ID()) s.Require().NoError(err) url := fmt.Sprintf("%s/cc/%s#%s", baseShareURL, channelID, shortKey) urlData, err := s.m.ParseSharedURL(url) s.Require().NoError(err) s.Require().NotNil(urlData) s.Require().NotNil(urlData.Community) s.Require().Equal(community.Identity().DisplayName, urlData.Community.DisplayName) s.Require().Equal(community.DescriptionText(), urlData.Community.Description) s.Require().Equal(uint32(community.MembersCount()), urlData.Community.MembersCount) s.Require().Equal(community.Identity().GetColor(), urlData.Community.Color) s.Require().Equal(community.TagsIndices(), urlData.Community.TagIndices) s.Require().NotNil(urlData.Channel) s.Require().Equal(channel.Identity.Emoji, urlData.Channel.Emoji) s.Require().Equal(channel.Identity.DisplayName, urlData.Channel.DisplayName) s.Require().Equal(channel.Identity.Color, urlData.Channel.Color) } func (s *MessengerShareUrlsSuite) TestShareCommunityChannelURLWithData() { community, channel, channelID := s.createCommunityWithChannel() request := &requests.CommunityChannelShareURL{ CommunityID: community.ID(), ChannelID: channelID, } url, err := s.m.ShareCommunityChannelURLWithData(request) s.Require().NoError(err) communityChannelData, chatKey, err := s.m.prepareEncodedCommunityChannelData(community, channel, channelID) s.Require().NoError(err) expectedURL := fmt.Sprintf("%s/cc/%s#%s", baseShareURL, communityChannelData, chatKey) s.Require().Equal(expectedURL, url) } func (s *MessengerShareUrlsSuite) TestParseCommunityChannelURLWithData() { url := "https://status.app/cc/G54AAKwObLdpiGjXnckYzRcOSq0QQAS_CURGfqVU42ceGHCObstUIknTTZDOKF3E8y2MSicncpO7fTskXnoACiPKeejvjtLTGWNxUhlT7fyQS7Jrr33UVHluxv_PLjV2ePGw5GQ33innzeK34pInIgUGs5RjdQifMVmURalxxQKwiuoY5zwIjixWWRHqjHM=#zQ3shYSHp7GoiXaauJMnDcjwU2yNjdzpXLosAWapPS4CFxc11" urlData, err := s.m.ParseSharedURL(url) s.Require().NoError(err) s.Require().NotNil(urlData) s.Require().NotNil(urlData.Community) s.Require().Equal("Doodles", urlData.Community.DisplayName) s.Require().NotNil(urlData.Channel) s.Require().Equal("🍿", urlData.Channel.Emoji) s.Require().Equal("design", urlData.Channel.DisplayName) s.Require().Equal("#131D2F", urlData.Channel.Color) } func (s *MessengerShareUrlsSuite) TestShareAndParseCommunityChannelURLWithData() { community, channel, channelID := s.createCommunityWithChannel() request := &requests.CommunityChannelShareURL{ CommunityID: community.ID(), ChannelID: channelID, } url, err := s.m.ShareCommunityChannelURLWithData(request) s.Require().NoError(err) urlData, err := s.m.ParseSharedURL(url) s.Require().NoError(err) s.Require().Equal(community.Identity().DisplayName, urlData.Community.DisplayName) s.Require().Equal(community.DescriptionText(), urlData.Community.Description) s.Require().Equal(uint32(community.MembersCount()), urlData.Community.MembersCount) s.Require().Equal(community.Identity().GetColor(), urlData.Community.Color) s.Require().Equal(community.TagsIndices(), urlData.Community.TagIndices) s.Require().NotNil(urlData.Channel) s.Require().Equal(channel.Identity.Emoji, urlData.Channel.Emoji) s.Require().Equal(channel.Identity.DisplayName, urlData.Channel.DisplayName) s.Require().Equal(channel.Identity.Color, urlData.Channel.Color) } func (s *MessengerShareUrlsSuite) TestShareUserURLWithChatKey() { _, contact := s.createContact() url, err := s.m.ShareUserURLWithChatKey(contact.ID) s.Require().NoError(err) publicKey, err := common.HexToPubkey(contact.ID) s.Require().NoError(err) shortKey, err := s.m.SerializePublicKey(crypto.CompressPubkey(publicKey)) s.Require().NoError(err) expectedURL := fmt.Sprintf("%s/u#%s", baseShareURL, shortKey) s.Require().Equal(expectedURL, url) } func (s *MessengerShareUrlsSuite) TestParseUserURLWithChatKey() { _, contact := s.createContact() publicKey, err := common.HexToPubkey(contact.ID) s.Require().NoError(err) shortKey, err := s.m.SerializePublicKey(crypto.CompressPubkey(publicKey)) s.Require().NoError(err) url := fmt.Sprintf("%s/u#%s", baseShareURL, shortKey) urlData, err := s.m.ParseSharedURL(url) s.Require().NoError(err) s.Require().NotNil(urlData) s.Require().NotNil(urlData.Contact) s.Require().Equal(contact.DisplayName, urlData.Contact.DisplayName) s.Require().Equal(contact.Bio, urlData.Contact.Description) } func (s *MessengerShareUrlsSuite) TestShareUserURLWithENS() { _, contact := s.createContact() url, err := s.m.ShareUserURLWithENS(contact.ID) s.Require().NoError(err) expectedURL := fmt.Sprintf("%s/u#%s", baseShareURL, contact.EnsName) s.Require().Equal(expectedURL, url) } // TODO: ens in the next ticket // func (s *MessengerShareUrlsSuite) TestParseUserURLWithENS() { // _, contact := s.createContact() // url := fmt.Sprintf("%s/u#%s", baseShareURL, contact.EnsName) // urlData, err := s.m.ParseSharedURL(url) // s.Require().NoError(err) // s.Require().NotNil(urlData) // s.Require().NotNil(urlData.Contact) // s.Require().Equal(contact.DisplayName, urlData.Contact.DisplayName) // s.Require().Equal(contact.Bio, urlData.Contact.DisplayName) // } func (s *MessengerShareUrlsSuite) TestParseUserURLWithData() { url := "https://status.app/u/G10A4B0JdgwyRww90WXtnP1oNH1ZLQNM0yX0Ja9YyAMjrqSZIYINOHCbFhrnKRAcPGStPxCMJDSZlGCKzmZrJcimHY8BbcXlORrElv_BbQEegnMDPx1g9C5VVNl0fE4y#zQ3shwQPhRuDJSjVGVBnTjCdgXy5i9WQaeVPdGJD6yTarJQSj" urlData, err := s.m.ParseSharedURL(url) s.Require().NoError(err) s.Require().NotNil(urlData) s.Require().NotNil(urlData.Contact) s.Require().Equal("Mark Cole", urlData.Contact.DisplayName) s.Require().Equal("Visual designer @Status, cat lover, pizza enthusiast, yoga afficionada", urlData.Contact.Description) s.Require().Equal("zQ3shwQPhRuDJSjVGVBnTjCdgXy5i9WQaeVPdGJD6yTarJQSj", urlData.Contact.PublicKey) } func (s *MessengerShareUrlsSuite) TestShareUserURLWithData() { _, contact := s.createContact() url, err := s.m.ShareUserURLWithData(contact.ID) s.Require().NoError(err) userData, chatKey, err := s.m.prepareEncodedUserData(contact) s.Require().NoError(err) expectedURL := fmt.Sprintf("%s/u/%s#%s", baseShareURL, userData, chatKey) s.Require().Equal(expectedURL, url) } func (s *MessengerShareUrlsSuite) TestShareAndParseUserURLWithData() { _, contact := s.createContact() pk, err := contact.PublicKey() s.Require().NoError(err) shortKey, err := s.m.SerializePublicKey(crypto.CompressPubkey(pk)) s.Require().NoError(err) url, err := s.m.ShareUserURLWithData(contact.ID) s.Require().NoError(err) urlData, err := s.m.ParseSharedURL(url) s.Require().NoError(err) s.Require().NotNil(urlData.Contact) s.Require().Equal(contact.DisplayName, urlData.Contact.DisplayName) s.Require().Equal(shortKey, urlData.Contact.PublicKey) }