-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* AST parser, placeholder renderer * Basic renderer * Implement proper renderer * Linter warnings * Better name for content parser struct
- Loading branch information
Showing
8 changed files
with
214 additions
and
20 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,3 +1,5 @@ | ||
| +++ | ||
| categories = ["Contributors"] | ||
| +++ | ||
|
|
||
| Discord user: <@106162668032802816> |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,167 @@ | ||
| package extensions | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "log/slog" | ||
|
|
||
| "github.com/bwmarrin/discordgo" | ||
| "github.com/yuin/goldmark" | ||
| "github.com/yuin/goldmark/ast" | ||
| "github.com/yuin/goldmark/parser" | ||
| "github.com/yuin/goldmark/renderer" | ||
| "github.com/yuin/goldmark/text" | ||
| "github.com/yuin/goldmark/util" | ||
| ) | ||
|
|
||
| var DiscordMentionKind = ast.NewNodeKind("DiscordMention") | ||
|
|
||
| type DiscordMentionNode struct { | ||
| ast.BaseInline | ||
|
|
||
| ID string | ||
| } | ||
|
|
||
| func (m *DiscordMentionNode) Dump(source []byte, level int) { | ||
| ast.DumpHelper( | ||
| m, | ||
| source, | ||
| level, | ||
| map[string]string{ | ||
| "ID": m.ID, | ||
| }, | ||
| nil, | ||
| ) | ||
| } | ||
|
|
||
| func (m *DiscordMentionNode) Kind() ast.NodeKind { | ||
| return DiscordMentionKind | ||
| } | ||
|
|
||
| var _ ast.Node = (*DiscordMentionNode)(nil) | ||
|
|
||
| type discordMentionParser struct{} | ||
|
|
||
| func (d discordMentionParser) Trigger() []byte { | ||
| return []byte("<") | ||
| } | ||
|
|
||
| func (d discordMentionParser) Parse(parent ast.Node, block text.Reader, pc parser.Context) ast.Node { | ||
| line, _ := block.PeekLine() | ||
|
|
||
| lineHead := 0 | ||
| userId := "" | ||
| inMention := false | ||
|
|
||
| for ; lineHead < len(line); lineHead++ { | ||
| char := line[lineHead] | ||
|
|
||
| if char == '@' { | ||
| inMention = true | ||
| } else if inMention && char == '>' { | ||
| inMention = false | ||
| break | ||
| } else if inMention { | ||
| userId += string(char) | ||
| } | ||
| } | ||
|
|
||
| if inMention || userId == "" { | ||
| return nil | ||
| } | ||
|
|
||
| block.Advance(lineHead + 1) | ||
|
|
||
| return &DiscordMentionNode{ | ||
| ID: userId, | ||
| } | ||
| } | ||
|
|
||
| var _ parser.InlineParser = (*discordMentionParser)(nil) | ||
|
|
||
| type discordMentionRenderer struct { | ||
| resolver *DiscordUserResolver | ||
| } | ||
|
|
||
| func (r *discordMentionRenderer) render( | ||
| w util.BufWriter, source []byte, n ast.Node, entering bool, | ||
| ) (ast.WalkStatus, error) { | ||
| if !entering { | ||
| return ast.WalkContinue, nil | ||
| } | ||
|
|
||
| username := fmt.Sprintf("<@%s>", n.(*DiscordMentionNode).ID) | ||
|
|
||
| if r.resolver != nil { | ||
| username = r.resolver.Resolve(n.(*DiscordMentionNode).ID) | ||
| } | ||
|
|
||
| _, err := w.WriteString(username) | ||
| if err != nil { | ||
| return ast.WalkStop, fmt.Errorf("failed to write string: %w", err) | ||
| } | ||
|
|
||
| return ast.WalkContinue, nil | ||
| } | ||
|
|
||
| func (r *discordMentionRenderer) RegisterFuncs(reg renderer.NodeRendererFuncRegisterer) { | ||
| reg.Register(DiscordMentionKind, r.render) | ||
| } | ||
|
|
||
| var _ renderer.NodeRenderer = (*discordMentionRenderer)(nil) | ||
|
|
||
| type DiscordMention struct { | ||
| resolver *DiscordUserResolver | ||
| } | ||
|
|
||
| func (d *DiscordMention) Extend(m goldmark.Markdown) { | ||
| m.Renderer(). | ||
| AddOptions(renderer.WithNodeRenderers(util.Prioritized(&discordMentionRenderer{resolver: d.resolver}, 500))) | ||
| m.Parser(). | ||
| AddOptions(parser.WithInlineParsers(util.Prioritized(&discordMentionParser{}, 500))) | ||
| } | ||
|
|
||
| type DiscordUserResolver struct { | ||
| cache map[string]string | ||
| discordClient *discordgo.Session | ||
| } | ||
|
|
||
| func (r *DiscordUserResolver) Resolve(userId string) string { | ||
| if cachedVal, ok := r.cache[userId]; ok { | ||
| return cachedVal | ||
| } | ||
|
|
||
| slog.Info("Resolving user for the first time, subsequent mentions will be cached...", slog.String("user_id", userId)) | ||
|
|
||
| var username string | ||
| user, err := r.discordClient.User(userId) | ||
| if err != nil { | ||
| username = fmt.Sprintf("<@%s>", userId) | ||
| slog.Error("Failed to resolve user", "error", err) | ||
| } else { | ||
| username = user.Username | ||
| } | ||
|
|
||
| r.cache[userId] = username | ||
|
|
||
| return username | ||
| } | ||
|
|
||
| func NewDiscordUserResolver(botToken string) (*DiscordUserResolver, error) { | ||
| if botToken == "" { | ||
| return nil, fmt.Errorf("bot token cannot be empty") | ||
| } | ||
|
|
||
| discordClient, err := discordgo.New("Bot " + botToken) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("failed to create Discord client: %w", err) | ||
| } | ||
|
|
||
| return &DiscordUserResolver{ | ||
| cache: make(map[string]string), | ||
| discordClient: discordClient, | ||
| }, nil | ||
| } | ||
|
|
||
| func NewDiscordMention(resolver *DiscordUserResolver) goldmark.Extender { | ||
| return &DiscordMention{resolver: resolver} | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.