82 lines
2.7 KiB
TypeScript
82 lines
2.7 KiB
TypeScript
import localforage from "localforage"
|
|
import { isNodeServer } from "./globalApi"
|
|
import { NodeStorage } from "./nodeStorage"
|
|
import { OpfsStorage } from "./opfsStorage"
|
|
import { alertConfirm, alertStore } from "../alert"
|
|
|
|
export class AutoStorage{
|
|
|
|
realStorage:LocalForage|NodeStorage|OpfsStorage
|
|
|
|
async setItem(key:string, value:Uint8Array) {
|
|
await this.Init()
|
|
return await this.realStorage.setItem(key, value)
|
|
}
|
|
async getItem(key:string):Promise<Buffer> {
|
|
await this.Init()
|
|
return await this.realStorage.getItem(key)
|
|
|
|
}
|
|
async keys():Promise<string[]>{
|
|
await this.Init()
|
|
return await this.realStorage.keys()
|
|
|
|
}
|
|
async removeItem(key:string){
|
|
await this.Init()
|
|
return await this.realStorage.removeItem(key)
|
|
}
|
|
|
|
private async Init(){
|
|
if(!this.realStorage){
|
|
if(isNodeServer){
|
|
console.log("using node storage")
|
|
this.realStorage = new NodeStorage()
|
|
return
|
|
}
|
|
else if(window.navigator?.storage?.getDirectory &&
|
|
FileSystemFileHandle?.prototype?.createWritable &&
|
|
localStorage.getItem('opfs_flag!') === "able"){
|
|
console.log("using opfs storage")
|
|
|
|
const forage = localforage.createInstance({
|
|
name: "risuai"
|
|
})
|
|
|
|
const i = await forage.getItem("database/database.bin")
|
|
|
|
if((!i) || (await forage.getItem("migrated"))){
|
|
this.realStorage = new OpfsStorage()
|
|
return
|
|
}
|
|
else if(!(await forage.getItem("denied_opfs"))){
|
|
console.log("migrating")
|
|
const keys = await forage.keys()
|
|
let i = 0;
|
|
const opfs = new OpfsStorage()
|
|
for(const key of keys){
|
|
alertStore.set({
|
|
type: "wait",
|
|
msg: `Migrating your data...(${i}/${keys.length})`
|
|
})
|
|
await opfs.setItem(key,await forage.getItem(key))
|
|
i += 1
|
|
}
|
|
this.realStorage = opfs
|
|
alertStore.set({
|
|
type: "none",
|
|
msg: ""
|
|
})
|
|
await forage.setItem("migrated", true)
|
|
return
|
|
}
|
|
}
|
|
console.log("using forage storage")
|
|
this.realStorage = localforage.createInstance({
|
|
name: "risuai"
|
|
})
|
|
}
|
|
}
|
|
|
|
listItem = this.keys
|
|
} |