2020-01-02 09:21:11 +01:00
|
|
|
|
2020-01-11 12:47:11 +01:00
|
|
|
require "json"
|
|
|
|
|
2020-01-02 09:21:11 +01:00
|
|
|
# keep track of connected users and their requests
|
|
|
|
# TODO: requests should be handled concurrently
|
|
|
|
class User
|
|
|
|
property uid : Int32
|
|
|
|
property token : FileStorage::Token
|
2020-01-04 15:07:04 +01:00
|
|
|
property uploads : Array(FileStorage::UploadRequest)
|
|
|
|
property downloads : Array(FileStorage::DownloadRequest)
|
2020-01-02 09:21:11 +01:00
|
|
|
|
2020-01-04 15:07:04 +01:00
|
|
|
def initialize(@token,
|
|
|
|
@uploads = Array(FileStorage::UploadRequest).new,
|
|
|
|
@downloads = Array(FileStorage::DownloadRequest).new)
|
2020-01-02 09:21:11 +01:00
|
|
|
@uid = token.uid
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2020-01-11 12:47:11 +01:00
|
|
|
class TransferInfo
|
|
|
|
include JSON::Serializable
|
|
|
|
|
|
|
|
property owner : Int32
|
|
|
|
property file_info : FileInfo
|
|
|
|
property chunks : Hash(Int32, Bool)
|
|
|
|
|
|
|
|
def initialize(@owner, @file_info)
|
|
|
|
@chunks = Hash(Int32, Bool).new
|
|
|
|
@file_info.nb_chunks.times do |n|
|
|
|
|
@chunks[n] = false
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2020-01-02 09:21:11 +01:00
|
|
|
class Context
|
|
|
|
class_property service_name = "filestorage"
|
|
|
|
class_property storage_directory = "./storage"
|
2020-01-11 12:47:11 +01:00
|
|
|
class_property file_info_directory = "./file-infos"
|
|
|
|
|
|
|
|
class_property db : DODB::DataBase(TransferInfo) = self.init_db
|
|
|
|
|
|
|
|
def init_db
|
|
|
|
@@db = DODB::DataBase(TransferInfo).new @@file_info_directory
|
|
|
|
|
|
|
|
# init index, partitions and tags
|
|
|
|
Context.db.new_index "filedigest", &.file_info.digest
|
|
|
|
Context.db.new_partition "owner", &.owner
|
|
|
|
Context.db.new_tags "tags", &.tags
|
|
|
|
end
|
2020-01-02 09:21:11 +01:00
|
|
|
|
|
|
|
# list of connected users (fd => uid)
|
|
|
|
class_property connected_users = Hash(Int32, Int32).new
|
|
|
|
|
|
|
|
# users_status: keep track of the users' status even if they are
|
|
|
|
# disconnected, allowing the application to handle connection problems
|
|
|
|
class_property users_status = Hash(Int32, User).new
|
|
|
|
|
|
|
|
class_property service : IPC::Service? = nil
|
|
|
|
end
|