using System; using System.Collections.Generic; using System.Linq; using System.Threading.Tasks; using Microsoft.AspNetCore.Mvc; using Bit.Core.Repositories; using Microsoft.AspNetCore.Authorization; using Bit.Api.Models; using Bit.Core.Exceptions; using Bit.Core.Domains; using Microsoft.AspNetCore.Identity; using Bit.Core.Services; namespace Bit.Api.Controllers { [Route("folders")] [Authorize("Application")] public class FoldersController : Controller { private readonly ICipherRepository _cipherRepository; private readonly ICipherService _cipherService; private readonly UserManager _userManager; public FoldersController( ICipherRepository cipherRepository, ICipherService cipherService, UserManager userManager) { _cipherRepository = cipherRepository; _cipherService = cipherService; _userManager = userManager; } [HttpGet("{id}")] public async Task Get(string id) { var folder = await _cipherRepository.GetByIdAsync(new Guid(id), new Guid(_userManager.GetUserId(User))); if(folder == null || folder.Type != Core.Enums.CipherType.Folder) { throw new NotFoundException(); } return new FolderResponseModel(folder); } [HttpGet("")] public async Task> Get() { ICollection folders = await _cipherRepository.GetManyByTypeAndUserIdAsync(Core.Enums.CipherType.Folder, new Guid(_userManager.GetUserId(User))); var responses = folders.Select(f => new FolderResponseModel(f)); return new ListResponseModel(responses); } [HttpPost("")] public async Task Post([FromBody]FolderRequestModel model) { var folder = model.ToCipher(_userManager.GetUserId(User)); await _cipherService.SaveAsync(folder); return new FolderResponseModel(folder); } [HttpPut("{id}")] [HttpPost("{id}")] public async Task Put(string id, [FromBody]FolderRequestModel model) { var folder = await _cipherRepository.GetByIdAsync(new Guid(id), new Guid(_userManager.GetUserId(User))); if(folder == null || folder.Type != Core.Enums.CipherType.Folder) { throw new NotFoundException(); } await _cipherService.SaveAsync(model.ToCipher(folder)); return new FolderResponseModel(folder); } [HttpDelete("{id}")] [HttpPost("{id}/delete")] public async Task Delete(string id) { var folder = await _cipherRepository.GetByIdAsync(new Guid(id), new Guid(_userManager.GetUserId(User))); if(folder == null || folder.Type != Core.Enums.CipherType.Folder) { throw new NotFoundException(); } await _cipherService.DeleteAsync(folder); } } }