-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtransfer_file.js
96 lines (63 loc) · 2.56 KB
/
transfer_file.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
require('dotenv').config();
var crypto = require("crypto");
const mongoose = require("mongoose");
// const { path } = require('./schema/user_detail');
const path = require('path');
let link = process.env.DB_LINK;
// var json_data;
var user_detail_schema = require("./schema/user_detail");
function connect_to_db() {
mongoose.connect(link, { useNewUrlParser: true, useUnifiedTopology: true, useCreateIndex: true }).catch(error => { });
}
async function transfer_file(json_data) {
let result;
let model = mongoose.models["user_detail"] === undefined ? mongoose.model("user_detail", user_detail_schema) : mongoose.model("user_detail");
result = await model.findOne({ email: json_data.email, token: json_data.token, u_id: json_data.u_id });
if (result == null || result.account_status != "active") {
return { status: "error", message: "Not a valid user" }
}
let result2;
let folder_name;
let file_ext = path.extname(json_data.file_name);
if (!result.folder_name) {
let new_folder_name ;
while (true) {
new_folder_name = "fi" + crypto.randomBytes(10).toString('hex');
//
result2 = await model.findOne({ folder_name: new_folder_name });
if (result2 == null) {
break;
}
}
result2 = await model.updateOne({ u_id: json_data.u_id }, { $set: { folder_name: new_folder_name } });
folder_name = new_folder_name;
}else{
folder_name = result.folder_name;
}
//create new file name
let new_file_name;
// let total_file_count =
let table_file = {};
for (let i = 0; i < result.files.length; i++) {
table_file[result.files[i].file_name] = true;
}
while (true) {
new_file_name = "fi" + crypto.randomBytes(10).toString('hex') + file_ext;
if (table_file[new_file_name] == null) {
break;
}
}
result2 = await model.updateOne({ u_id: json_data.u_id }, { $push: { files: { file_name: json_data.file_name,new_file_name: new_file_name,sender_u_id: json_data.u_id,rec_u_id : json_data.curr_f_id,mime_type: json_data.mime_type } } });
if (result2.nModified == 1) {
return { status: "ok", file_name: json_data.file_name, curr_file_name: new_file_name,folder_name:folder_name }
} else {
return { status: "error", message: "not able to save file_name" };
}
}
async function main(data) {
connect_to_db();
let result;
result = await transfer_file(data);
return result;
}
module.exports = main;