Skip to content

Commit

Permalink
WIP: Implement support for multiple notes
Browse files Browse the repository at this point in the history
Refactor Vue <-> Editor <-> CodeMirror code.
Introduce Pinia store to keep global state, in order to get rid of a lot of event juggling between Editor class/child components and the root App component.
  • Loading branch information
heyman committed Jul 24, 2024
1 parent f156320 commit 487a4cb
Show file tree
Hide file tree
Showing 20 changed files with 1,175 additions and 291 deletions.
48 changes: 27 additions & 21 deletions electron/main/buffer.js
Original file line number Diff line number Diff line change
Expand Up @@ -16,15 +16,15 @@ const untildify = (pathWithTilde) => {
: pathWithTilde;
}

export function constructBufferFilePath(directoryPath) {
return join(untildify(directoryPath), isDev ? "buffer-dev.txt" : "buffer.txt")
export function constructBufferFilePath(directoryPath, path) {
return join(untildify(directoryPath), path)
}

export function getBufferFilePath() {
export function getFullBufferFilePath(path) {
let defaultPath = app.getPath("userData")
let configPath = CONFIG.get("settings.bufferPath")
let bufferPath = configPath.length ? configPath : defaultPath
let bufferFilePath = constructBufferFilePath(bufferPath)
let bufferFilePath = constructBufferFilePath(bufferPath, path)
try {
// use realpathSync to resolve a potential symlink
return fs.realpathSync(bufferFilePath)
Expand Down Expand Up @@ -103,39 +103,45 @@ export class Buffer {


// Buffer
let buffer
export function loadBuffer() {
if (buffer) {
buffer.close()
let buffers = {}
export function loadBuffer(path) {
if (buffers[path]) {
buffers[path].close()
}
buffer = new Buffer({
filePath: getBufferFilePath(),
buffers[path] = new Buffer({
filePath: getFullBufferFilePath(path),
onChange: (content) => {
win?.webContents.send("buffer-content:change", content)
console.log("Old buffer.js onChange")
win?.webContents.send("buffer-content:change", path, content)
},
})
return buffer
return buffers[path]
}

ipcMain.handle('buffer-content:load', async () => {
if (buffer.exists() && !(eraseInitialContent && isDev)) {
return await buffer.load()
ipcMain.handle('buffer-content:load', async (event, path) => {
if (!buffers[path]) {
loadBuffer(path)
}
if (buffers[path].exists() && !(eraseInitialContent && isDev)) {
return await buffers[path].load()
} else {
return isDev ? initialDevContent : initialContent
}
});

async function save(content) {
return await buffer.save(content)
async function save(path, content) {
return await buffers[path].save(content)
}

ipcMain.handle('buffer-content:save', async (event, content) => {
return await save(content)
ipcMain.handle('buffer-content:save', async (event, path, content) => {
return await save(path, content)
});

export let contentSaved = false
ipcMain.handle('buffer-content:saveAndQuit', async (event, content) => {
await save(content)
ipcMain.handle('buffer-content:saveAndQuit', async (event, contents) => {
for (const [path, content] of contents) {
await save(path, content)
}
contentSaved = true
app.quit()
})
Expand Down
210 changes: 210 additions & 0 deletions electron/main/file-library.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,210 @@
import fs from "fs"
import os from "node:os"
import { join, dirname, basename } from "path"

import * as jetpack from "fs-jetpack";
import { app, ipcMain, dialog } from "electron"


const untildify = (pathWithTilde) => {
const homeDir = os.homedir()
return homeDir ? pathWithTilde.replace(/^~(?=$|\/|\\)/, homeDir) : pathWithTilde
}

async function readNoteMetadata(filePath) {
const chunks = []
for await (let chunk of fs.createReadStream(filePath, { start: 0, end:4000 })) {
chunks.push(chunk)
}
const headContent = Buffer.concat(chunks).toString("utf8")
const firstSeparator = headContent.indexOf("\n∞∞∞")
if (firstSeparator === -1) {
return null
}
try {
const metadata = JSON.parse(headContent.slice(0, firstSeparator).trim())
return {"name": metadata.name, "tags": metadata.tags}
} catch (e) {
return {}
}
}


export class FileLibrary {
constructor(basePath) {
basePath = untildify(basePath)
if (jetpack.exists(basePath) !== "dir") {
throw new Error(`Path directory does not exist: ${basePath}`)
}
this.basePath = fs.realpathSync(basePath)
this.jetpack = jetpack.cwd(this.basePath)
this.files = {};
this.watcher = null;
this.contentSaved = false
this.onChangeCallback = null
}

async exists(path) {
return this.jetpack.exists(path) === "file"
}

async load(path) {
if (this.files[path]) {
return this.files[path].read()
}
const fullPath = fs.realpathSync(join(this.basePath, path))
this.files[path] = new NoteBuffer({fullPath, library:this})
return await this.files[path].read()
}

async save(path, content) {
if (!this.files[path]) {
throw new Error(`File not loaded: ${path}`)
}
return await this.files[path].save(content)
}

async getList() {
console.log("Loading notes")
const notes = {}
const files = await this.jetpack.findAsync(this.basePath, {
matching: "*.txt",
recursive: true,
})
const promises = []
for (const file of files) {
promises.push(readNoteMetadata(join(this.basePath, file)))
}
const metadataList = await Promise.all(promises)
metadataList.forEach((metadata, i) => {
const path = files[i]
notes[path] = metadata
})
return notes
}

setupWatcher(win) {
if (!this.watcher) {
this.watcher = fs.watch(
this.basePath,
{
persistent: true,
recursive: true,
encoding: "utf8",
},
async (eventType, changedPath) => {
console.log("File changed", eventType, changedPath)
//if (changedPath.toLowerCase().endsWith(".txt")) {
// console.log("txt", this.notes)
// if (await this.exists(changedPath)) {
// console.log("file exists!")
// const newMetadata = await readNoteMetadata(join(this.basePath, changedPath))
// if (!(changedPath in this.notes) || newMetadata.name !== this.notes[changedPath].name) {
// this.notes[changedPath] = newMetadata
// win.webContents.send("buffer:noteMetadataChanged", changedPath, newMetadata)
// console.log("metadata changed")
// } else {
// console.log("no metadata change")
// }
// } else if (changedPath in this.notes) {
// console.log("note removed", changedPath)
// delete this.notes[changedPath]
// win.webContents.send("buffer:noteRemoved", changedPath)
// }
//}
for (const [path, buffer] of Object.entries(this.files)) {
if (changedPath === basename(path)) {
const content = await buffer.read()
if (buffer._lastSavedContent !== content) {
win.webContents.send("buffer:change", path, content)
}
}
}
}
)
}
}

closeFile(path) {
if (this.files[path]) {
delete this.files[path]
}
}

close() {
for (const buffer of Object.values(this.files)) {
this.closeFile(buffer.filePath)
}
this.stopWatcher()
}

stopWatcher() {
if (this.watcher) {
this.watcher.close()
this.watcher = null
}
}
}



export class NoteBuffer {
constructor({fullPath, library}) {
this.fullPath = fullPath
this._lastSavedContent = null
this.library = library
}

async read() {
return await this.library.jetpack.read(this.fullPath, 'utf8')
}

async save(content) {
this._lastSavedContent = content
const saveResult = await this.library.jetpack.write(this.fullPath, content, {
atomic: true,
mode: '600',
})
return saveResult
}

exists() {
return jetpack.exists(this.fullPath) === "file"
}
}


export function setupFileLibraryEventHandlers(library, win) {
ipcMain.handle('buffer:load', async (event, path) => {
console.log("buffer:load", path)
return await library.load(path)
});


ipcMain.handle('buffer:save', async (event, path, content) => {
return await library.save(path, content)
});

ipcMain.handle('buffer:getList', async (event) => {
return await library.getList()
});

ipcMain.handle('buffer:exists', async (event, path) => {
return await library.exists(path)
});

ipcMain.handle('buffer:close', async (event, path) => {
return await library.closeFile(path)
});

ipcMain.handle('buffer:saveAndQuit', async (event, contents) => {
library.stopWatcher()
for (const [path, content] of contents) {
await library.save(path, content)
}
library.contentSaved = true
app.quit()
})

library.setupWatcher(win)
}
17 changes: 14 additions & 3 deletions electron/main/index.ts
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ import { isDev, isLinux, isMac, isWindows } from '../detect-platform';
import { initializeAutoUpdate, checkForUpdates } from './auto-update';
import { fixElectronCors } from './cors';
import { loadBuffer, contentSaved } from './buffer';
import { FileLibrary, setupFileLibraryEventHandlers } from './file-library';


// The built directory structure
Expand Down Expand Up @@ -49,6 +50,7 @@ Menu.setApplicationMenu(menu)
// process.env['ELECTRON_DISABLE_SECURITY_WARNINGS'] = 'true'

export let win: BrowserWindow | null = null
let fileLibrary: FileLibrary | null = null
let tray: Tray | null = null;
let initErrors: string[] = []
// Here, you can also use other preload
Expand Down Expand Up @@ -139,7 +141,7 @@ async function createWindow() {
}
// Prevent the window from closing, and send a message to the renderer which will in turn
// send a message to the main process to save the current buffer and close the window.
if (!contentSaved) {
if (!!fileLibrary && !fileLibrary.contentSaved) {
event.preventDefault()
win?.webContents.send(WINDOW_CLOSE_EVENT)
} else {
Expand Down Expand Up @@ -303,6 +305,7 @@ function registerAlwaysOnTop() {
}

app.whenReady().then(createWindow).then(async () => {
setupFileLibraryEventHandlers(fileLibrary, win)
initializeAutoUpdate(win)
registerGlobalHotkey()
registerShowInDock()
Expand Down Expand Up @@ -344,8 +347,16 @@ ipcMain.handle('dark-mode:set', (event, mode) => {

ipcMain.handle('dark-mode:get', () => nativeTheme.themeSource)

// load buffer on app start
loadBuffer()
// Initialize note/file library
const customLibraryPath = CONFIG.get("settings.bufferPath")
const libraryPath = customLibraryPath ? customLibraryPath : join(app.getPath("userData"), "notes")
console.log("libraryPath", libraryPath)
try {
fileLibrary = new FileLibrary(libraryPath)
} catch (error) {
initErrors.push(`Error: ${error.message}`)
}

ipcMain.handle("getInitErrors", () => {
return initErrors
})
Expand Down
Loading

0 comments on commit 487a4cb

Please sign in to comment.