71 lines
1.8 KiB
TypeScript
71 lines
1.8 KiB
TypeScript
import { dangerousGetDB, closeDB } from "."
|
|
|
|
class Replication {
|
|
source: any
|
|
target: any
|
|
replication: any
|
|
|
|
/**
|
|
*
|
|
* @param {String} source - the DB you want to replicate or rollback to
|
|
* @param {String} target - the DB you want to replicate to, or rollback from
|
|
*/
|
|
constructor({ source, target }: any) {
|
|
this.source = dangerousGetDB(source)
|
|
this.target = dangerousGetDB(target)
|
|
}
|
|
|
|
close() {
|
|
return Promise.all([closeDB(this.source), closeDB(this.target)])
|
|
}
|
|
|
|
promisify(operation: any, opts = {}) {
|
|
return new Promise(resolve => {
|
|
operation(this.target, opts)
|
|
.on("denied", function (err: any) {
|
|
// a document failed to replicate (e.g. due to permissions)
|
|
throw new Error(`Denied: Document failed to replicate ${err}`)
|
|
})
|
|
.on("complete", function (info: any) {
|
|
return resolve(info)
|
|
})
|
|
.on("error", function (err: any) {
|
|
throw new Error(`Replication Error: ${err}`)
|
|
})
|
|
})
|
|
}
|
|
|
|
/**
|
|
* Two way replication operation, intended to be promise based.
|
|
* @param {Object} opts - PouchDB replication options
|
|
*/
|
|
sync(opts = {}) {
|
|
this.replication = this.promisify(this.source.sync, opts)
|
|
return this.replication
|
|
}
|
|
|
|
/**
|
|
* One way replication operation, intended to be promise based.
|
|
* @param {Object} opts - PouchDB replication options
|
|
*/
|
|
replicate(opts = {}) {
|
|
this.replication = this.promisify(this.source.replicate.to, opts)
|
|
return this.replication
|
|
}
|
|
|
|
/**
|
|
* Rollback the target DB back to the state of the source DB
|
|
*/
|
|
async rollback() {
|
|
await this.target.destroy()
|
|
// Recreate the DB again
|
|
this.target = dangerousGetDB(this.target.name)
|
|
await this.replicate()
|
|
}
|
|
|
|
cancel() {
|
|
this.replication.cancel()
|
|
}
|
|
}
|
|
|
|
export default Replication
|