feat(imsize): rework image styling to get more control (#549)

* feat(imsize): rework image styling to get more control

* chore: add more tests with different size configs, refactor styling block

* chore: hide inline size styling behind feature flag

* chore: add forcedSanitizeWhiteList

* chore: rename params

Author: St1ggy

src/transform/md.ts

@@ -168,7 +168,9 @@ function initCompiler(md: MarkdownIt, options: OptionsType, env: EnvType) {
         const html = md.renderer.render(tokens, md.options, env);
 
         // Sanitize the page
-        return needToSanitizeHtml ? sanitizeHtml(html, sanitizeOptions) : html;
+        return needToSanitizeHtml
+            ? sanitizeHtml(html, sanitizeOptions, {cssWhiteList: env.additionalOptionsCssWhiteList})
+            : html;
     };
 }
 

src/transform/plugins/imsize/const.ts

@@ -4,4 +4,5 @@ export enum ImsizeAttr {
     Title = 'title',
     Width = 'width',
     Height = 'height',
+    Style = 'style',
 }

src/transform/plugins/imsize/index.ts

@@ -1,15 +1,15 @@
-import {PluginSimple} from 'markdown-it';
+import {PluginWithOptions} from 'markdown-it';
 
-import {imageWithSize} from './plugin';
+import {ImsizeOptions, imageWithSize} from './plugin';
 
 /**
  * Imsize plugin for markdown-it.
  * This plugin overloads original image renderer.
  * Forked from https://github.com/tatsy/markdown-it-imsize
  */
 
-const imsize: PluginSimple = (md) => {
-    md.inline.ruler.before('emphasis', 'image', imageWithSize(md));
+const imsize: PluginWithOptions<ImsizeOptions> = (md, opts) => {
+    md.inline.ruler.before('emphasis', 'image', imageWithSize(md, opts));
 };
 
 export = imsize;

src/transform/plugins/imsize/plugin.ts

@@ -5,7 +5,11 @@ import type Token from 'markdown-it/lib/token';
 import {ImsizeAttr} from './const';
 import {parseImageSize} from './helpers';
 
-export const imageWithSize = (md: MarkdownIt): ParserInline.RuleInline => {
+export type ImsizeOptions = {
+    enableInlineStyling?: boolean;
+};
+
+export const imageWithSize = (md: MarkdownIt, opts?: ImsizeOptions): ParserInline.RuleInline => {
     // eslint-disable-next-line complexity
     return (state, silent) => {
         if (state.src.charCodeAt(state.pos) !== 0x21 /* ! */) {
@@ -206,6 +210,33 @@ export const imageWithSize = (md: MarkdownIt): ParserInline.RuleInline => {
             if (height !== '') {
                 token.attrs.push([ImsizeAttr.Height, height]);
             }
+
+            if (opts?.enableInlineStyling) {
+                let style: string | undefined = '';
+
+                const widthWithPercent = width.includes('%');
+                const heightWithPercent = height.includes('%');
+
+                if (width !== '') {
+                    const widthString = widthWithPercent ? width : `${width}px`;
+                    style += `width: ${widthString};`;
+                }
+
+                if (height !== '') {
+                    if (width !== '' && !heightWithPercent && !widthWithPercent) {
+                        style += `aspect-ratio: ${width} / ${height};height: auto;`;
+                        state.env.additionalOptionsCssWhiteList ??= {};
+                        state.env.additionalOptionsCssWhiteList['aspect-ratio'] = true;
+                    } else {
+                        const heightString = heightWithPercent ? height : `${height}px`;
+                        style += `height: ${heightString};`;
+                    }
+                }
+
+                if (style) {
+                    token.attrs.push([ImsizeAttr.Style, style]);
+                }
+            }
         }
 
         state.pos = pos;

src/transform/sanitize.ts

@@ -4,6 +4,8 @@ import cssfilter from 'cssfilter';
 import * as cheerio from 'cheerio';
 import css from 'css';
 
+import {CssWhiteList} from './typings';
+
 const htmlTags = [
     'a',
     'abbr',
@@ -492,8 +494,6 @@ const allowedTags = Array.from(
 );
 const allowedAttributes = Array.from(new Set([...htmlAttrs, ...svgAttrs, ...yfmHtmlAttrs]));
 
-export type CssWhiteList = {[property: string]: boolean};
-
 export interface SanitizeOptions extends sanitizeHtml.IOptions {
     cssWhiteList?: CssWhiteList;
     disableStyleSanitizer?: boolean;
@@ -598,9 +598,20 @@ function sanitizeStyles(html: string, options: SanitizeOptions) {
     return styles + content;
 }
 
-export default function sanitize(html: string, options?: SanitizeOptions) {
+export default function sanitize(
+    html: string,
+    options?: SanitizeOptions,
+    additionalOptions?: SanitizeOptions,
+) {
     const sanitizeOptions = options || defaultOptions;
 
+    if (additionalOptions?.cssWhiteList) {
+        sanitizeOptions.cssWhiteList = {
+            ...sanitizeOptions.cssWhiteList,
+            ...additionalOptions.cssWhiteList,
+        };
+    }
+
     const needToSanitizeStyles = !(sanitizeOptions.disableStyleSanitizer ?? false);
 
     const modifiedHtml = needToSanitizeStyles ? sanitizeStyles(html, sanitizeOptions) : html;

src/transform/typings.ts

@@ -74,6 +74,7 @@ export type EnvType<Extras extends {} = {}> = {
     assets?: unknown[];
     meta?: object;
     changelogs?: ChangelogItem[];
+    additionalOptionsCssWhiteList?: CssWhiteList;
 } & Extras;
 
 export interface MarkdownItPluginOpts {
@@ -98,3 +99,5 @@ export type MarkdownItPluginCb<T extends {} = {}> = {
 export type MarkdownItPreprocessorCb<T extends unknown = {}> = {
     (input: string, opts: T & Partial<MarkdownItPluginOpts>, md?: MarkdownIt): string;
 };
+
+export type CssWhiteList = {[property: string]: boolean};

test/data/imsize/imsize-fixtures.txt

@@ -283,6 +283,26 @@ Coverage. Image
 .
 <p><img src="test" alt="test" width="100%"></p>
 .
+.
+![test](test =x100%)
+.
+<p><img src="test" alt="test" height="100%"></p>
+.
+.
+![test](test =100%x100%)
+.
+<p><img src="test" alt="test" width="100%" height="100%"></p>
+.
+.
+![test](test =100%x200)
+.
+<p><img src="test" alt="test" width="100%" height="200"></p>
+.
+.
+![test](test =100x100%)
+.
+<p><img src="test" alt="test" width="100" height="100%"></p>
+.
 
 Coverage. Link
 .

test/data/imsize/imsize-inlineSizeStyling-fixtures.txt

@@ -0,0 +1,81 @@
+Coverage. Image with inlineStyling
+.
+![test]( x =100x200)
+.
+<p><img src="x" alt="test" width="100" height="200" style="width: 100px;aspect-ratio: 100 / 200;height: auto;"></p>
+.
+.
+![test]( x =x)
+.
+<p><img src="x" alt="test"></p>
+.
+.
+![test]( x =100x)
+.
+<p><img src="x" alt="test" width="100" style="width: 100px;"></p>
+.
+.
+![test]( x =x200)
+.
+<p><img src="x" alt="test" height="200" style="height: 200px;"></p>
+.
+.
+![test]( x "title" =100x200)
+.
+<p><img src="x" alt="test" title="title" width="100" height="200" style="width: 100px;aspect-ratio: 100 / 200;height: auto;"></p>
+.
+.
+![test]( x =WxH )
+.
+<p>![test]( x =WxH )</p>
+.
+.
+![test]( x = 100x200 )
+.
+<p>![test]( x = 100x200 )</p>
+.
+.
+![test]( x =aaaxbbb )
+.
+<p>![test]( x =aaaxbbb )</p>
+.
+.
+![test](http://this.is.test.jpg =100x200)
+.
+<p><img src="http://this.is.test.jpg" alt="test" width="100" height="200" style="width: 100px;aspect-ratio: 100 / 200;height: auto;"></p>
+.
+.
+![test](<x =100x200)
+.
+<p>![test](&lt;x =100x200)</p>
+.
+.
+![test](<x> =100x200)
+.
+<p><img src="x" alt="test" width="100" height="200" style="width: 100px;aspect-ratio: 100 / 200;height: auto;"></p>
+.
+.
+![test](test =100%x)
+.
+<p><img src="test" alt="test" width="100%" style="width: 100%;"></p>
+.
+.
+![test](test =x100%)
+.
+<p><img src="test" alt="test" height="100%" style="height: 100%;"></p>
+.
+.
+![test](test =100%x100%)
+.
+<p><img src="test" alt="test" width="100%" height="100%" style="width: 100%;height: 100%;"></p>
+.
+.
+![test](test =100%x200)
+.
+<p><img src="test" alt="test" width="100%" height="200" style="width: 100%;height: 200px;"></p>
+.
+.
+![test](test =100x100%)
+.
+<p><img src="test" alt="test" width="100" height="100%" style="width: 100px;height: 100%;"></p>
+.

test/imsize.test.ts

@@ -14,3 +14,13 @@ describe('imsize', () => {
 
     generate(path.join(__dirname, 'data/imsize/imsize-fixtures.txt'), md);
 });
+
+describe('imsize with inlineStyling', () => {
+    const md = new MarkdownIt({
+        html: true,
+        linkify: false,
+        typographer: false,
+    }).use(imsize, {enableInlineStyling: true});
+
+    generate(path.join(__dirname, 'data/imsize/imsize-inlineSizeStyling-fixtures.txt'), md);
+});