|
|
|
@ -1,42 +1,24 @@
|
|
|
|
|
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))
|
|
|
|
|
})
|
|
|
|
|
}
|
|
|
|
|
const tmp = require('tmp-promise')
|
|
|
|
|
const CombinedStream = require('combined-stream')
|
|
|
|
|
|
|
|
|
|
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) {
|
|
|
|
|
// Check if we have a temporary files which have been written to.
|
|
|
|
|
// If so, we should "flush" the temporary files to the proper storage.
|
|
|
|
|
const placeholders = message.socket.session.temp_write_files?.[descriptor]
|
|
|
|
|
if ( !placeholders ) {
|
|
|
|
|
return message.send_response(
|
|
|
|
|
message.fresh()
|
|
|
|
|
)
|
|
|
|
@ -48,7 +30,6 @@ console.log({placeholder})
|
|
|
|
|
descriptor_type: NodeDescriptorType.File,
|
|
|
|
|
})
|
|
|
|
|
|
|
|
|
|
console.log({node})
|
|
|
|
|
if ( !node ) {
|
|
|
|
|
return message.send_response(
|
|
|
|
|
message.fresh().error(Errors.NodeDoesNotExist)
|
|
|
|
@ -56,30 +37,45 @@ console.log({node})
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
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
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Sort the placeholders by the byte starting position
|
|
|
|
|
placeholders.sort((a, b) => {
|
|
|
|
|
if ( a.position < b.position ) return -1
|
|
|
|
|
if ( a.position > b.position ) return 1
|
|
|
|
|
return 0
|
|
|
|
|
})
|
|
|
|
|
|
|
|
|
|
// Combine them into a single stream
|
|
|
|
|
const stream = CombinedStream.create()
|
|
|
|
|
const combined_file = await tmp.file()
|
|
|
|
|
|
|
|
|
|
for ( const placeholder of placeholders ) {
|
|
|
|
|
stream.append(fs.createReadStream(placeholder.path))
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
const pipe = stream.pipe(fs.createWriteStream(combined_file.path))
|
|
|
|
|
|
|
|
|
|
await new Promise(res => pipe.on('finish', res))
|
|
|
|
|
|
|
|
|
|
// Store the temporary file
|
|
|
|
|
const new_file = await di.upload.provider().store({
|
|
|
|
|
temp_path: placeholder.path,
|
|
|
|
|
temp_path: combined_file.path,
|
|
|
|
|
original_name: node.pied_name,
|
|
|
|
|
mime_type: 'application/octet-stream', // TODO determine from file extension?
|
|
|
|
|
})
|
|
|
|
|
|
|
|
|
|
console.log({new_file})
|
|
|
|
|
const stat = await fs.promises.stat(combined_file.path)
|
|
|
|
|
|
|
|
|
|
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]')
|
|
|
|
|
}
|
|
|
|
|