Create FLUSH handler and update RELEASE to write temp file to backend

master
Garrett Mills 4 years ago
parent 8ea69dd006
commit ba8e0396b8
Signed by: garrettmills
GPG Key ID: D2BF5FBA8298F246

@ -88,7 +88,7 @@ class ServerUnit extends Unit {
}
const encoded_buffer = await this._bufferStream(stream)
const decoded_buffer = new Buffer(encoded_buffer.toString(), 'base64')
const decoded_buffer = Buffer.from(encoded_buffer.toString(), 'base64')
const combined_stream = CombinedStream.create()
combined_stream.append(decoded_buffer)

@ -0,0 +1,85 @@
const { Errors } = require('../../shared')
const { NodeDescriptorType } = require('../../enum')
const fs = require('fs')
const exists = async file => {
return new Promise(res => {
fs.promises.stat(file).then(() => res(true)).catch(() => res(false))
})
}
module.exports = exports = async (message, di) => {
console.log('[FLUSH]')
const Node = di.models.get('fs:Node')
const { descriptor } = message.data()
const node_uuid = message.socket.session.file_descriptors?.[descriptor]
console.log({node_uuid})
if ( !node_uuid ) {
return message.send_response(
message.fresh().error(Errors.NodeDoesNotExist)
)
}
// Check if we have a temporary file which has been written to.
// If so, we should "flush" the temporary file to the proper storage.
const placeholder = message.socket.session.temp_write_files?.[node_uuid]
console.log({placeholder})
if ( !placeholder ) {
return message.send_response(
message.fresh()
)
}
let stat;
try {
stat = await fs.promises.stat(placeholder.path)
} catch (e) {
return message.send_response(
message.fresh()
)
}
const node = await Node.findOne({
uuid: node_uuid,
deleted: false,
descriptor_type: NodeDescriptorType.File,
})
console.log({node})
if ( !node ) {
return message.send_response(
message.fresh().error(Errors.NodeDoesNotExist)
)
}
const existing_file = await node.uploaded_file()
console.log({existing_file})
if ( existing_file ) {
// This is a new write, so delete the old file
await existing_file.delete()
delete node.uploaded_file_id
}
// Store the temporary file
const new_file = await di.upload.provider().store({
temp_path: placeholder.path,
original_name: node.pied_name,
mime_type: 'application/octet-stream', // TODO determine from file extension?
})
console.log({new_file})
node.uploaded_file_id = new_file.id
node.size = stat.size
await node.save()
console.log({saved_node: node})
message.send_response(
message.fresh()
)
console.log('[/FLUSH]')
}

@ -1,11 +1,28 @@
const fs = require('fs')
const exists = async file => {
return new Promise(res => {
fs.promises.stat(file).then(() => res(true)).catch(() => res(false))
})
}
module.exports = exports = async (message, di) => {
console.log('[RELEASE]')
const { descriptor } = message.data()
if ( message.socket.session.file_descriptors ) {
const node_uuid = message.socket.session.file_descriptors?.[descriptor]
if ( node_uuid ) {
delete message.socket.session.file_descriptors[descriptor]
}
const placeholder = message.socket.session.temp_write_files?.[node_uuid]
if ( placeholder && (await exists(placeholder.path)) ) {
await fs.promises.unlink(placeholder.path)
}
message.send_response(
message.fresh()
)
console.log('[/RELEASE]')
}

Loading…
Cancel
Save