-
Notifications
You must be signed in to change notification settings - Fork 80
/
directives.go
39 lines (33 loc) · 770 Bytes
/
directives.go
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
// Copyright 2020 Frederik Zipp. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package gocyclo
import (
"go/ast"
"strings"
)
type directives []string
func (ds directives) HasIgnore() bool {
return ds.isPresent("ignore")
}
func (ds directives) isPresent(name string) bool {
for _, d := range ds {
if d == name {
return true
}
}
return false
}
func parseDirectives(doc *ast.CommentGroup) directives {
if doc == nil {
return directives{}
}
const prefix = "//gocyclo:"
var ds directives
for _, comment := range doc.List {
if strings.HasPrefix(comment.Text, prefix) {
ds = append(ds, strings.TrimSpace(strings.TrimPrefix(comment.Text, prefix)))
}
}
return ds
}