nhyatt/bitwarden
1
0
Fork 0
mirror of https://github.com/bitwarden/server.git synced 2025-07-01 08:02:49 -05:00
Code Activity

database adjustments and storage for attachments

Browse Source
This commit is contained in:
Kyle Spearrin
2017-06-30 14:41:57 -04:00
parent 6cea556ae1
commit 284078e946
17 changed files with 178 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/Core/Models/Data/Attachment.cs
View File

@ -1,4 +1,5 @@
using System;
using Newtonsoft.Json;
using System;
namespace Bit.Core.Models.Data
{
@ -12,7 +13,23 @@ namespace Bit.Core.Models.Data
public class MetaData
{
public long Size { get; set; }
private long _size;
[JsonIgnore]
public long Size
{
get { return _size; }
set { _size = value; }
}
// We serialize Size as a string since JSON (or Javascript) doesn't support full precision for long numbers
[JsonProperty("Size")]
public string SizeString
{
get { return _size.ToString(); }
set { _size = Convert.ToInt64(value); }
}
public string FileName { get; set; }
}
}

18
src/Core/Models/Table/Organization.cs
View File

@ -16,6 +16,8 @@ namespace Bit.Core.Models.Table
public short? MaxCollections { get; set; }
public bool UseGroups { get; set; }
public bool UseDirectory { get; set; }
public long? Storage { get; set; }
public short? MaxStorageGb { get; set; }
public string StripeCustomerId { get; set; }
public string StripeSubscriptionId { get; set; }
public bool Enabled { get; set; } = true;
@ -29,5 +31,21 @@ namespace Bit.Core.Models.Table
Id = CoreHelpers.GenerateComb();
}
}
public long StorageBytesRemaining()
{
if(!MaxStorageGb.HasValue)
{
return 0;
}
var maxStorageBytes = MaxStorageGb.Value * 1073741824L;
if(!Storage.HasValue)
{
return maxStorageBytes;
}
return maxStorageBytes - Storage.Value;
}
}
}

19
src/Core/Models/Table/User.cs
View File

@ -27,6 +27,9 @@ namespace Bit.Core.Models.Table
public string Key { get; set; }
public string PublicKey { get; set; }
public string PrivateKey { get; set; }
public bool Premium { get; set; }
public long? Storage { get; set; }
public short? MaxStorageGb { get; set; }
public DateTime CreationDate { get; internal set; } = DateTime.UtcNow;
public DateTime RevisionDate { get; internal set; } = DateTime.UtcNow;
@ -99,5 +102,21 @@ namespace Bit.Core.Models.Table
return providers[provider];
}
public long StorageBytesRemaining()
{
if(!MaxStorageGb.HasValue)
{
return 0;
}
var maxStorageBytes = MaxStorageGb.Value * 1073741824L;
if(!Storage.HasValue)
{
return maxStorageBytes;
}
return maxStorageBytes - Storage.Value;
}
}
}