mirror of
https://github.com/zadam/trilium.git
synced 2025-10-20 07:08:55 +02:00
429 lines
16 KiB
TypeScript
429 lines
16 KiB
TypeScript
import { t } from "../../services/i18n.js";
|
||
import dateNoteService from "../../services/date_notes.js";
|
||
import server from "../../services/server.js";
|
||
import appContext from "../../components/app_context.js";
|
||
import RightDropdownButtonWidget from "./right_dropdown_button.js";
|
||
import toastService from "../../services/toast.js";
|
||
import options from "../../services/options.js";
|
||
import { Dropdown } from "bootstrap";
|
||
import type { EventData } from "../../components/app_context.js";
|
||
import dayjs, { Dayjs } from "dayjs";
|
||
import isoWeek from "dayjs/plugin/isoWeek.js";
|
||
import utc from "dayjs/plugin/utc.js";
|
||
import isSameOrAfter from "dayjs/plugin/isSameOrAfter.js";
|
||
import "../../stylesheets/calendar.css";
|
||
import type { AttributeRow, OptionDefinitions } from "@triliumnext/commons";
|
||
|
||
dayjs.extend(utc);
|
||
dayjs.extend(isSameOrAfter);
|
||
dayjs.extend(isoWeek);
|
||
|
||
const MONTHS = [
|
||
t("calendar.january"),
|
||
t("calendar.february"),
|
||
t("calendar.march"),
|
||
t("calendar.april"),
|
||
t("calendar.may"),
|
||
t("calendar.june"),
|
||
t("calendar.july"),
|
||
t("calendar.august"),
|
||
t("calendar.september"),
|
||
t("calendar.october"),
|
||
t("calendar.november"),
|
||
t("calendar.december")
|
||
];
|
||
|
||
const DROPDOWN_TPL = `
|
||
<div class="calendar-dropdown-widget">
|
||
<style>
|
||
.calendar-dropdown-widget {
|
||
width: 400px;
|
||
}
|
||
</style>
|
||
|
||
<div class="calendar-header">
|
||
<div class="calendar-month-selector">
|
||
<button class="calendar-btn tn-tool-button bx bx-chevron-left" data-calendar-toggle="previous"></button>
|
||
|
||
<button class="btn dropdown-toggle select-button" type="button"
|
||
data-bs-toggle="dropdown" data-bs-auto-close="true"
|
||
aria-expanded="false"
|
||
data-calendar-input="month"></button>
|
||
<ul class="dropdown-menu" data-calendar-input="month-list">
|
||
${Object.entries(MONTHS)
|
||
.map(([i, month]) => `<li><button class="dropdown-item" data-value=${i}>${month}</button></li>`)
|
||
.join("")}
|
||
</ul>
|
||
|
||
<button class="calendar-btn tn-tool-button bx bx-chevron-right" data-calendar-toggle="next"></button>
|
||
</div>
|
||
|
||
<div class="calendar-year-selector">
|
||
<button class="calendar-btn tn-tool-button bx bx-chevron-left" data-calendar-toggle="previousYear"></button>
|
||
|
||
<input type="number" min="1900" max="2999" step="1" data-calendar-input="year" />
|
||
|
||
<button class="calendar-btn tn-tool-button bx bx-chevron-right" data-calendar-toggle="nextYear"></button>
|
||
</div>
|
||
</div>
|
||
|
||
<div class="calendar-week"></div>
|
||
<div class="calendar-body" data-calendar-area="month"></div>
|
||
</div>`;
|
||
|
||
const DAYS_OF_WEEK = [
|
||
t("calendar.sun"),
|
||
t("calendar.mon"),
|
||
t("calendar.tue"),
|
||
t("calendar.wed"),
|
||
t("calendar.thu"),
|
||
t("calendar.fri"),
|
||
t("calendar.sat")
|
||
];
|
||
|
||
interface DateNotesForMonth {
|
||
[date: string]: string;
|
||
}
|
||
|
||
interface WeekCalculationOptions {
|
||
firstWeekType: number;
|
||
minDaysInFirstWeek: number;
|
||
}
|
||
|
||
export default class CalendarWidget extends RightDropdownButtonWidget {
|
||
private $month!: JQuery<HTMLElement>;
|
||
private $weekHeader!: JQuery<HTMLElement>;
|
||
private $monthSelect!: JQuery<HTMLElement>;
|
||
private $yearSelect!: JQuery<HTMLElement>;
|
||
private $next!: JQuery<HTMLElement>;
|
||
private $previous!: JQuery<HTMLElement>;
|
||
private $nextYear!: JQuery<HTMLElement>;
|
||
private $previousYear!: JQuery<HTMLElement>;
|
||
private monthDropdown!: Dropdown;
|
||
// stored in ISO 1–7
|
||
private firstDayOfWeekISO!: number;
|
||
private weekCalculationOptions!: WeekCalculationOptions;
|
||
private activeDate: Dayjs | null = null;
|
||
private todaysDate!: Dayjs;
|
||
private date!: Dayjs;
|
||
private weekNoteEnable: boolean = false;
|
||
private weekNotes: string[] = [];
|
||
|
||
constructor(title: string = "", icon: string = "") {
|
||
super(title, icon, DROPDOWN_TPL);
|
||
}
|
||
|
||
doRender() {
|
||
super.doRender();
|
||
|
||
this.$month = this.$dropdownContent.find('[data-calendar-area="month"]');
|
||
this.$weekHeader = this.$dropdownContent.find(".calendar-week");
|
||
|
||
this.manageFirstDayOfWeek();
|
||
this.initWeekCalculation();
|
||
|
||
// Month navigation
|
||
this.$monthSelect = this.$dropdownContent.find('[data-calendar-input="month"]');
|
||
this.$monthSelect.on("show.bs.dropdown", (e) => {
|
||
// Don't trigger dropdownShown() at widget level when the month selection dropdown is shown, since it would cause a redundant refresh.
|
||
e.stopPropagation();
|
||
});
|
||
this.monthDropdown = Dropdown.getOrCreateInstance(this.$monthSelect[0]);
|
||
this.$dropdownContent.find('[data-calendar-input="month-list"] button').on("click", (e) => {
|
||
const target = e.target as HTMLElement;
|
||
const value = target.dataset.value;
|
||
if (value) {
|
||
this.date = this.date.month(parseInt(value));
|
||
this.createMonth();
|
||
}
|
||
});
|
||
|
||
this.$next = this.$dropdownContent.find('[data-calendar-toggle="next"]');
|
||
this.$next.on("click", () => {
|
||
this.date = this.date.add(1, 'month');
|
||
this.createMonth();
|
||
});
|
||
this.$previous = this.$dropdownContent.find('[data-calendar-toggle="previous"]');
|
||
this.$previous.on("click", () => {
|
||
this.date = this.date.subtract(1, 'month');
|
||
this.createMonth();
|
||
});
|
||
|
||
// Year navigation
|
||
this.$yearSelect = this.$dropdownContent.find('[data-calendar-input="year"]');
|
||
this.$yearSelect.on("input", (e) => {
|
||
const target = e.target as HTMLInputElement;
|
||
this.date = this.date.year(parseInt(target.value));
|
||
this.createMonth();
|
||
});
|
||
|
||
this.$nextYear = this.$dropdownContent.find('[data-calendar-toggle="nextYear"]');
|
||
this.$nextYear.on("click", () => {
|
||
this.date = this.date.add(1, 'year');
|
||
this.createMonth();
|
||
});
|
||
|
||
this.$previousYear = this.$dropdownContent.find('[data-calendar-toggle="previousYear"]');
|
||
this.$previousYear.on("click", () => {
|
||
this.date = this.date.subtract(1, 'year');
|
||
this.createMonth();
|
||
});
|
||
|
||
// Date click
|
||
this.$dropdownContent.on("click", ".calendar-date", async (ev) => {
|
||
const date = $(ev.target).closest(".calendar-date").attr("data-calendar-date");
|
||
if (date) {
|
||
const note = await dateNoteService.getDayNote(date);
|
||
if (note) {
|
||
appContext.tabManager.getActiveContext()?.setNote(note.noteId);
|
||
this.dropdown?.hide();
|
||
} else {
|
||
toastService.showError(t("calendar.cannot_find_day_note"));
|
||
}
|
||
}
|
||
ev.stopPropagation();
|
||
});
|
||
|
||
// Week click
|
||
this.$dropdownContent.on("click", ".calendar-week-number", async (ev) => {
|
||
if (!this.weekNoteEnable) {
|
||
return;
|
||
}
|
||
|
||
const week = $(ev.target).closest(".calendar-week-number").attr("data-calendar-week-number");
|
||
|
||
if (week) {
|
||
const note = await dateNoteService.getWeekNote(week);
|
||
|
||
if (note) {
|
||
appContext.tabManager.getActiveContext()?.setNote(note.noteId);
|
||
this.dropdown?.hide();
|
||
} else {
|
||
toastService.showError(t("calendar.cannot_find_week_note"));
|
||
}
|
||
}
|
||
|
||
ev.stopPropagation();
|
||
});
|
||
|
||
// Handle click events for the entire calendar widget
|
||
this.$dropdownContent.on("click", (e) => {
|
||
const $target = $(e.target);
|
||
|
||
// Keep dropdown open when clicking on month select button or year selector area
|
||
if ($target.closest('.btn.dropdown-toggle.select-button').length ||
|
||
$target.closest('.calendar-year-selector').length) {
|
||
e.stopPropagation();
|
||
return;
|
||
}
|
||
|
||
// Hide dropdown for all other cases
|
||
this.monthDropdown.hide();
|
||
// Prevent dismissing the calendar popup by clicking on an empty space inside it.
|
||
e.stopPropagation();
|
||
});
|
||
}
|
||
|
||
private async getWeekNoteEnable() {
|
||
const noteId = await server.get<string[]>(`search/${encodeURIComponent('#calendarRoot')}`);
|
||
if (noteId.length === 0) {
|
||
this.weekNoteEnable = false;
|
||
return;
|
||
}
|
||
const noteAttributes = await server.get<AttributeRow[]>(`notes/${noteId}/attributes`);
|
||
this.weekNoteEnable = noteAttributes.some(a => a.name === 'enableWeekNote');
|
||
}
|
||
|
||
// Store firstDayOfWeek as ISO (1–7)
|
||
manageFirstDayOfWeek() {
|
||
const rawFirstDayOfWeek = options.getInt("firstDayOfWeek") || 0;
|
||
this.firstDayOfWeekISO = rawFirstDayOfWeek === 0 ? 7 : rawFirstDayOfWeek;
|
||
|
||
let localeDaysOfWeek = [...DAYS_OF_WEEK];
|
||
const shifted = localeDaysOfWeek.splice(0, rawFirstDayOfWeek);
|
||
localeDaysOfWeek = ['', ...localeDaysOfWeek, ...shifted];
|
||
this.$weekHeader.html(localeDaysOfWeek.map((el) => `<span>${el}</span>`).join(''));
|
||
}
|
||
|
||
initWeekCalculation() {
|
||
this.weekCalculationOptions = {
|
||
firstWeekType: options.getInt("firstWeekOfYear") || 0,
|
||
minDaysInFirstWeek: options.getInt("minDaysInFirstWeek") || 4
|
||
};
|
||
}
|
||
|
||
getWeekStartDate(date: Dayjs): Dayjs {
|
||
const currentISO = date.isoWeekday();
|
||
const diff = (currentISO - this.firstDayOfWeekISO + 7) % 7;
|
||
return date.clone().subtract(diff, "day").startOf("day");
|
||
}
|
||
|
||
getWeekNumber(date: Dayjs): number {
|
||
const weekStart = this.getWeekStartDate(date);
|
||
return weekStart.isoWeek();
|
||
}
|
||
|
||
async dropdownShown() {
|
||
await this.getWeekNoteEnable();
|
||
this.weekNotes = await server.get<string[]>(`attribute-values/weekNote`);
|
||
this.init(appContext.tabManager.getActiveContextNote()?.getOwnedLabelValue("dateNote") ?? null);
|
||
}
|
||
|
||
init(activeDate: string | null) {
|
||
this.activeDate = activeDate ? dayjs(`${activeDate}T12:00:00`) : null;
|
||
this.todaysDate = dayjs();
|
||
this.date = dayjs(this.activeDate || this.todaysDate).startOf('month');
|
||
this.createMonth();
|
||
}
|
||
|
||
createDay(dateNotesForMonth: DateNotesForMonth, num: number) {
|
||
const $newDay = $("<a>")
|
||
.addClass("calendar-date")
|
||
.attr("data-calendar-date", this.date.local().format('YYYY-MM-DD'));
|
||
const $date = $("<span>").html(String(num));
|
||
const dateNoteId = dateNotesForMonth[this.date.local().format('YYYY-MM-DD')];
|
||
|
||
if (dateNoteId) {
|
||
$newDay.addClass("calendar-date-exists").attr("data-href", `#root/${dateNoteId}`);
|
||
}
|
||
|
||
if (this.date.isSame(this.activeDate, 'day')) $newDay.addClass("calendar-date-active");
|
||
if (this.date.isSame(this.todaysDate, 'day')) $newDay.addClass("calendar-date-today");
|
||
|
||
$newDay.append($date);
|
||
return $newDay;
|
||
}
|
||
|
||
createWeekNumber(weekNumber: number) {
|
||
const weekNoteId = this.date.local().format('YYYY-') + 'W' + String(weekNumber).padStart(2, '0');
|
||
let $newWeekNumber;
|
||
|
||
if (this.weekNoteEnable) {
|
||
$newWeekNumber = $("<a>").addClass("calendar-date");
|
||
if (this.weekNotes.includes(weekNoteId)) {
|
||
$newWeekNumber.addClass("calendar-date-exists").attr("data-href", `#root/${weekNoteId}`);
|
||
}
|
||
} else {
|
||
$newWeekNumber = $("<span>").addClass("calendar-week-number-disabled");
|
||
}
|
||
|
||
$newWeekNumber.addClass("calendar-week-number").attr("data-calendar-week-number", weekNoteId);
|
||
$newWeekNumber.append($("<span>").html(String(weekNumber)));
|
||
return $newWeekNumber;
|
||
}
|
||
|
||
// Use isoWeekday() consistently
|
||
private getPrevMonthDays(firstDayISO: number): { weekNumber: number, dates: Dayjs[] } {
|
||
const prevMonthLastDay = this.date.subtract(1, 'month').endOf('month');
|
||
const daysToAdd = (firstDayISO - this.firstDayOfWeekISO + 7) % 7;
|
||
const dates: Dayjs[] = [];
|
||
|
||
const firstDay = this.date.startOf('month');
|
||
const weekNumber = this.getWeekNumber(firstDay);
|
||
|
||
// Get dates from previous month
|
||
for (let i = daysToAdd - 1; i >= 0; i--) {
|
||
dates.push(prevMonthLastDay.subtract(i, 'day'));
|
||
}
|
||
|
||
return { weekNumber, dates };
|
||
}
|
||
|
||
private getNextMonthDays(lastDayISO: number): Dayjs[] {
|
||
const nextMonthFirstDay = this.date.add(1, 'month').startOf('month');
|
||
const dates: Dayjs[] = [];
|
||
|
||
const lastDayOfUserWeek = ((this.firstDayOfWeekISO + 6 - 1) % 7) + 1; // ISO wrap
|
||
const daysToAdd = (lastDayOfUserWeek - lastDayISO + 7) % 7;
|
||
|
||
for (let i = 0; i < daysToAdd; i++) {
|
||
dates.push(nextMonthFirstDay.add(i, 'day'));
|
||
}
|
||
return dates;
|
||
}
|
||
|
||
async createMonth() {
|
||
const month = this.date.format('YYYY-MM');
|
||
const dateNotesForMonth: DateNotesForMonth = await server.get(`special-notes/notes-for-month/${month}`);
|
||
|
||
this.$month.empty();
|
||
|
||
const firstDay = this.date.startOf('month');
|
||
const firstDayISO = firstDay.isoWeekday();
|
||
|
||
// Previous month filler
|
||
if (firstDayISO !== this.firstDayOfWeekISO) {
|
||
const { weekNumber, dates } = this.getPrevMonthDays(firstDayISO);
|
||
const prevMonth = this.date.subtract(1, 'month').format('YYYY-MM');
|
||
const dateNotesForPrevMonth: DateNotesForMonth = await server.get(`special-notes/notes-for-month/${prevMonth}`);
|
||
|
||
const $weekNumber = this.createWeekNumber(weekNumber);
|
||
this.$month.append($weekNumber);
|
||
|
||
dates.forEach(date => {
|
||
const tempDate = this.date;
|
||
this.date = date;
|
||
const $day = this.createDay(dateNotesForPrevMonth, date.date());
|
||
$day.addClass('calendar-date-prev-month');
|
||
this.$month.append($day);
|
||
this.date = tempDate;
|
||
});
|
||
}
|
||
|
||
const currentMonth = this.date.month();
|
||
|
||
// Main month
|
||
while (this.date.month() === currentMonth) {
|
||
const weekNumber = this.getWeekNumber(this.date);
|
||
if (this.date.isoWeekday() === this.firstDayOfWeekISO) {
|
||
const $weekNumber = this.createWeekNumber(weekNumber);
|
||
this.$month.append($weekNumber);
|
||
}
|
||
|
||
const $day = this.createDay(dateNotesForMonth, this.date.date());
|
||
this.$month.append($day);
|
||
this.date = this.date.add(1, 'day');
|
||
}
|
||
// while loop trips over and day is at 30/31, bring it back
|
||
this.date = this.date.startOf('month').subtract(1, 'month');
|
||
|
||
// Add dates from next month
|
||
const lastDayOfMonth = this.date.endOf('month');
|
||
const lastDayISO = lastDayOfMonth.isoWeekday();
|
||
const lastDayOfUserWeek = ((this.firstDayOfWeekISO + 6 - 1) % 7) + 1;
|
||
|
||
if (lastDayISO !== lastDayOfUserWeek) {
|
||
const dates = this.getNextMonthDays(lastDayISO);
|
||
const nextMonth = this.date.add(1, 'month').format('YYYY-MM');
|
||
const dateNotesForNextMonth: DateNotesForMonth = await server.get(`special-notes/notes-for-month/${nextMonth}`);
|
||
|
||
dates.forEach(date => {
|
||
const tempDate = this.date;
|
||
this.date = date;
|
||
const $day = this.createDay(dateNotesForNextMonth, date.date());
|
||
$day.addClass('calendar-date-next-month');
|
||
this.$month.append($day);
|
||
this.date = tempDate;
|
||
});
|
||
}
|
||
|
||
this.$monthSelect.text(MONTHS[this.date.month()]);
|
||
this.$yearSelect.val(this.date.year());
|
||
}
|
||
|
||
async entitiesReloadedEvent({ loadResults }: EventData<"entitiesReloaded">) {
|
||
const WEEK_OPTIONS: (keyof OptionDefinitions)[] = [
|
||
"firstDayOfWeek",
|
||
"firstWeekOfYear",
|
||
"minDaysInFirstWeek",
|
||
];
|
||
if (!WEEK_OPTIONS.some(opt => loadResults.getOptionNames().includes(opt))) {
|
||
return;
|
||
}
|
||
|
||
this.manageFirstDayOfWeek();
|
||
this.initWeekCalculation();
|
||
this.createMonth();
|
||
}
|
||
}
|