SemianiakaVY/BotSharp
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Import agent from zip file. Add swagger file mock FileUploadOperation.

Browse source
This commit is contained in:
botsharp2018 2018-08-27 07:59:47 -05:00
parent 39f508ae08
commit 94592bb911
3 changed files with 52 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
BotSharp.RestApi/AgentController.cs
View file

@ -4,6 +4,7 @@ using BotSharp.Core.Engines.BotSharp;
using BotSharp.Core.Models;
using DotNetToolkit;
using EntityFrameworkCore.BootKit;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Configuration;
using Newtonsoft.Json;
@ -12,6 +13,7 @@ using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace BotSharp.RestApi
{
@ -43,23 +45,38 @@ namespace BotSharp.RestApi
/// <summary>
/// Restore a agent from a uploaded zip file
/// </summary>
/// <param name="agentId"></param>
/// <param name="file"></param>
/// <returns></returns>
[HttpGet("{agentId}")]
public ActionResult Restore([FromRoute] String agentId)
[HttpPost]
public async Task<ActionResult> Restore(IFormFile file)
{
var botsHeaderFilePath = Path.Combine(AppDomain.CurrentDomain.GetData("DataPath").ToString(), $"DbInitializer{Path.DirectorySeparatorChar}Agents{Path.DirectorySeparatorChar}agents.json");
System.IO.Compression.ZipFile.ExtractToDirectory("", "");
if (file == null || file.Length == 0)
{
return BadRequest();
}
var filePath = Path.GetTempFileName();
using (var stream = new FileStream(filePath, FileMode.Create))
{
await file.CopyToAsync(stream);
}
_platform.LoadAgentFromFile<AgentImporterInDialogflow>("", new AgentImportHeader { });
/*var botsHeaderFilePath = Path.Combine(AppDomain.CurrentDomain.GetData("DataPath").ToString(), $"DbInitializer{Path.DirectorySeparatorChar}Agents{Path.DirectorySeparatorChar}agents.json");
var agents = JsonConvert.DeserializeObject<List<AgentImportHeader>>(System.IO.File.ReadAllText(botsHeaderFilePath));
var rasa = new BotSharpAi();
var agentHeader = agents.First(x => x.Id == agentId);
rasa.RestoreAgent<AgentImporterInDialogflow>(agentHeader);
rasa.RestoreAgent<AgentImporterInDialogflow>(agentHeader);*/
return Ok();
}
/// <summary>
/// Restore a agent from a uploaded zip file
/// Train agent
/// </summary>
/// <param name="agentId"></param>
/// <returns></returns>

1
BotSharp.WebHost/Startup.cs
View file

@ -59,6 +59,7 @@ namespace BotSharp.WebHost
var filePath = Path.Combine(PlatformServices.Default.Application.ApplicationBasePath, "BotSharp.RestApi.xml");
c.IncludeXmlComments(filePath);
c.OperationFilter<SwaggerFileUploadOperation>();
});
// register platform dependency

28
BotSharp.WebHost/SwaggerFileUploadOperation.cs Normal file
View file

@ -0,0 +1,28 @@
using Swashbuckle.AspNetCore.Swagger;
using Swashbuckle.AspNetCore.SwaggerGen;
using System;
using System.Collections.Generic;
using System.Text;
namespace BotSharp.WebHost
{
public class SwaggerFileUploadOperation : IOperationFilter
{
public void Apply(Operation operation, OperationFilterContext context)
{
if (operation.OperationId == "V1AgentRestorePost")
{
operation.Parameters.Clear();
operation.Parameters.Add(new NonBodyParameter
{
Name = "uploadedFile",
In = "formData",
Description = "Upload File",
Required = true,
Type = "file"
});
operation.Consumes.Add("multipart/form-data");
}
}
}
}