-
Notifications
You must be signed in to change notification settings - Fork 38
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
SYMPHONYP-1106 upload attachment feature in famework
- Loading branch information
1 parent
9571941
commit 15451c9
Showing
9 changed files
with
239 additions
and
18 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
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