@struktoai/mirage-core
Version:
A Unified Virtual Filesystem For AI Agents
89 lines • 3.8 kB
JavaScript
// ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
import { CONDITION_SUPPORT } from "./constants.js";
import { requireSupported } from "../../cache/context.js";
import { chunks } from "../../io/cooperative.js";
import { ByteSpool } from "../../io/spool.js";
import { closeQuietly } from "../../io/stream.js";
import { nativeCondition, writeCondition } from "../../cache/context.js";
import { record, startOp } from "../../observe/context.js";
import { uploadStat } from "../../utils/upload.js";
import { lookup, refused } from "./api.js";
import { liveOf } from "./fingerprint.js";
import { dropboxUpload } from "./client.js";
import { checkPath, dropboxPathOf } from "./paths.js";
import { statFromEntry } from "./readdir.js";
// Large spooled inputs use upload sessions, and a stream is spooled first, as
// its size picks the upload; `signal` stops it. A held version costs one
// get_metadata and goes out as the file's rev in `update` mode. Answers the
// stored file's stat, null for a reply that names none.
export async function write(accessor, path, data, signal) {
if (!(data instanceof Uint8Array) && !(data instanceof ByteSpool)) {
const spool = new ByteSpool();
try {
await spool.fill(chunks(data, signal));
return await write(accessor, path, spool);
}
finally {
await spool.close();
}
}
const condition = await writeCondition(path, 'put');
await checkPath(accessor, path);
return uploadPrepared(accessor, path, data, condition);
}
export async function replace(accessor, request, signal) {
try {
await checkPath(accessor, request.path);
requireSupported(CONDITION_SUPPORT, 'dropbox', request.path, 'put', request.condition);
}
catch (error) {
await closeQuietly(request.body);
throw error;
}
if (!(request.body instanceof Uint8Array)) {
const spool = new ByteSpool();
try {
await spool.fill(chunks(request.body, signal));
return await uploadPrepared(accessor, request.path, spool, request.condition);
}
finally {
await spool.close();
}
}
return uploadPrepared(accessor, request.path, request.body, request.condition);
}
async function uploadPrepared(accessor, path, data, cond) {
const tm = accessor.tokenManager;
const apiPath = dropboxPathOf(accessor, path);
requireSupported(CONDITION_SUPPORT, 'dropbox', path, 'put', cond);
let rev = null;
if (cond.kind === 'match' && cond.version !== '') {
rev = await nativeCondition(path, cond, liveOf(await lookup(tm, apiPath)), 'put');
}
const timer = startOp();
let entry;
try {
entry = await dropboxUpload(tm, apiPath, data, rev);
}
catch (err) {
throw (await refused(path, err, cond, rev)) ?? err;
}
const stored = uploadStat(entry, statFromEntry, path.virtual);
record('write', path.virtual, 'dropbox', data instanceof ByteSpool ? data.size : data.byteLength, timer, {
fingerprint: stored?.fingerprint ?? null,
});
return stored;
}
//# sourceMappingURL=write.js.map