/
index.js
151 lines (132 loc) · 4.52 KB
/
index.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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
const request = require('request')
const purest = require('purest')({ request })
const { promisify } = require('node:util')
const Provider = require('../Provider')
const logger = require('../../logger')
const adapter = require('./adapter')
const { ProviderApiError, ProviderAuthError } = require('../error')
const { requestStream } = require('../../helpers/utils')
/**
* Adapter for API https://docs.microsoft.com/en-us/onedrive/developer/rest-api/
*/
class OneDrive extends Provider {
constructor (options) {
super(options)
this.authProvider = OneDrive.authProvider
this.client = purest({
...options,
provider: OneDrive.authProvider,
})
}
static get authProvider () {
return 'microsoft'
}
_userInfo ({ token }, done) {
this.client
.get('me')
.auth(token)
.request(done)
}
/**
* Makes 2 requests in parallel - 1. to get files, 2. to get user email
* it then waits till both requests are done before proceeding with the callback
*
* @param {object} options
* @param {string} options.directory
* @param {any} options.query
* @param {string} options.token
* @param {Function} done
*/
_list ({ directory, query, token }, done) {
const path = directory ? `items/${directory}` : 'root'
const rootPath = query.driveId ? `/drives/${query.driveId}` : '/me/drive'
const qs = { $expand: 'thumbnails' }
if (query.cursor) {
qs.$skiptoken = query.cursor
}
this.client
.get(`${rootPath}/${path}/children`)
.qs(qs)
.auth(token)
.request((err, resp, body) => {
if (err || resp.statusCode !== 200) {
err = this._error(err, resp)
logger.error(err, 'provider.onedrive.list.error')
return done(err)
}
this._userInfo({ token }, (err, infoResp) => {
if (err || infoResp.statusCode !== 200) {
err = this._error(err, infoResp)
logger.error(err, 'provider.onedrive.user.error')
return done(err)
}
done(null, this.adaptData(body, infoResp.body.mail || infoResp.body.userPrincipalName))
})
})
}
async download ({ id, token, query }) {
try {
const rootPath = query.driveId ? `/drives/${query.driveId}` : '/me/drive'
const req = this.client
.get(`${rootPath}/items/${id}/content`)
.auth(token)
.request()
return await requestStream(req, async (res) => this._error(null, res))
} catch (err) {
logger.error(err, 'provider.onedrive.download.error')
throw err
}
}
async thumbnail () {
// not implementing this because a public thumbnail from onedrive will be used instead
logger.error('call to thumbnail is not implemented', 'provider.onedrive.thumbnail.error')
throw new Error('call to thumbnail is not implemented')
}
_size ({ id, query, token }, done) {
const rootPath = query.driveId ? `/drives/${query.driveId}` : '/me/drive'
return this.client
.get(`${rootPath}/items/${id}`)
.auth(token)
.request((err, resp, body) => {
if (err || resp.statusCode !== 200) {
err = this._error(err, resp)
logger.error(err, 'provider.onedrive.size.error')
return done(err)
}
done(null, body.size)
})
}
async logout () {
return { revoked: false, manual_revoke_url: 'https://account.live.com/consent/Manage' }
}
adaptData (res, username) {
const data = { username, items: [] }
const items = adapter.getItemSubList(res)
items.forEach((item) => {
data.items.push({
isFolder: adapter.isFolder(item),
icon: adapter.getItemIcon(item),
name: adapter.getItemName(item),
mimeType: adapter.getMimeType(item),
id: adapter.getItemId(item),
thumbnail: adapter.getItemThumbnailUrl(item),
requestPath: adapter.getItemRequestPath(item),
modifiedDate: adapter.getItemModifiedDate(item),
size: adapter.getItemSize(item),
})
})
data.nextPagePath = adapter.getNextPagePath(res)
return data
}
_error (err, resp) {
if (resp) {
const fallbackMsg = `request to ${this.authProvider} returned ${resp.statusCode}`
const errMsg = (resp.body || {}).error ? resp.body.error.message : fallbackMsg
return resp.statusCode === 401 ? new ProviderAuthError() : new ProviderApiError(errMsg, resp.statusCode)
}
return err
}
}
OneDrive.prototype.list = promisify(OneDrive.prototype._list)
OneDrive.prototype.size = promisify(OneDrive.prototype._size)
module.exports = OneDrive