mirror of
				https://github.com/go-gitea/gitea.git
				synced 2025-11-04 15:04:00 +01:00 
			
		
		
		
	Change all license headers to comply with REUSE specification. Fix #16132 Co-authored-by: flynnnnnnnnnn <flynnnnnnnnnn@github> Co-authored-by: John Olheiser <john.olheiser@gmail.com>
		
			
				
	
	
		
			108 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			108 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright 2022 The Gitea Authors. All rights reserved.
 | 
						|
// SPDX-License-Identifier: MIT
 | 
						|
 | 
						|
package math
 | 
						|
 | 
						|
import (
 | 
						|
	"github.com/yuin/goldmark"
 | 
						|
	"github.com/yuin/goldmark/parser"
 | 
						|
	"github.com/yuin/goldmark/renderer"
 | 
						|
	"github.com/yuin/goldmark/util"
 | 
						|
)
 | 
						|
 | 
						|
// Extension is a math extension
 | 
						|
type Extension struct {
 | 
						|
	enabled           bool
 | 
						|
	parseDollarInline bool
 | 
						|
	parseDollarBlock  bool
 | 
						|
}
 | 
						|
 | 
						|
// Option is the interface Options should implement
 | 
						|
type Option interface {
 | 
						|
	SetOption(e *Extension)
 | 
						|
}
 | 
						|
 | 
						|
type extensionFunc func(e *Extension)
 | 
						|
 | 
						|
func (fn extensionFunc) SetOption(e *Extension) {
 | 
						|
	fn(e)
 | 
						|
}
 | 
						|
 | 
						|
// Enabled enables or disables this extension
 | 
						|
func Enabled(enable ...bool) Option {
 | 
						|
	value := true
 | 
						|
	if len(enable) > 0 {
 | 
						|
		value = enable[0]
 | 
						|
	}
 | 
						|
	return extensionFunc(func(e *Extension) {
 | 
						|
		e.enabled = value
 | 
						|
	})
 | 
						|
}
 | 
						|
 | 
						|
// WithInlineDollarParser enables or disables the parsing of $...$
 | 
						|
func WithInlineDollarParser(enable ...bool) Option {
 | 
						|
	value := true
 | 
						|
	if len(enable) > 0 {
 | 
						|
		value = enable[0]
 | 
						|
	}
 | 
						|
	return extensionFunc(func(e *Extension) {
 | 
						|
		e.parseDollarInline = value
 | 
						|
	})
 | 
						|
}
 | 
						|
 | 
						|
// WithBlockDollarParser enables or disables the parsing of $$...$$
 | 
						|
func WithBlockDollarParser(enable ...bool) Option {
 | 
						|
	value := true
 | 
						|
	if len(enable) > 0 {
 | 
						|
		value = enable[0]
 | 
						|
	}
 | 
						|
	return extensionFunc(func(e *Extension) {
 | 
						|
		e.parseDollarBlock = value
 | 
						|
	})
 | 
						|
}
 | 
						|
 | 
						|
// Math represents a math extension with default rendered delimiters
 | 
						|
var Math = &Extension{
 | 
						|
	enabled:           true,
 | 
						|
	parseDollarBlock:  true,
 | 
						|
	parseDollarInline: true,
 | 
						|
}
 | 
						|
 | 
						|
// NewExtension creates a new math extension with the provided options
 | 
						|
func NewExtension(opts ...Option) *Extension {
 | 
						|
	r := &Extension{
 | 
						|
		enabled:           true,
 | 
						|
		parseDollarBlock:  true,
 | 
						|
		parseDollarInline: true,
 | 
						|
	}
 | 
						|
 | 
						|
	for _, o := range opts {
 | 
						|
		o.SetOption(r)
 | 
						|
	}
 | 
						|
	return r
 | 
						|
}
 | 
						|
 | 
						|
// Extend extends goldmark with our parsers and renderers
 | 
						|
func (e *Extension) Extend(m goldmark.Markdown) {
 | 
						|
	if !e.enabled {
 | 
						|
		return
 | 
						|
	}
 | 
						|
 | 
						|
	m.Parser().AddOptions(parser.WithBlockParsers(
 | 
						|
		util.Prioritized(NewBlockParser(e.parseDollarBlock), 701),
 | 
						|
	))
 | 
						|
 | 
						|
	inlines := []util.PrioritizedValue{
 | 
						|
		util.Prioritized(NewInlineBracketParser(), 501),
 | 
						|
	}
 | 
						|
	if e.parseDollarInline {
 | 
						|
		inlines = append(inlines, util.Prioritized(NewInlineDollarParser(), 501))
 | 
						|
	}
 | 
						|
	m.Parser().AddOptions(parser.WithInlineParsers(inlines...))
 | 
						|
 | 
						|
	m.Renderer().AddOptions(renderer.WithNodeRenderers(
 | 
						|
		util.Prioritized(NewBlockRenderer(), 501),
 | 
						|
		util.Prioritized(NewInlineRenderer(), 502),
 | 
						|
	))
 | 
						|
}
 |