summaryrefslogtreecommitdiffstats
path: root/utils/markdown/indented_code.go
blob: dc5dce1accf3223eaee92ee2453734bd6a4c0304 (plain)
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
// Copyright (c) 2017-present Mattermost, Inc. All Rights Reserved.
// See License.txt for license information.

package markdown

import (
	"strings"
)

type IndentedCodeLine struct {
	Indentation int
	Range       Range
}

type IndentedCode struct {
	blockBase
	markdown string

	RawCode []IndentedCodeLine
}

func (b *IndentedCode) Code() (result string) {
	for _, code := range b.RawCode {
		result += strings.Repeat(" ", code.Indentation) + b.markdown[code.Range.Position:code.Range.End]
	}
	return
}

func (b *IndentedCode) Continuation(indentation int, r Range) *continuation {
	if indentation >= 4 {
		return &continuation{
			Indentation: indentation - 4,
			Remaining:   r,
		}
	}
	s := b.markdown[r.Position:r.End]
	if strings.TrimSpace(s) == "" {
		return &continuation{
			Remaining: r,
		}
	}
	return nil
}

func (b *IndentedCode) AddLine(indentation int, r Range) bool {
	b.RawCode = append(b.RawCode, IndentedCodeLine{
		Indentation: indentation,
		Range:       r,
	})
	return true
}

func (b *IndentedCode) Close() {
	for {
		last := b.RawCode[len(b.RawCode)-1]
		s := b.markdown[last.Range.Position:last.Range.End]
		if strings.TrimRight(s, "\r\n") == "" {
			b.RawCode = b.RawCode[:len(b.RawCode)-1]
		} else {
			break
		}
	}
}

func (b *IndentedCode) AllowsBlockStarts() bool {
	return false
}

func indentedCodeStart(markdown string, indentation int, r Range, matchedBlocks, unmatchedBlocks []Block) []Block {
	if len(unmatchedBlocks) > 0 {
		if _, ok := unmatchedBlocks[len(unmatchedBlocks)-1].(*Paragraph); ok {
			return nil
		}
	} else if len(matchedBlocks) > 0 {
		if _, ok := matchedBlocks[len(matchedBlocks)-1].(*Paragraph); ok {
			return nil
		}
	}

	if indentation < 4 {
		return nil
	}

	s := markdown[r.Position:r.End]
	if strings.TrimSpace(s) == "" {
		return nil
	}

	return []Block{
		&IndentedCode{
			markdown: markdown,
			RawCode: []IndentedCodeLine{{
				Indentation: indentation - 4,
				Range:       r,
			}},
		},
	}
}