• Index
• Search by APIs
• Search by Words
• Project Search

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• vscode-drawio-master
  • .github
    • FUNDING.yml
    • workflows
      • test.yml
      • master.yml
      • release.yml
      • opened-issues-triage.yml
  • src
    • DrawioExtensionApi.ts
    • utils
      • formatValue.ts
      • path.ts
      • mapObject.ts
      • buffer.ts
      • fromResource.ts
      • autorunTrackDisposables.ts
      • SimpleTemplate.ts
      • groupBy.ts
      • registerFailableCommand.ts
    • DrawioClient
      • webview-content.html
      • html.d.ts
      • CustomizedDrawioClient.ts
      • index.ts
      • simpleDrawioLibrary.ts
      • DrawioClientFactory.ts
      • DrawioClient.ts
      • DrawioTypes.ts
    • DrawioEditorProviderText.ts
    • vscode-utils
      • VirtualFileSystemProvider.ts
      • VsCodeSetting.ts
    • index.ts
    • Config.ts
    • DrawioEditorService.ts
    • types.d.ts
    • features
      • ActivtyTracking.ts
      • EditDiagramAsTextFeature.ts
      • CodeLinkFeature.ts
      • LiveshareFeature
        • index.ts
        • LiveshareFeature.ts
        • CurrentViewState.ts
        • LiveshareSession.ts
        • SessionModel.ts
        • assets
          • package.json
    • DrawioEditorProviderBinary.ts
    • Extension.ts
  • .prettierrc.json
  • NOTES.md
  • .vscodeignore
  • examples
    • use-cases
      • wireframes.dio
      • class-diagrams.dio
      • cloud-architecture.drawio
      • packages.dio
    • tooltips-plugin.js
    • linking
      • main.dio
      • demo-src
        • Baz.ts
        • tsconfig.json
        • Foo.ts
        • test.cc
    • temp
      • Example.drawio
      • TestLibrary.xml
    • data
    • .vscode
      • settings.json
    • formats
      • README.md
  • .gitmodules
  • tslint.json
  • webpack.config.ts
  • .gitattributes
  • CHANGELOG.md
  • LICENSE.md
  • README.md
  • package.json
  • scripts
    • check-version.ts
    • pub.ts
    • run-script.js
    • prepare-insiders-build
      • README_INSIDERS_BUILD.md
      • index.ts
      • package-insiders-build.json
    • shared.ts
    • trigger-prerelease.ts
    • tsconfig.json
  • tsconfig.json
  • .gitignore
  • .prettierignore
  • docs
    • code-link.md
  • .vscode
    • settings.json
    • launch.json
    • tasks.json
  • drawio-custom-plugins
    • src
      • donationDialog.ts
      • linkSelectedNodeWithData.ts
      • styles.css
      • propertiesDialog.ts
      • index.ts
      • menu-entries.ts
      • focus.ts
      • drawio-types.d.ts
      • types.d.ts
      • vscode.ts
      • liveshare.ts
    • webpack.config.ts
    • tsconfig.json

import {
	FileSystemProvider,
	Event,
	Uri,
	FileStat,
	FileType,
	FileChangeEvent,
	EventEmitter,
	workspace,
	FileChangeType,
} from "vscode";
import { Disposable } from "@hediet/std/disposable";
import { BufferImpl } from "../utils/buffer";

export class DrawioFileSystemController {
	public readonly dispose = Disposable.fn();

	private readonly fs = new VirtualFileSystemProvider();
	public readonly scheme = "drawio";

	constructor() {
		this.dispose.track(
			workspace.registerFileSystemProvider(this.scheme, this.fs, {
				isCaseSensitive: true,
				isReadonly: false,
			})
		);
	}

	public getOrCreateFileForUri(uri: Uri): {
		file: File;
		didFileExist: boolean;
	} {
		return this.fs.getOrCreateFile(uri);
	}

	/*public getRandomFile(extensionWithDot: string): File {
		const id1 = new Date().getTime();
		const id2 = id++;
		return this.fs.getOrCreateFile(`/${id1}_${id2}${extensionWithDot}`);
	}*/
}

export class VirtualFileSystemProvider implements FileSystemProvider {
	private fileChangedEmitter = new EventEmitter<FileChangeEvent[]>();
	public readonly onDidChangeFile = this.fileChangedEmitter.event;

	private readonly files = new Map<string, File>();

	public getOrCreateFile(uri: Uri): { file: File; didFileExist: boolean } {
		const key = uri.toString();

		const f = this.files.get(key);
		if (f) {
			return { file: f, didFileExist: true };
		}

		const newFile = new File(uri, Uint8Array.from([]));
		newFile.onDidChangeFile(() =>
			this.fileChangedEmitter.fire([
				{ type: FileChangeType.Changed, uri: newFile.uri },
			])
		);
		this.files.set(key, newFile);
		return { file: newFile, didFileExist: false };
	}

	readFile(uri: Uri): Uint8Array | Thenable<Uint8Array> {
		return this.getOrCreateFile(uri).file.data;
	}

	writeFile(
		uri: Uri,
		content: Uint8Array,
		options: { create: boolean; overwrite: boolean }
	): void | Thenable<void> {
		return this.getOrCreateFile(uri).file.write(content);
	}

	stat(uri: Uri): FileStat {
		const f = this.getOrCreateFile(uri).file;
		return {
			type: FileType.File,
			ctime: 0,
			mtime: 0,
			size: f.data.length,
		};
	}

	watch(
		uri: Uri,
		options: { recursive: boolean; excludes: string[] }
	): Disposable {
		return Disposable.empty;
	}

	readDirectory(
		uri: Uri
	):
		| [string, import("vscode").FileType][]
		| Thenable<[string, import("vscode").FileType][]> {
		throw new Error("Method not implemented.");
	}

	createDirectory(uri: Uri): void | Thenable<void> {
		throw new Error("Method not implemented.");
	}

	delete(uri: Uri, options: { recursive: boolean }): void | Thenable<void> {
		throw new Error("Method not implemented.");
	}

	rename(
		oldUri: Uri,
		newUri: Uri,
		options: { overwrite: boolean }
	): void | Thenable<void> {
		throw new Error("Method not implemented.");
	}
}

export class File {
	private readonly fileChangedEmitter = new EventEmitter();
	public readonly onDidChangeFile = this.fileChangedEmitter.event;

	constructor(public readonly uri: Uri, public data: Uint8Array) {}

	public write(data: Uint8Array): void {
		this.data = data;
		this.fileChangedEmitter.fire(undefined);
	}

	public writeString(str: string): void {
		this.write(Uint8Array.from(BufferImpl.from(str, "utf-8")));
	}

	public readString(): string {
		return BufferImpl.from(this.data).toString("utf-8");
	}
}