nexusweb/plugins.go at main · tkdeng/nexusweb · GitHub

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
99
100
101
102
103
104
105
package nxweb

import (
	"strconv"

	lorem "github.com/drhodes/golorem"
	"github.com/tkdeng/nexusweb/plugins"
)

func init() {
	plugins.New("lorem", func(args map[string]string, cont []byte, static bool) ([]byte, error) {
		t := byte('p')
		min := 3
		max := 5

		if val, ok := args["0"]; ok {
			t = val[0]
		}

		if val, ok := args["1"]; ok {
			if s, e := strconv.Atoi(val); e == nil && s > 0 {
				min = s
			}
		}

		if val, ok := args["2"]; ok {
			if s, e := strconv.Atoi(val); e == nil && s > 0 {
				max = s
			}
		} else if _, ok := args["1"]; ok {
			max = min
		}

		if min > max {
			min, max = max, min
		}

		switch t {
		case 'p':
			return []byte(lorem.Paragraph(min, max)), nil
		case 's':
			return []byte(lorem.Sentence(min, max)), nil
		case 'w':
			return []byte(lorem.Word(min, max)), nil
		case 'e':
			return []byte(lorem.Email()), nil
		case 'h':
			return []byte(lorem.Host()), nil
		case 'u':
			return []byte(lorem.Url()), nil
		default:
			return []byte(lorem.Paragraph(min, max)), nil
		}
	}, true)

	plugins.New("lorem-d", func(args map[string]string, cont []byte, static bool) ([]byte, error) {
		t := byte('p')
		min := 3
		max := 5

		if val, ok := args["0"]; ok {
			t = val[0]
		}

		if val, ok := args["1"]; ok {
			if s, e := strconv.Atoi(val); e == nil && s > 0 {
				min = s
			}
		}

		if val, ok := args["2"]; ok {
			if s, e := strconv.Atoi(val); e == nil && s > 0 {
				max = s
			}
		} else if _, ok := args["1"]; ok {
			max = min
		}

		if min > max {
			min, max = max, min
		}

		switch t {
		case 'p':
			return []byte(lorem.Paragraph(min, max)), nil
		case 's':
			return []byte(lorem.Sentence(min, max)), nil
		case 'w':
			return []byte(lorem.Word(min, max)), nil
		case 'e':
			return []byte(lorem.Email()), nil
		case 'h':
			return []byte(lorem.Host()), nil
		case 'u':
			return []byte(lorem.Url()), nil
		default:
			return []byte(lorem.Paragraph(min, max)), nil
		}
	})

	plugins.New("embed", func(args map[string]string, cont []byte, static bool) ([]byte, error) {
		//todo: smart embed youtube videos
		return []byte(""), nil
	}, true)
}