-
Notifications
You must be signed in to change notification settings - Fork 7
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #35 from wireapp/staging
Release
- Loading branch information
Showing
9 changed files
with
215 additions
and
183 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,127 @@ | ||
| package com.wire.bots.roman; | ||
|
|
||
|
|
||
| import com.wire.bots.cryptobox.CryptoException; | ||
| import com.wire.bots.roman.model.Attachment; | ||
| import com.wire.bots.roman.model.IncomingMessage; | ||
| import com.wire.bots.roman.model.Mention; | ||
| import com.wire.bots.sdk.ClientRepo; | ||
| import com.wire.bots.sdk.WireClient; | ||
| import com.wire.bots.sdk.assets.*; | ||
| import com.wire.bots.sdk.models.AssetKey; | ||
| import com.wire.bots.sdk.server.model.Conversation; | ||
| import com.wire.bots.sdk.tools.Logger; | ||
|
|
||
| import javax.annotation.Nullable; | ||
| import java.io.IOException; | ||
| import java.util.Base64; | ||
| import java.util.UUID; | ||
|
|
||
| public class Sender { | ||
|
|
||
| private final ClientRepo repo; | ||
|
|
||
| public Sender(ClientRepo repo) { | ||
| this.repo = repo; | ||
| } | ||
|
|
||
| public UUID send(IncomingMessage message, UUID botId) throws Exception { | ||
| try (WireClient client = repo.getClient(botId)) { | ||
| return send(message, client); | ||
| } | ||
| } | ||
|
|
||
| @Nullable | ||
| private UUID send(IncomingMessage message, @Nullable WireClient client) throws Exception { | ||
| if (client == null) | ||
| return null; | ||
|
|
||
| switch (message.type) { | ||
| case "text": { | ||
| MessageText text = new MessageText(message.text.data); | ||
| if (message.text.mentions != null) { | ||
| for (Mention mention : message.text.mentions) | ||
| text.addMention(mention.userId, mention.offset, mention.length); | ||
| } | ||
| client.send(text); | ||
| return text.getMessageId(); | ||
| } | ||
| case "attachment": { | ||
| if (message.attachment.mimeType.startsWith("image")) { | ||
| final byte[] decode = Base64.getDecoder().decode(message.attachment.data); | ||
| return client.sendPicture(decode, message.attachment.mimeType); | ||
| } | ||
|
|
||
| return sendAttachment(message, client); | ||
| } | ||
| case "poll": { | ||
| if (message.poll.type.equals("create")) { | ||
| return sendNewPoll(message, client); | ||
| } | ||
| if (message.poll.type.equals("confirmation")) { | ||
| return sendPollConfirmation(message, client); | ||
| } | ||
| } | ||
| break; | ||
| } | ||
|
|
||
| return null; | ||
| } | ||
|
|
||
| private UUID sendNewPoll(IncomingMessage message, WireClient client) throws Exception { | ||
| MessageText messageText = new MessageText(message.text.data); | ||
| if (message.text.mentions != null) { | ||
| for (Mention mention : message.text.mentions) | ||
| messageText.addMention(mention.userId, mention.offset, mention.length); | ||
| } | ||
|
|
||
| Poll poll = new Poll(message.poll.id); | ||
| poll.addText(messageText); | ||
|
|
||
| int i = 0; | ||
| for (String caption : message.poll.buttons) { | ||
| poll.addButton("" + i++, caption); | ||
| } | ||
|
|
||
| Logger.info("poll.create: id: %s", message.poll.id); | ||
|
|
||
| client.send(poll); | ||
|
|
||
| return poll.getMessageId(); | ||
| } | ||
|
|
||
| private UUID sendAttachment(IncomingMessage message, WireClient client) throws Exception { | ||
| UUID messageId = UUID.randomUUID(); | ||
|
|
||
| final Attachment attachment = message.attachment; | ||
| final byte[] decode = Base64.getDecoder().decode(attachment.data); | ||
|
|
||
| FileAssetPreview preview = new FileAssetPreview(attachment.filename, attachment.mimeType, decode.length, messageId); | ||
| FileAsset asset = new FileAsset(decode, attachment.mimeType, messageId); | ||
|
|
||
| client.send(preview); | ||
| final AssetKey assetKey = client.uploadAsset(asset); | ||
| asset.setAssetKey(assetKey.key); | ||
| asset.setAssetToken(assetKey.token); | ||
| client.send(asset); | ||
| return messageId; | ||
| } | ||
|
|
||
| private UUID sendPollConfirmation(IncomingMessage message, WireClient client) throws Exception { | ||
| ButtonActionConfirmation confirmation = new ButtonActionConfirmation( | ||
| message.poll.id, | ||
| message.poll.offset.toString()); | ||
|
|
||
| Logger.info("poll.confirmation: pollId: %s, offset: %s", message.poll.id, message.poll.offset); | ||
|
|
||
| client.send(confirmation, message.poll.userId); | ||
|
|
||
| return confirmation.getMessageId(); | ||
| } | ||
|
|
||
| public Conversation getConversation(UUID botId) throws IOException, CryptoException { | ||
| try (WireClient client = repo.getClient(botId)) { | ||
| return client.getConversation(); | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.