feat: pass logger to integrations (#7816)

This commit is contained in:
Emanuele Stoppa 2023-07-28 09:23:54 +01:00
parent 364d861bd5
commit 036388f66d
7 changed files with 154 additions and 30 deletions

View file

@ -0,0 +1,23 @@
---
'astro': minor
---
Integrations can now log messages using Astros built-in logger.
The logger is available to all hooks as an additional parameter:
```ts
import {AstroIntegration} from "./astro";
// integration.js
export function myIntegration(): AstroIntegration {
return {
name: "my-integration",
hooks: {
"astro:config:done": ({ logger }) => {
logger.info("Configure integration...");
}
}
}
}
```

View file

@ -22,6 +22,7 @@ import type { AstroCookies } from '../core/cookies';
import type { LogOptions, LoggerLevel } from '../core/logger/core'; import type { LogOptions, LoggerLevel } from '../core/logger/core';
import type { AstroComponentFactory, AstroComponentInstance } from '../runtime/server'; import type { AstroComponentFactory, AstroComponentInstance } from '../runtime/server';
import type { SUPPORTED_MARKDOWN_FILE_EXTENSIONS } from './../core/constants.js'; import type { SUPPORTED_MARKDOWN_FILE_EXTENSIONS } from './../core/constants.js';
import { AstroIntegrationLogger } from '../core/logger/core';
export type { export type {
MarkdownHeading, MarkdownHeading,
MarkdownMetadata, MarkdownMetadata,
@ -1879,6 +1880,7 @@ export interface AstroIntegration {
injectScript: (stage: InjectedScriptStage, content: string) => void; injectScript: (stage: InjectedScriptStage, content: string) => void;
injectRoute: (injectRoute: InjectedRoute) => void; injectRoute: (injectRoute: InjectedRoute) => void;
addClientDirective: (directive: ClientDirectiveConfig) => void; addClientDirective: (directive: ClientDirectiveConfig) => void;
logger: AstroIntegrationLogger;
// TODO: Add support for `injectElement()` for full HTML element injection, not just scripts. // TODO: Add support for `injectElement()` for full HTML element injection, not just scripts.
// This may require some refactoring of `scripts`, `styles`, and `links` into something // This may require some refactoring of `scripts`, `styles`, and `links` into something
// more generalized. Consider the SSR use-case as well. // more generalized. Consider the SSR use-case as well.
@ -1887,10 +1889,17 @@ export interface AstroIntegration {
'astro:config:done'?: (options: { 'astro:config:done'?: (options: {
config: AstroConfig; config: AstroConfig;
setAdapter: (adapter: AstroAdapter) => void; setAdapter: (adapter: AstroAdapter) => void;
logger: AstroIntegrationLogger;
}) => void | Promise<void>; }) => void | Promise<void>;
'astro:server:setup'?: (options: { server: vite.ViteDevServer }) => void | Promise<void>; 'astro:server:setup'?: (options: {
'astro:server:start'?: (options: { address: AddressInfo }) => void | Promise<void>; server: vite.ViteDevServer;
'astro:server:done'?: () => void | Promise<void>; logger: AstroIntegrationLogger;
}) => void | Promise<void>;
'astro:server:start'?: (options: {
address: AddressInfo;
logger: AstroIntegrationLogger;
}) => void | Promise<void>;
'astro:server:done'?: (options: { logger: AstroIntegrationLogger }) => void | Promise<void>;
'astro:build:ssr'?: (options: { 'astro:build:ssr'?: (options: {
manifest: SerializedSSRManifest; manifest: SerializedSSRManifest;
/** /**
@ -1902,19 +1911,25 @@ export interface AstroIntegration {
* File path of the emitted middleware * File path of the emitted middleware
*/ */
middlewareEntryPoint: URL | undefined; middlewareEntryPoint: URL | undefined;
logger: AstroIntegrationLogger;
}) => void | Promise<void>; }) => void | Promise<void>;
'astro:build:start'?: () => void | Promise<void>; 'astro:build:start'?: (options: { logger: AstroIntegrationLogger }) => void | Promise<void>;
'astro:build:setup'?: (options: { 'astro:build:setup'?: (options: {
vite: vite.InlineConfig; vite: vite.InlineConfig;
pages: Map<string, PageBuildData>; pages: Map<string, PageBuildData>;
target: 'client' | 'server'; target: 'client' | 'server';
updateConfig: (newConfig: vite.InlineConfig) => void; updateConfig: (newConfig: vite.InlineConfig) => void;
logger: AstroIntegrationLogger;
}) => void | Promise<void>;
'astro:build:generated'?: (options: {
dir: URL;
logger: AstroIntegrationLogger;
}) => void | Promise<void>; }) => void | Promise<void>;
'astro:build:generated'?: (options: { dir: URL }) => void | Promise<void>;
'astro:build:done'?: (options: { 'astro:build:done'?: (options: {
pages: { pathname: string }[]; pages: { pathname: string }[];
dir: URL; dir: URL;
routes: RouteData[]; routes: RouteData[];
logger: AstroIntegrationLogger;
}) => void | Promise<void>; }) => void | Promise<void>;
}; };
} }

View file

@ -15,7 +15,7 @@ export const consoleLogDestination = {
function getPrefix() { function getPrefix() {
let prefix = ''; let prefix = '';
let type = event.type; let type = event.label;
if (type) { if (type) {
// hide timestamp when type is undefined // hide timestamp when type is undefined
prefix += dim(dateTimeFormat.format(new Date()) + ' '); prefix += dim(dateTimeFormat.format(new Date()) + ' ');

View file

@ -6,7 +6,6 @@ interface LogWritable<T> {
} }
export type LoggerLevel = 'debug' | 'info' | 'warn' | 'error' | 'silent'; // same as Pino export type LoggerLevel = 'debug' | 'info' | 'warn' | 'error' | 'silent'; // same as Pino
export type LoggerEvent = 'info' | 'warn' | 'error';
export interface LogOptions { export interface LogOptions {
dest: LogWritable<LogMessage>; dest: LogWritable<LogMessage>;
@ -29,7 +28,7 @@ export const dateTimeFormat = new Intl.DateTimeFormat([], {
}); });
export interface LogMessage { export interface LogMessage {
type: string | null; label: string | null;
level: LoggerLevel; level: LoggerLevel;
message: string; message: string;
} }
@ -43,11 +42,11 @@ export const levels: Record<LoggerLevel, number> = {
}; };
/** Full logging API */ /** Full logging API */
export function log(opts: LogOptions, level: LoggerLevel, type: string | null, message: string) { export function log(opts: LogOptions, level: LoggerLevel, label: string | null, message: string) {
const logLevel = opts.level; const logLevel = opts.level;
const dest = opts.dest; const dest = opts.dest;
const event: LogMessage = { const event: LogMessage = {
type, label,
level, level,
message, message,
}; };
@ -61,18 +60,18 @@ export function log(opts: LogOptions, level: LoggerLevel, type: string | null, m
} }
/** Emit a user-facing message. Useful for UI and other console messages. */ /** Emit a user-facing message. Useful for UI and other console messages. */
export function info(opts: LogOptions, type: string | null, message: string) { export function info(opts: LogOptions, label: string | null, message: string) {
return log(opts, 'info', type, message); return log(opts, 'info', label, message);
} }
/** Emit a warning message. Useful for high-priority messages that aren't necessarily errors. */ /** Emit a warning message. Useful for high-priority messages that aren't necessarily errors. */
export function warn(opts: LogOptions, type: string | null, message: string) { export function warn(opts: LogOptions, label: string | null, message: string) {
return log(opts, 'warn', type, message); return log(opts, 'warn', label, message);
} }
/** Emit a error message, Useful when Astro can't recover from some error. */ /** Emit a error message, Useful when Astro can't recover from some error. */
export function error(opts: LogOptions, type: string | null, message: string) { export function error(opts: LogOptions, label: string | null, message: string) {
return log(opts, 'error', type, message); return log(opts, 'error', label, message);
} }
type LogFn = typeof info | typeof warn | typeof error; type LogFn = typeof info | typeof warn | typeof error;
@ -127,3 +126,53 @@ export function timerMessage(message: string, startTime: number = Date.now()) {
timeDiff < 750 ? `${Math.round(timeDiff)}ms` : `${(timeDiff / 1000).toFixed(1)}s`; timeDiff < 750 ? `${Math.round(timeDiff)}ms` : `${(timeDiff / 1000).toFixed(1)}s`;
return `${message} ${dim(timeDisplay)}`; return `${message} ${dim(timeDisplay)}`;
} }
export class Logger {
options: LogOptions;
constructor(options: LogOptions) {
this.options = options;
}
info(label: string, message: string) {
info(this.options, label, message);
}
warn(label: string, message: string) {
warn(this.options, label, message);
}
error(label: string, message: string) {
error(this.options, label, message);
}
debug(label: string, message: string) {
debug(this.options, label, message);
}
}
export class AstroIntegrationLogger {
options: LogOptions;
label: string;
constructor(logging: LogOptions, label: string) {
this.options = logging;
this.label = label;
}
/**
* Creates a new logger instance with a new label, but the same log options.
*/
fork(label: string): AstroIntegrationLogger {
return new AstroIntegrationLogger(this.options, label);
}
info(message: string) {
info(this.options, this.label, message);
}
warn(message: string) {
warn(this.options, this.label, message);
}
error(message: string) {
error(this.options, this.label, message);
}
debug(message: string) {
debug(this.options, this.label, message);
}
}

View file

@ -21,19 +21,19 @@ export const nodeLogDestination = new Writable({
function getPrefix() { function getPrefix() {
let prefix = ''; let prefix = '';
let type = event.type; let label = event.label;
if (type) { if (label) {
// hide timestamp when type is undefined // hide timestamp when type is undefined
prefix += dim(dateTimeFormat.format(new Date()) + ' '); prefix += dim(dateTimeFormat.format(new Date()) + ' ');
if (event.level === 'info') { if (event.level === 'info') {
type = bold(cyan(`[${type}]`)); label = bold(cyan(`[${label}]`));
} else if (event.level === 'warn') { } else if (event.level === 'warn') {
type = bold(yellow(`[${type}]`)); label = bold(yellow(`[${label}]`));
} else if (event.level === 'error') { } else if (event.level === 'error') {
type = bold(red(`[${type}]`)); label = bold(red(`[${label}]`));
} }
prefix += `${type} `; prefix += `${label} `;
} }
return reset(prefix); return reset(prefix);
} }
@ -87,7 +87,7 @@ export const nodeLogOptions: Required<LogOptions> = {
}; };
export interface LogMessage { export interface LogMessage {
type: string | null; label: string | null;
level: LoggerLevel; level: LoggerLevel;
message: string; message: string;
} }

View file

@ -5,6 +5,7 @@ import { fileURLToPath } from 'node:url';
import type { InlineConfig, ViteDevServer } from 'vite'; import type { InlineConfig, ViteDevServer } from 'vite';
import type { import type {
AstroConfig, AstroConfig,
AstroIntegration,
AstroRenderer, AstroRenderer,
AstroSettings, AstroSettings,
ContentEntryType, ContentEntryType,
@ -16,7 +17,7 @@ import type { SerializedSSRManifest } from '../core/app/types';
import type { PageBuildData } from '../core/build/types'; import type { PageBuildData } from '../core/build/types';
import { buildClientDirectiveEntrypoint } from '../core/client-directive/index.js'; import { buildClientDirectiveEntrypoint } from '../core/client-directive/index.js';
import { mergeConfig } from '../core/config/index.js'; import { mergeConfig } from '../core/config/index.js';
import { info, type LogOptions } from '../core/logger/core.js'; import { info, type LogOptions, AstroIntegrationLogger } from '../core/logger/core.js';
import { isServerLikeOutput } from '../prerender/utils.js'; import { isServerLikeOutput } from '../prerender/utils.js';
async function withTakingALongTimeMsg<T>({ async function withTakingALongTimeMsg<T>({
@ -38,6 +39,19 @@ async function withTakingALongTimeMsg<T>({
return result; return result;
} }
// Used internally to store instances of loggers.
const Loggers = new WeakMap<AstroIntegration, AstroIntegrationLogger>();
function getLogger(integration: AstroIntegration, logging: LogOptions) {
if (Loggers.has(integration)) {
// SAFETY: we check the existence in the if block
return Loggers.get(integration)!;
}
const logger = new AstroIntegrationLogger(logging, integration.name);
Loggers.set(integration, logger);
return logger;
}
export async function runHookConfigSetup({ export async function runHookConfigSetup({
settings, settings,
command, command,
@ -72,6 +86,8 @@ export async function runHookConfigSetup({
* ``` * ```
*/ */
if (integration.hooks?.['astro:config:setup']) { if (integration.hooks?.['astro:config:setup']) {
const logger = getLogger(integration, logging);
const hooks: HookParameters<'astro:config:setup'> = { const hooks: HookParameters<'astro:config:setup'> = {
config: updatedConfig, config: updatedConfig,
command, command,
@ -107,6 +123,7 @@ export async function runHookConfigSetup({
} }
addedClientDirectives.set(name, buildClientDirectiveEntrypoint(name, entrypoint)); addedClientDirectives.set(name, buildClientDirectiveEntrypoint(name, entrypoint));
}, },
logger,
}; };
// --- // ---
@ -167,6 +184,7 @@ export async function runHookConfigDone({
logging: LogOptions; logging: LogOptions;
}) { }) {
for (const integration of settings.config.integrations) { for (const integration of settings.config.integrations) {
const logger = getLogger(integration, logging);
if (integration?.hooks?.['astro:config:done']) { if (integration?.hooks?.['astro:config:done']) {
await withTakingALongTimeMsg({ await withTakingALongTimeMsg({
name: integration.name, name: integration.name,
@ -180,6 +198,7 @@ export async function runHookConfigDone({
} }
settings.adapter = adapter; settings.adapter = adapter;
}, },
logger,
}), }),
logging, logging,
}); });
@ -198,9 +217,10 @@ export async function runHookServerSetup({
}) { }) {
for (const integration of config.integrations) { for (const integration of config.integrations) {
if (integration?.hooks?.['astro:server:setup']) { if (integration?.hooks?.['astro:server:setup']) {
const logger = getLogger(integration, logging);
await withTakingALongTimeMsg({ await withTakingALongTimeMsg({
name: integration.name, name: integration.name,
hookResult: integration.hooks['astro:server:setup']({ server }), hookResult: integration.hooks['astro:server:setup']({ server, logger }),
logging, logging,
}); });
} }
@ -217,10 +237,12 @@ export async function runHookServerStart({
logging: LogOptions; logging: LogOptions;
}) { }) {
for (const integration of config.integrations) { for (const integration of config.integrations) {
const logger = getLogger(integration, logging);
if (integration?.hooks?.['astro:server:start']) { if (integration?.hooks?.['astro:server:start']) {
await withTakingALongTimeMsg({ await withTakingALongTimeMsg({
name: integration.name, name: integration.name,
hookResult: integration.hooks['astro:server:start']({ address }), hookResult: integration.hooks['astro:server:start']({ address, logger }),
logging, logging,
}); });
} }
@ -235,10 +257,12 @@ export async function runHookServerDone({
logging: LogOptions; logging: LogOptions;
}) { }) {
for (const integration of config.integrations) { for (const integration of config.integrations) {
const logger = getLogger(integration, logging);
if (integration?.hooks?.['astro:server:done']) { if (integration?.hooks?.['astro:server:done']) {
await withTakingALongTimeMsg({ await withTakingALongTimeMsg({
name: integration.name, name: integration.name,
hookResult: integration.hooks['astro:server:done'](), hookResult: integration.hooks['astro:server:done']({ logger }),
logging, logging,
}); });
} }
@ -254,9 +278,11 @@ export async function runHookBuildStart({
}) { }) {
for (const integration of config.integrations) { for (const integration of config.integrations) {
if (integration?.hooks?.['astro:build:start']) { if (integration?.hooks?.['astro:build:start']) {
const logger = getLogger(integration, logging);
await withTakingALongTimeMsg({ await withTakingALongTimeMsg({
name: integration.name, name: integration.name,
hookResult: integration.hooks['astro:build:start'](), hookResult: integration.hooks['astro:build:start']({ logger }),
logging, logging,
}); });
} }
@ -280,6 +306,8 @@ export async function runHookBuildSetup({
for (const integration of config.integrations) { for (const integration of config.integrations) {
if (integration?.hooks?.['astro:build:setup']) { if (integration?.hooks?.['astro:build:setup']) {
const logger = getLogger(integration, logging);
await withTakingALongTimeMsg({ await withTakingALongTimeMsg({
name: integration.name, name: integration.name,
hookResult: integration.hooks['astro:build:setup']({ hookResult: integration.hooks['astro:build:setup']({
@ -289,6 +317,7 @@ export async function runHookBuildSetup({
updateConfig: (newConfig) => { updateConfig: (newConfig) => {
updatedConfig = mergeConfig(updatedConfig, newConfig); updatedConfig = mergeConfig(updatedConfig, newConfig);
}, },
logger,
}), }),
logging, logging,
}); });
@ -315,12 +344,15 @@ export async function runHookBuildSsr({
}: RunHookBuildSsr) { }: RunHookBuildSsr) {
for (const integration of config.integrations) { for (const integration of config.integrations) {
if (integration?.hooks?.['astro:build:ssr']) { if (integration?.hooks?.['astro:build:ssr']) {
const logger = getLogger(integration, logging);
await withTakingALongTimeMsg({ await withTakingALongTimeMsg({
name: integration.name, name: integration.name,
hookResult: integration.hooks['astro:build:ssr']({ hookResult: integration.hooks['astro:build:ssr']({
manifest, manifest,
entryPoints, entryPoints,
middlewareEntryPoint, middlewareEntryPoint,
logger,
}), }),
logging, logging,
}); });
@ -338,10 +370,12 @@ export async function runHookBuildGenerated({
const dir = isServerLikeOutput(config) ? config.build.client : config.outDir; const dir = isServerLikeOutput(config) ? config.build.client : config.outDir;
for (const integration of config.integrations) { for (const integration of config.integrations) {
const logger = getLogger(integration, logging);
if (integration?.hooks?.['astro:build:generated']) { if (integration?.hooks?.['astro:build:generated']) {
await withTakingALongTimeMsg({ await withTakingALongTimeMsg({
name: integration.name, name: integration.name,
hookResult: integration.hooks['astro:build:generated']({ dir }), hookResult: integration.hooks['astro:build:generated']({ dir, logger }),
logging, logging,
}); });
} }
@ -361,12 +395,15 @@ export async function runHookBuildDone({ config, pages, routes, logging }: RunHo
for (const integration of config.integrations) { for (const integration of config.integrations) {
if (integration?.hooks?.['astro:build:done']) { if (integration?.hooks?.['astro:build:done']) {
const logger = getLogger(integration, logging);
await withTakingALongTimeMsg({ await withTakingALongTimeMsg({
name: integration.name, name: integration.name,
hookResult: integration.hooks['astro:build:done']({ hookResult: integration.hooks['astro:build:done']({
pages: pages.map((p) => ({ pathname: p })), pages: pages.map((p) => ({ pathname: p })),
dir, dir,
routes, routes,
logger,
}), }),
logging, logging,
}); });

View file

@ -175,7 +175,7 @@ describe('Static build', () => {
let found = false; let found = false;
for (const log of logs) { for (const log of logs) {
if ( if (
log.type === 'ssg' && log.label === 'ssg' &&
/[hH]eaders are not exposed in static \(SSG\) output mode/.test(log.message) /[hH]eaders are not exposed in static \(SSG\) output mode/.test(log.message)
) { ) {
found = true; found = true;