forked from finos/spring-bot
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Pull request finos#47: Feature/SYMPHONYP-1106 download feature in spr…
…ing bot team framework Merge in SYMPHONYP/symphony-java-toolkit from feature/SYMPHONYP-1106-download-feature-in-spring-bot-team-framework to spring-bot-master-db * commit '15451c97348710774d7276e65acacbba9271d0f1': SYMPHONYP-1106 upload attachment feature in famework Fix for finos#342 update pom Update CHANGES.md Update CHANGES.md
- Loading branch information
Showing
11 changed files
with
248 additions
and
19 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
44 changes: 44 additions & 0 deletions
44
...oot-starter/src/main/java/org/finos/springbot/teams/handlers/SimpleAttachmentHandler.java
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,44 @@ | ||
| package org.finos.springbot.teams.handlers; | ||
|
|
||
| import java.io.File; | ||
| import java.io.IOException; | ||
| import java.nio.file.Files; | ||
| import java.util.HashMap; | ||
| import java.util.Map; | ||
|
|
||
| import org.finos.springbot.workflow.response.AttachmentResponse; | ||
|
|
||
| import com.microsoft.bot.schema.Attachment; | ||
| import com.microsoft.bot.schema.teams.FileConsentCard; | ||
|
|
||
| public class SimpleAttachmentHandler implements AttachmentHandler { | ||
|
|
||
| public Attachment formatAttachment(AttachmentResponse ar) { | ||
|
|
||
| try { | ||
| Attachment a = new Attachment(); | ||
| String fileName = ar.getName().replaceAll("\\s+", "_") + "." + ar.getExtension(); | ||
| File f = null; | ||
| f = File.createTempFile("temp", fileName); | ||
| Files.write(f.toPath(), ar.getAttachment()); | ||
|
|
||
| Map<String, String> consentContext = new HashMap<>(); | ||
| consentContext.put("filename", fileName); | ||
| consentContext.put("filepath", f.getAbsolutePath()); | ||
|
|
||
| FileConsentCard fileCard = new FileConsentCard(); | ||
| fileCard.setDescription("This is the file I want to send you"); | ||
| fileCard.setSizeInBytes(ar.getAttachment().length); | ||
| fileCard.setAcceptContext(consentContext); | ||
| fileCard.setDeclineContext(consentContext); | ||
|
|
||
| a.setContent(fileCard); | ||
| a.setName(fileName); | ||
| a.setContentType(FileConsentCard.CONTENT_TYPE); | ||
| return a; | ||
| } catch (IOException e) { | ||
| throw new RuntimeException(e); | ||
| } | ||
|
|
||
| } | ||
| } |
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
148 changes: 148 additions & 0 deletions
148
...ng-boot-starter/src/main/java/org/finos/springbot/teams/messages/FileActivityHandler.java
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,148 @@ | ||
| package org.finos.springbot.teams.messages; | ||
|
|
||
| import java.io.BufferedReader; | ||
| import java.io.File; | ||
| import java.io.FileInputStream; | ||
| import java.io.InputStreamReader; | ||
| import java.io.OutputStream; | ||
| import java.net.HttpURLConnection; | ||
| import java.net.URL; | ||
| import java.util.List; | ||
| import java.util.Map; | ||
| import java.util.concurrent.CompletableFuture; | ||
| import java.util.concurrent.atomic.AtomicReference; | ||
|
|
||
| import org.finos.springbot.teams.content.serialization.TeamsHTMLParser; | ||
| import org.finos.springbot.teams.conversations.TeamsConversations; | ||
| import org.finos.springbot.teams.state.TeamsStateStorage; | ||
| import org.finos.springbot.workflow.actions.consumers.ActionConsumer; | ||
| import org.finos.springbot.workflow.form.FormConverter; | ||
| import org.finos.springbot.workflow.form.FormValidationProcessor; | ||
| import org.slf4j.Logger; | ||
| import org.slf4j.LoggerFactory; | ||
|
|
||
| import com.microsoft.bot.builder.MessageFactory; | ||
| import com.microsoft.bot.builder.TurnContext; | ||
| import com.microsoft.bot.schema.Activity; | ||
| import com.microsoft.bot.schema.Attachment; | ||
| import com.microsoft.bot.schema.ResultPair; | ||
| import com.microsoft.bot.schema.TextFormatTypes; | ||
| import com.microsoft.bot.schema.teams.FileConsentCardResponse; | ||
| import com.microsoft.bot.schema.teams.FileInfoCard; | ||
|
|
||
| public class FileActivityHandler extends MessageActivityHandler { | ||
|
|
||
| private static final Logger LOG = LoggerFactory.getLogger(FileActivityHandler.class); | ||
|
|
||
|
|
||
| public FileActivityHandler(List<ActionConsumer> messageConsumers, TeamsConversations teamsConversations, | ||
| TeamsStateStorage teamsStateStorage, TeamsHTMLParser parser, FormConverter formConverter, | ||
| FormValidationProcessor validationProcessor) { | ||
| super(messageConsumers, teamsConversations, teamsStateStorage, parser, formConverter, validationProcessor); | ||
| } | ||
|
|
||
| @Override | ||
| protected CompletableFuture<Void> onTeamsFileConsentAccept(TurnContext turnContext, | ||
| FileConsentCardResponse fileConsentCardResponse) { | ||
|
|
||
| return upload(fileConsentCardResponse) | ||
| .thenCompose(result -> !result.result() ? fileUploadFailed(turnContext, result.value()) | ||
| : fileUploadCompleted(turnContext, fileConsentCardResponse)); | ||
| } | ||
|
|
||
| @Override | ||
| protected CompletableFuture<Void> onTeamsFileConsentDecline(TurnContext turnContext, | ||
| FileConsentCardResponse fileConsentCardResponse) { | ||
| Map<String, String> context = (Map<String, String>) fileConsentCardResponse.getContext(); | ||
|
|
||
| Activity reply = MessageFactory | ||
| .text(String.format("Declined. We won't upload file <b>%s</b>.", context.get("filename"))); | ||
| reply.setTextFormat(TextFormatTypes.XML); | ||
|
|
||
| return turnContext.sendActivityBlind(reply); | ||
| } | ||
|
|
||
| private CompletableFuture<ResultPair<String>> upload(FileConsentCardResponse fileConsentCardResponse) { | ||
| AtomicReference<ResultPair<String>> result = new AtomicReference<>(); | ||
|
|
||
| return CompletableFuture.runAsync(() -> { | ||
| Map<String, String> context = (Map<String, String>) fileConsentCardResponse.getContext(); | ||
| LOG.info("filepath - {}", context.get("filepath")); | ||
| LOG.info("File upload endpoint : {}", fileConsentCardResponse.getUploadInfo().getUploadUrl()); | ||
| File filePath = new File(context.get("filepath")); | ||
|
|
||
| HttpURLConnection connection = null; | ||
| try { | ||
| URL url = new URL(fileConsentCardResponse.getUploadInfo().getUploadUrl()); | ||
| connection = (HttpURLConnection) url.openConnection(); | ||
| connection.setRequestMethod("PUT"); | ||
| connection.setDoOutput(true); | ||
| connection.setRequestProperty("Content-Length", Long.toString(filePath.length())); | ||
| connection.setRequestProperty("Content-Range", | ||
| String.format("bytes 0-%d/%d", filePath.length() - 1, filePath.length())); | ||
|
|
||
| try (FileInputStream fileStream = new FileInputStream(filePath); | ||
| OutputStream uploadStream = connection.getOutputStream()) { | ||
| byte[] buffer = new byte[4096]; | ||
| int bytes_read; | ||
| while ((bytes_read = fileStream.read(buffer)) != -1) { | ||
| uploadStream.write(buffer, 0, bytes_read); | ||
| } | ||
|
|
||
| uploadStream.flush(); | ||
| } | ||
|
|
||
| try { | ||
| BufferedReader in = new BufferedReader(new InputStreamReader(connection.getInputStream())); | ||
| String inputLine; | ||
| while ((inputLine = in.readLine()) != null) | ||
| LOG.info(inputLine); | ||
| in.close(); | ||
| } catch (Exception e) { | ||
| LOG.error("Exception occured while reading steam.. ignore this error " + e); | ||
| } | ||
| result.set(new ResultPair<String>(true, null)); | ||
| } catch (Throwable t) { | ||
| result.set(new ResultPair<String>(false, t.getLocalizedMessage())); | ||
| } finally { | ||
| if (connection != null) { | ||
| connection.disconnect(); | ||
| } | ||
| } | ||
| }).thenApply(aVoid -> result.get()); | ||
| } | ||
|
|
||
| private CompletableFuture<Void> fileUploadFailed(TurnContext turnContext, String error) { | ||
| Activity reply = MessageFactory.text("<b>File upload failed.</b> Error: <pre>" + error + "</pre>"); | ||
| reply.setTextFormat(TextFormatTypes.XML); | ||
| return turnContext.sendActivityBlind(reply); | ||
| } | ||
|
|
||
| private CompletableFuture<Void> fileDownloadCompleted(TurnContext turnContext, Attachment attachment) { | ||
| Activity reply = MessageFactory.text(String.format("<b>%s</b> received and saved.", attachment.getName())); | ||
| reply.setTextFormat(TextFormatTypes.XML); | ||
|
|
||
| return turnContext.sendActivityBlind(reply); | ||
| } | ||
|
|
||
| private CompletableFuture<Void> fileUploadCompleted(TurnContext turnContext, | ||
| FileConsentCardResponse fileConsentCardResponse) { | ||
| FileInfoCard downloadCard = new FileInfoCard(); | ||
| downloadCard.setUniqueId(fileConsentCardResponse.getUploadInfo().getUniqueId()); | ||
| downloadCard.setFileType(fileConsentCardResponse.getUploadInfo().getFileType()); | ||
|
|
||
| Attachment asAttachment = new Attachment(); | ||
| asAttachment.setContent(downloadCard); | ||
| asAttachment.setContentType(FileInfoCard.CONTENT_TYPE); | ||
| asAttachment.setName(fileConsentCardResponse.getUploadInfo().getName()); | ||
| asAttachment.setContentUrl(fileConsentCardResponse.getUploadInfo().getContentUrl()); | ||
|
|
||
| Activity reply = MessageFactory | ||
| .text(String.format("<b>File uploaded.</b> Your file <b>%s</b> is ready to download", | ||
| fileConsentCardResponse.getUploadInfo().getName())); | ||
| reply.setTextFormat(TextFormatTypes.XML); | ||
| reply.setAttachment(asAttachment); | ||
|
|
||
| return turnContext.sendActivityBlind(reply); | ||
| } | ||
| } |
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
Oops, something went wrong.