ProjectNotes/ProjectNotes.py at master · budlime/ProjectNotes · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
# Copyright (c) 2022, budRich. All rights reserved.
# SPDX-License-Identifier: 0BSD

import sublime
import sublime_plugin

from os.path import isdir, isfile, expanduser, split, join, basename, splitext
from os import getenv, makedirs, listdir

from typing import List

def tilde_prefix(target: str):
    home = getenv("HOME")
    target = expanduser(target)

    if home and target.startswith(home):
        target = "~" + target[len(home) :]

    return target

def notes_root_dir(settings: sublime.Settings):
    note_dir = settings.setdefault("notes_dir_path", "budlabs made me do it")
    note_dir = expanduser(note_dir)

    if not isdir(note_dir):
        note_dir = join(sublime.packages_path(), "User", "notes")

    return note_dir


def add_directory_to_project(target: str, name: str) -> None:
    win = sublime.active_window()
    project_data = win.project_data() or {}
    project_folders = project_data.get("folders") or []

    directory = tilde_prefix(target)

    folder = dict(
        path=directory,
        name=name,
    )

    if all(folder["path"] != directory for folder in project_folders):
        project_data.setdefault("folders", []).append(folder)
        win.set_project_data(project_data)

class ProjectNotesListCommand(sublime_plugin.WindowCommand):

    notes: List[str]
    settings: sublime.Settings
    note_dir: str

    def run(self):

        self.settings = sublime.load_settings("ProjectNotes.sublime-settings")
        self.notes = []
        self.note_dir = notes_root_dir(self.settings)

        for dir in listdir(self.note_dir):
            note_path = join(self.note_dir,dir,self.settings.setdefault("note_file_name", "todo"))
            if isfile(note_path):
                self.notes.append(dir)

        self.window.show_quick_panel(self.notes, self.on_done)

    def on_done(self, i: int):

        # -1, escape was pressed
        if (i == -1): return

        dir = self.notes[i]
        note_path = join(self.note_dir, dir, self.settings.setdefault("note_file_name", "todo"))
        # we know that the file exist here
        sublime.active_window().open_file(note_path)

class ProjectNotesOpenCommand(sublime_plugin.WindowCommand):

    def run(self):
        project_path = self.window.project_file_name()

        # current window has no associated project
        if not project_path: return

        settings = sublime.load_settings("ProjectNotes.sublime-settings")

        project_name, _ = splitext(basename(project_path))
        note_name = settings.setdefault("note_file_name", "todo")
        note_dir = notes_root_dir(settings)

        note_path = join(note_dir, project_name, note_name)

        note_dir, _ = split(note_path)

        if not isdir(note_dir): makedirs(note_dir)

        if (settings.get("add_note_dir_to_project")):
            add_directory_to_project(note_dir, settings.setdefault("note_dir_name", "notes"))

        if not isfile(note_path):
            sublime.status_message("Created new buffer '" + note_path + "'")
        sublime.active_window().open_file(note_path)