-
-
Notifications
You must be signed in to change notification settings - Fork 110
/
resolvers.mjs
68 lines (59 loc) · 1.67 KB
/
resolvers.mjs
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
import fs from 'fs'
import promisesAll from 'promises-all'
import mkdirp from 'mkdirp'
import shortid from 'shortid'
import lowdb from 'lowdb'
import FileSync from 'lowdb/adapters/FileSync'
import { GraphQLUpload } from 'apollo-upload-server'
const uploadDir = './uploads'
const db = lowdb(new FileSync('db.json'))
// Seed an empty DB
db.defaults({ uploads: [] }).write()
// Ensure upload directory exists
mkdirp.sync(uploadDir)
const storeFS = ({ stream, filename }) => {
const id = shortid.generate()
const path = `${uploadDir}/${id}-${filename}`
return new Promise((resolve, reject) =>
stream
.on('error', error => {
if (stream.truncated)
// Delete the truncated file
fs.unlinkSync(path)
reject(error)
})
.on('end', () => resolve({ id, path }))
.pipe(fs.createWriteStream(path))
)
}
const storeDB = file =>
db
.get('uploads')
.push(file)
.last()
.write()
const processUpload = async upload => {
const { stream, filename, mimetype, encoding } = await upload
const { id, path } = await storeFS({ stream, filename })
return storeDB({ id, filename, mimetype, encoding, path })
}
export default {
Upload: GraphQLUpload,
Query: {
uploads: () => db.get('uploads').value()
},
Mutation: {
singleUpload: (obj, { file }) => processUpload(file),
multipleUpload: async (obj, { files }) => {
const { resolve, reject } = await promisesAll.all(
files.map(processUpload)
)
if (reject.length)
reject.forEach(({ name, message }) =>
// eslint-disable-next-line no-console
console.error(`${name}: ${message}`)
)
return resolve
}
}
}