mincomk/risuai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[feat] new pngchunk encode/decoderr

Browse Source
This commit is contained in:
kwaroran
2024-01-03 04:13:19 +09:00
parent 239d819383
commit a8e50dcd77
10 changed files with 196 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/ts/characterCards.ts
View File

@@ -9,8 +9,9 @@ import { checkCharOrder, downloadFile, readImage, saveAsset } from "./storage/gl
import { cloneDeep } from "lodash"
import { selectedCharID } from "./stores"
import { convertImage } from "./parser"
import * as yuso from 'yuso'
import { reencodeImage } from "./image"
import { PngChunk } from "./pngChunk"
export const hubURL = "https://sv.risuai.xyz"
@@ -60,7 +61,11 @@ async function importCharacterProcess(f:{
await sleep(10)
const img = f.data
const readed = yuso.decode(img, 'chara')
const readed = PngChunk.read(img, ['chara'])?.['chara']
if(!readed){
alertError(language.errors.noData)
return
}
{
const charaData:CharacterCardV2 = JSON.parse(Buffer.from(readed, 'base64').toString('utf-8'))
if(await importSpecv2(charaData, img)){
@@ -504,13 +509,16 @@ export async function exportSpecV2(char:character, type:'png'|'json' = 'png') {
await sleep(10)
img = await reencodeImage(img)
img = yuso.encode(img, "chara",Buffer.from(JSON.stringify(card)).toString('base64'))
alertStore.set({
type: 'wait',
msg: 'Loading... (Writing)'
})
img = (await PngChunk.write(img, {
"chara":Buffer.from(JSON.stringify(card)).toString('base64')
})) as Uint8Array
char.image = ''
await sleep(10)
await downloadFile(`${char.name.replace(/[<>:"/\\|?*\.\,]/g, "")}_export.png`, img)
@@ -519,6 +527,7 @@ export async function exportSpecV2(char:character, type:'png'|'json' = 'png') {
}
catch(e){
console.error(e, e.stack)
alertError(`${e}`)
}
}

4
src/ts/characters.ts
View File

@@ -7,9 +7,9 @@ import { checkNullish, findCharacterbyId, selectMultipleFile, selectSingleFile,
import { v4 as uuidv4 } from 'uuid';
import { selectedCharID } from "./stores";
import { checkCharOrder, downloadFile, getFileSrc } from "./storage/globalApi";
import * as yuso from 'yuso'
import { reencodeImage } from "./image";
import { updateInlayScreen } from "./process/inlayScreen";
import { PngChunk } from "./pngChunk";
export function createNewCharacter() {
let db = get(DataBase)
@@ -473,7 +473,7 @@ export async function addDefaultCharacters() {
for(const img of imgs){
const imgBuffer = await (await img).arrayBuffer()
const readed = yuso.decode(Buffer.from(imgBuffer), "risuai")
const readed = PngChunk.read(Buffer.from(imgBuffer), ["risuai"])?.risuai
await sleep(10)
const va = decodeMsgpack(Buffer.from(readed,'base64')) as any
if(va.type !== 101){

6
src/ts/drive/backuplocal.ts
View File

@@ -50,7 +50,7 @@ export async function SaveLocalBackup(){
if(!key || !key.endsWith('.png')){
continue
}
await writer.write(key, await readBinaryFile(asset.path))
await writer.writeBackup(key, await readBinaryFile(asset.path))
}
}
else{
@@ -63,7 +63,7 @@ export async function SaveLocalBackup(){
if(!key || !key.endsWith('.png')){
continue
}
await writer.write(key, await forageStorage.getItem(key))
await writer.writeBackup(key, await forageStorage.getItem(key))
if(forageStorage.isAccount){
await sleep(1000)
}
@@ -74,7 +74,7 @@ export async function SaveLocalBackup(){
alertWait(`Saving local Backup... (Saving database)`)
await writer.write('database.risudat', dbData)
await writer.writeBackup('database.risudat', dbData)
alertNormal('Success')

1
src/ts/image.ts
View File

@@ -98,6 +98,7 @@ export function supportsInlayImage(){
}
export async function reencodeImage(img:Uint8Array){
const canvas = document.createElement('canvas')
const imgObj = new Image()
imgObj.src = URL.createObjectURL(new Blob([img], {type: `image/png`}))

13
src/ts/persona.ts
View File

@@ -2,11 +2,11 @@ import { get } from "svelte/store"
import { DataBase, saveImage, setDatabase } from "./storage/database"
import { selectSingleFile, sleep } from "./util"
import { alertError, alertNormal, alertStore } from "./alert"
import * as yuso from 'yuso'
import { downloadFile, readImage } from "./storage/globalApi"
import { language } from "src/lang"
import { cloneDeep } from "lodash"
import { reencodeImage } from "./image"
import { PngChunk } from "./pngChunk"
export async function selectUserImg() {
const selected = await selectSingleFile(['png'])
@@ -81,7 +81,9 @@ export async function exportUserPersona(){
await sleep(10)
img = yuso.encode(await reencodeImage(img), "persona",Buffer.from(JSON.stringify(card)).toString('base64'))
img = (await PngChunk.write(await reencodeImage(img), {
"persona":Buffer.from(JSON.stringify(card)).toString('base64')
})) as Uint8Array
alertStore.set({
type: 'wait',
@@ -98,7 +100,12 @@ export async function importUserPersona(){
try {
const v = await selectSingleFile(['png'])
const data:PersonaCard = JSON.parse(Buffer.from(yuso.decode(v.data, "persona"), 'base64').toString('utf-8'))
const decoded = PngChunk.read(v.data, ['persona'])?.persona
if(!decoded){
alertError(language.errors.noData)
return
}
const data:PersonaCard = JSON.parse(Buffer.from(decoded, 'base64').toString('utf-8'))
if(data.name && data.personaPrompt){
let db = get(DataBase)
db.personas.push({

144
src/ts/pngChunk.ts Normal file
View File

@@ -0,0 +1,144 @@
import { Buffer } from 'buffer';
import crc32 from 'crc/crc32';
import type { LocalWriter } from './storage/globalApi';
export const PngChunk = {
read: (data:Uint8Array, chunkName:string[], arg:{checkCrc?:boolean} = {}) => {
let pos = 8
let chunks:{[key:string]:string} = {}
while(pos < data.length){
const len = data[pos] * 0x1000000 + data[pos+1] * 0x10000 + data[pos+2] * 0x100 + data[pos+3]
const type = data.slice(pos+4,pos+8)
const typeString = new TextDecoder().decode(type)
console.log(typeString, len)
if(arg.checkCrc){
const crc = data[pos+8+len] * 0x1000000 + data[pos+9+len] * 0x10000 + data[pos+10+len] * 0x100 + data[pos+11+len]
const crcCheck = crc32(data.slice(pos+4,pos+8+len))
if(crc !== crcCheck){
throw new Error('crc check failed')
}
}
if(typeString === 'IEND'){
break
}
if(typeString === 'tEXt'){
const chunkData = data.slice(pos+8,pos+8+len)
let key=''
let value=''
for(let i=0;i<10;i++){
if(chunkData[i] === 0){
key = new TextDecoder().decode(chunkData.slice(0,i))
value = new TextDecoder().decode(chunkData.slice(i))
break
}
}
if(chunkName.includes(key)){
chunks[key] = value
}
}
pos += 12 + len
}
return chunks
},
trim: (data:Uint8Array) => {
let pos = 8
let newData:Uint8Array[] = []
while(pos < data.length){
const len = data[pos] * 0x1000000 + data[pos+1] * 0x10000 + data[pos+2] * 0x100 + data[pos+3]
const type = data.slice(pos+4,pos+8)
const typeString = new TextDecoder().decode(type)
if(typeString === 'IEND'){
newData.push(data.slice(pos,pos+12+len))
break
}
if(typeString === 'tEXt'){
pos += 12 + len
}
else{
newData.push(data.slice(pos,pos+12+len))
pos += 12 + len
}
}
newData.push(data.slice(pos))
return Buffer.concat(newData)
},
write: async (data:Uint8Array, chunks:{[key:string]:string}, options:{writer?:LocalWriter} = {}):Promise<void | Buffer> => {
let pos = 8
let newData:Uint8Array[] = []
async function pushData(data:Uint8Array){
if(options.writer){
await options.writer.write(data)
}
else{
newData.push(data)
}
}
await pushData(data.slice(0,8))
while(pos < data.length){
const len = data[pos] * 0x1000000 + data[pos+1] * 0x10000 + data[pos+2] * 0x100 + data[pos+3]
const type = data.slice(pos+4,pos+8)
const typeString = new TextDecoder().decode(type)
if(typeString === 'IEND'){
break
}
if(typeString === 'tEXt'){
pos += 12 + len
}
else{
await pushData(data.slice(pos,pos+12+len))
pos += 12 + len
}
}
for(const key in chunks){
const keyData = new TextEncoder().encode(key)
const value = Buffer.from(chunks[key])
const lenNum = value.byteLength + keyData.byteLength + 1
//idk, but uint32array is not working
const length = new Uint8Array([
lenNum / 0x1000000 % 0x100,
lenNum / 0x10000 % 0x100,
lenNum / 0x100 % 0x100,
lenNum % 0x100
])
const type = new TextEncoder().encode('tEXt')
await pushData(length)
await pushData(type)
await pushData(keyData)
await pushData(new Uint8Array([0]))
await pushData(value)
const crc = crc32(Buffer.concat([type,keyData,new Uint8Array([0]),value]))
await pushData(new Uint8Array([
crc / 0x1000000 % 0x100,
crc / 0x10000 % 0x100,
crc / 0x100 % 0x100,
crc % 0x100
]))
}
//create IEND chunk
{
const length = new Uint8Array((new Uint32Array([0])).buffer)
const type = new TextEncoder().encode('IEND')
await pushData(length)
await pushData(type)
const crc = crc32(type)
await pushData(new Uint8Array([
crc / 0x1000000 % 0x100,
crc / 0x10000 % 0x100,
crc / 0x100 % 0x100,
crc % 0x100
]))
}
if(options.writer){
await options.writer.close()
}
else{
return Buffer.concat(newData)
}
},
}

5
src/ts/storage/globalApi.ts
View File

@@ -1195,7 +1195,7 @@ export class LocalWriter{
this.writer = writableStream.getWriter()
return true
}
async write(name:string,data: Uint8Array){
async writeBackup(name:string,data: Uint8Array){
const encodedName = new TextEncoder().encode(getBasename(name))
const nameLength = new Uint32Array([encodedName.byteLength])
await this.writer.write(new Uint8Array(nameLength.buffer))
@@ -1204,6 +1204,9 @@ export class LocalWriter{
await this.writer.write(new Uint8Array(dataLength.buffer))
await this.writer.write(data)
}
async write(data:Uint8Array) {
await this.writer.write(data)
}
async close(){
await this.writer.close()
}