ec1f89217a
Full Mattermost server source with integrated Community Enterprise features. Includes vendor directory for offline/air-gapped builds. Structure: - enterprise-impl/: Enterprise feature implementations - enterprise-community/: Init files that register implementations - enterprise/: Bridge imports (community_imports.go) - vendor/: All dependencies for offline builds Build (online): go build ./cmd/mattermost Build (offline/air-gapped): go build -mod=vendor ./cmd/mattermost 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
128 lines
2.7 KiB
Go
128 lines
2.7 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package asthelpers
|
|
|
|
import (
|
|
"go/ast"
|
|
"go/types"
|
|
|
|
"github.com/pkg/errors"
|
|
"golang.org/x/tools/go/packages"
|
|
)
|
|
|
|
func GetPackage(pkgPath string) (*packages.Package, error) {
|
|
cfg := &packages.Config{
|
|
Mode: packages.NeedName | packages.NeedTypes | packages.NeedSyntax | packages.NeedTypesInfo,
|
|
}
|
|
pkgs, err := packages.Load(cfg, pkgPath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if len(pkgs) == 0 {
|
|
return nil, errors.Errorf("could not find package %s", pkgPath)
|
|
}
|
|
return pkgs[0], nil
|
|
}
|
|
|
|
func FindInterface(name string, files []*ast.File) (*ast.InterfaceType, error) {
|
|
iface, _, err := FindInterfaceWithIdent(name, files)
|
|
return iface, err
|
|
}
|
|
|
|
func FindInterfaceWithIdent(name string, files []*ast.File) (*ast.InterfaceType, *ast.Ident, error) {
|
|
var (
|
|
ident *ast.Ident
|
|
iface *ast.InterfaceType
|
|
)
|
|
|
|
for _, f := range files {
|
|
ast.Inspect(f, func(n ast.Node) bool {
|
|
if t, ok := n.(*ast.TypeSpec); ok {
|
|
if iface != nil {
|
|
return false
|
|
}
|
|
|
|
if i, ok := t.Type.(*ast.InterfaceType); ok && t.Name.Name == name {
|
|
ident = t.Name
|
|
iface = i
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
})
|
|
|
|
if iface != nil {
|
|
return iface, ident, nil
|
|
}
|
|
}
|
|
return nil, nil, errors.Errorf("could not find %s interface", name)
|
|
}
|
|
|
|
func FindMethodsCalledOnType(info *types.Info, typ types.Type, caller *ast.FuncDecl) []string {
|
|
var methods []string
|
|
|
|
ast.Inspect(caller, func(n ast.Node) bool {
|
|
if s, ok := n.(*ast.SelectorExpr); ok {
|
|
var receiver *ast.Ident
|
|
switch r := s.X.(type) {
|
|
case *ast.Ident:
|
|
// Left-hand side of the selector is an identifier, eg:
|
|
//
|
|
// a := p.API
|
|
// a.GetTeams()
|
|
//
|
|
receiver = r
|
|
case *ast.SelectorExpr:
|
|
// Left-hand side of the selector is a selector, eg:
|
|
//
|
|
// p.API.GetTeams()
|
|
//
|
|
receiver = r.Sel
|
|
}
|
|
|
|
if receiver != nil {
|
|
obj := info.ObjectOf(receiver)
|
|
if obj != nil && types.Identical(obj.Type(), typ) {
|
|
methods = append(methods, s.Sel.Name)
|
|
}
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
})
|
|
|
|
return methods
|
|
}
|
|
|
|
func FindReceiverMethods(receiverName string, files []*ast.File) []*ast.FuncDecl {
|
|
var fns []*ast.FuncDecl
|
|
for _, f := range files {
|
|
ast.Inspect(f, func(n ast.Node) bool {
|
|
if fn, ok := n.(*ast.FuncDecl); ok {
|
|
r := extractReceiverTypeName(fn)
|
|
if r == receiverName {
|
|
fns = append(fns, fn)
|
|
}
|
|
}
|
|
return true
|
|
})
|
|
}
|
|
return fns
|
|
}
|
|
|
|
func extractReceiverTypeName(fn *ast.FuncDecl) string {
|
|
if fn.Recv != nil {
|
|
t := fn.Recv.List[0].Type
|
|
// Unwrap the pointer type (a star expression)
|
|
if se, ok := t.(*ast.StarExpr); ok {
|
|
t = se.X
|
|
}
|
|
if id, ok := t.(*ast.Ident); ok {
|
|
return id.Name
|
|
}
|
|
}
|
|
return ""
|
|
}
|