mirror of https://github.com/go-gitea/gitea
Rename scripts to build and add revive command as a new build tool command (#10942)
Co-authored-by: techknowlogick <techknowlogick@gitea.io>pull/10872/head^2
Lunny Xiao
5 years ago
committed by
GitHub
parent
4af7c47b38
commit
4f63f283c4
@ -0,0 +1,325 @@ |
||||
// Copyright 2020 The Gitea Authors. All rights reserved.
|
||||
// Copyright (c) 2018 Minko Gechev. All rights reserved.
|
||||
// Use of this source code is governed by a MIT-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// +build ignore
|
||||
|
||||
package main |
||||
|
||||
import ( |
||||
"flag" |
||||
"fmt" |
||||
"io/ioutil" |
||||
"os" |
||||
"path/filepath" |
||||
"strings" |
||||
|
||||
"github.com/BurntSushi/toml" |
||||
"github.com/mgechev/dots" |
||||
"github.com/mgechev/revive/formatter" |
||||
"github.com/mgechev/revive/lint" |
||||
"github.com/mgechev/revive/rule" |
||||
"github.com/mitchellh/go-homedir" |
||||
) |
||||
|
||||
func fail(err string) { |
||||
fmt.Fprintln(os.Stderr, err) |
||||
os.Exit(1) |
||||
} |
||||
|
||||
var defaultRules = []lint.Rule{ |
||||
&rule.VarDeclarationsRule{}, |
||||
&rule.PackageCommentsRule{}, |
||||
&rule.DotImportsRule{}, |
||||
&rule.BlankImportsRule{}, |
||||
&rule.ExportedRule{}, |
||||
&rule.VarNamingRule{}, |
||||
&rule.IndentErrorFlowRule{}, |
||||
&rule.IfReturnRule{}, |
||||
&rule.RangeRule{}, |
||||
&rule.ErrorfRule{}, |
||||
&rule.ErrorNamingRule{}, |
||||
&rule.ErrorStringsRule{}, |
||||
&rule.ReceiverNamingRule{}, |
||||
&rule.IncrementDecrementRule{}, |
||||
&rule.ErrorReturnRule{}, |
||||
&rule.UnexportedReturnRule{}, |
||||
&rule.TimeNamingRule{}, |
||||
&rule.ContextKeysType{}, |
||||
&rule.ContextAsArgumentRule{}, |
||||
} |
||||
|
||||
var allRules = append([]lint.Rule{ |
||||
&rule.ArgumentsLimitRule{}, |
||||
&rule.CyclomaticRule{}, |
||||
&rule.FileHeaderRule{}, |
||||
&rule.EmptyBlockRule{}, |
||||
&rule.SuperfluousElseRule{}, |
||||
&rule.ConfusingNamingRule{}, |
||||
&rule.GetReturnRule{}, |
||||
&rule.ModifiesParamRule{}, |
||||
&rule.ConfusingResultsRule{}, |
||||
&rule.DeepExitRule{}, |
||||
&rule.UnusedParamRule{}, |
||||
&rule.UnreachableCodeRule{}, |
||||
&rule.AddConstantRule{}, |
||||
&rule.FlagParamRule{}, |
||||
&rule.UnnecessaryStmtRule{}, |
||||
&rule.StructTagRule{}, |
||||
&rule.ModifiesValRecRule{}, |
||||
&rule.ConstantLogicalExprRule{}, |
||||
&rule.BoolLiteralRule{}, |
||||
&rule.RedefinesBuiltinIDRule{}, |
||||
&rule.ImportsBlacklistRule{}, |
||||
&rule.FunctionResultsLimitRule{}, |
||||
&rule.MaxPublicStructsRule{}, |
||||
&rule.RangeValInClosureRule{}, |
||||
&rule.RangeValAddress{}, |
||||
&rule.WaitGroupByValueRule{}, |
||||
&rule.AtomicRule{}, |
||||
&rule.EmptyLinesRule{}, |
||||
&rule.LineLengthLimitRule{}, |
||||
&rule.CallToGCRule{}, |
||||
&rule.DuplicatedImportsRule{}, |
||||
&rule.ImportShadowingRule{}, |
||||
&rule.BareReturnRule{}, |
||||
&rule.UnusedReceiverRule{}, |
||||
&rule.UnhandledErrorRule{}, |
||||
&rule.CognitiveComplexityRule{}, |
||||
&rule.StringOfIntRule{}, |
||||
}, defaultRules...) |
||||
|
||||
var allFormatters = []lint.Formatter{ |
||||
&formatter.Stylish{}, |
||||
&formatter.Friendly{}, |
||||
&formatter.JSON{}, |
||||
&formatter.NDJSON{}, |
||||
&formatter.Default{}, |
||||
&formatter.Unix{}, |
||||
&formatter.Checkstyle{}, |
||||
&formatter.Plain{}, |
||||
} |
||||
|
||||
func getFormatters() map[string]lint.Formatter { |
||||
result := map[string]lint.Formatter{} |
||||
for _, f := range allFormatters { |
||||
result[f.Name()] = f |
||||
} |
||||
return result |
||||
} |
||||
|
||||
func getLintingRules(config *lint.Config) []lint.Rule { |
||||
rulesMap := map[string]lint.Rule{} |
||||
for _, r := range allRules { |
||||
rulesMap[r.Name()] = r |
||||
} |
||||
|
||||
lintingRules := []lint.Rule{} |
||||
for name := range config.Rules { |
||||
rule, ok := rulesMap[name] |
||||
if !ok { |
||||
fail("cannot find rule: " + name) |
||||
} |
||||
lintingRules = append(lintingRules, rule) |
||||
} |
||||
|
||||
return lintingRules |
||||
} |
||||
|
||||
func parseConfig(path string) *lint.Config { |
||||
config := &lint.Config{} |
||||
file, err := ioutil.ReadFile(path) |
||||
if err != nil { |
||||
fail("cannot read the config file") |
||||
} |
||||
_, err = toml.Decode(string(file), config) |
||||
if err != nil { |
||||
fail("cannot parse the config file: " + err.Error()) |
||||
} |
||||
return config |
||||
} |
||||
|
||||
func normalizeConfig(config *lint.Config) { |
||||
if config.Confidence == 0 { |
||||
config.Confidence = 0.8 |
||||
} |
||||
severity := config.Severity |
||||
if severity != "" { |
||||
for k, v := range config.Rules { |
||||
if v.Severity == "" { |
||||
v.Severity = severity |
||||
} |
||||
config.Rules[k] = v |
||||
} |
||||
for k, v := range config.Directives { |
||||
if v.Severity == "" { |
||||
v.Severity = severity |
||||
} |
||||
config.Directives[k] = v |
||||
} |
||||
} |
||||
} |
||||
|
||||
func getConfig() *lint.Config { |
||||
config := defaultConfig() |
||||
if configPath != "" { |
||||
config = parseConfig(configPath) |
||||
} |
||||
normalizeConfig(config) |
||||
return config |
||||
} |
||||
|
||||
func getFormatter() lint.Formatter { |
||||
formatters := getFormatters() |
||||
formatter := formatters["default"] |
||||
if formatterName != "" { |
||||
f, ok := formatters[formatterName] |
||||
if !ok { |
||||
fail("unknown formatter " + formatterName) |
||||
} |
||||
formatter = f |
||||
} |
||||
return formatter |
||||
} |
||||
|
||||
func buildDefaultConfigPath() string { |
||||
var result string |
||||
if homeDir, err := homedir.Dir(); err == nil { |
||||
result = filepath.Join(homeDir, "revive.toml") |
||||
if _, err := os.Stat(result); err != nil { |
||||
result = "" |
||||
} |
||||
} |
||||
|
||||
return result |
||||
} |
||||
|
||||
func defaultConfig() *lint.Config { |
||||
defaultConfig := lint.Config{ |
||||
Confidence: 0.0, |
||||
Severity: lint.SeverityWarning, |
||||
Rules: map[string]lint.RuleConfig{}, |
||||
} |
||||
for _, r := range defaultRules { |
||||
defaultConfig.Rules[r.Name()] = lint.RuleConfig{} |
||||
} |
||||
return &defaultConfig |
||||
} |
||||
|
||||
func normalizeSplit(strs []string) []string { |
||||
res := []string{} |
||||
for _, s := range strs { |
||||
t := strings.Trim(s, " \t") |
||||
if len(t) > 0 { |
||||
res = append(res, t) |
||||
} |
||||
} |
||||
return res |
||||
} |
||||
|
||||
func getPackages() [][]string { |
||||
globs := normalizeSplit(flag.Args()) |
||||
if len(globs) == 0 { |
||||
globs = append(globs, ".") |
||||
} |
||||
|
||||
packages, err := dots.ResolvePackages(globs, normalizeSplit(excludePaths)) |
||||
if err != nil { |
||||
fail(err.Error()) |
||||
} |
||||
|
||||
return packages |
||||
} |
||||
|
||||
type arrayFlags []string |
||||
|
||||
func (i *arrayFlags) String() string { |
||||
return strings.Join([]string(*i), " ") |
||||
} |
||||
|
||||
func (i *arrayFlags) Set(value string) error { |
||||
*i = append(*i, value) |
||||
return nil |
||||
} |
||||
|
||||
var configPath string |
||||
var excludePaths arrayFlags |
||||
var formatterName string |
||||
var help bool |
||||
|
||||
var originalUsage = flag.Usage |
||||
|
||||
func init() { |
||||
flag.Usage = func() { |
||||
originalUsage() |
||||
} |
||||
// command line help strings
|
||||
const ( |
||||
configUsage = "path to the configuration TOML file, defaults to $HOME/revive.toml, if present (i.e. -config myconf.toml)" |
||||
excludeUsage = "list of globs which specify files to be excluded (i.e. -exclude foo/...)" |
||||
formatterUsage = "formatter to be used for the output (i.e. -formatter stylish)" |
||||
) |
||||
|
||||
defaultConfigPath := buildDefaultConfigPath() |
||||
|
||||
flag.StringVar(&configPath, "config", defaultConfigPath, configUsage) |
||||
flag.Var(&excludePaths, "exclude", excludeUsage) |
||||
flag.StringVar(&formatterName, "formatter", "", formatterUsage) |
||||
flag.Parse() |
||||
} |
||||
|
||||
func main() { |
||||
config := getConfig() |
||||
formatter := getFormatter() |
||||
packages := getPackages() |
||||
|
||||
revive := lint.New(func(file string) ([]byte, error) { |
||||
return ioutil.ReadFile(file) |
||||
}) |
||||
|
||||
lintingRules := getLintingRules(config) |
||||
|
||||
failures, err := revive.Lint(packages, lintingRules, *config) |
||||
if err != nil { |
||||
fail(err.Error()) |
||||
} |
||||
|
||||
formatChan := make(chan lint.Failure) |
||||
exitChan := make(chan bool) |
||||
|
||||
var output string |
||||
go (func() { |
||||
output, err = formatter.Format(formatChan, *config) |
||||
if err != nil { |
||||
fail(err.Error()) |
||||
} |
||||
exitChan <- true |
||||
})() |
||||
|
||||
exitCode := 0 |
||||
for f := range failures { |
||||
if f.Confidence < config.Confidence { |
||||
continue |
||||
} |
||||
if exitCode == 0 { |
||||
exitCode = config.WarningCode |
||||
} |
||||
if c, ok := config.Rules[f.RuleName]; ok && c.Severity == lint.SeverityError { |
||||
exitCode = config.ErrorCode |
||||
} |
||||
if c, ok := config.Directives[f.RuleName]; ok && c.Severity == lint.SeverityError { |
||||
exitCode = config.ErrorCode |
||||
} |
||||
|
||||
formatChan <- f |
||||
} |
||||
|
||||
close(formatChan) |
||||
<-exitChan |
||||
if output != "" { |
||||
fmt.Println(output) |
||||
} |
||||
|
||||
os.Exit(exitCode) |
||||
} |
||||
@ -0,0 +1,18 @@ |
||||
// Copyright 2020 The Gitea Authors. All rights reserved.
|
||||
// Use of this source code is governed by a MIT-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package build |
||||
|
||||
import ( |
||||
// for lint
|
||||
_ "github.com/BurntSushi/toml" |
||||
_ "github.com/mgechev/dots" |
||||
_ "github.com/mgechev/revive/formatter" |
||||
_ "github.com/mgechev/revive/lint" |
||||
_ "github.com/mgechev/revive/rule" |
||||
_ "github.com/mitchellh/go-homedir" |
||||
|
||||
// for embed
|
||||
_ "github.com/shurcooL/vfsgen" |
||||
) |
||||
@ -0,0 +1,20 @@ |
||||
The MIT License (MIT) |
||||
|
||||
Copyright (c) 2013 Fatih Arslan |
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy of |
||||
this software and associated documentation files (the "Software"), to deal in |
||||
the Software without restriction, including without limitation the rights to |
||||
use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of |
||||
the Software, and to permit persons to whom the Software is furnished to do so, |
||||
subject to the following conditions: |
||||
|
||||
The above copyright notice and this permission notice shall be included in all |
||||
copies or substantial portions of the Software. |
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS |
||||
FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR |
||||
COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER |
||||
IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN |
||||
CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. |
||||
@ -0,0 +1,182 @@ |
||||
# Archived project. No maintenance. |
||||
|
||||
This project is not maintained anymore and is archived. Feel free to fork and |
||||
make your own changes if needed. For more detail read my blog post: [Taking an indefinite sabbatical from my projects](https://arslan.io/2018/10/09/taking-an-indefinite-sabbatical-from-my-projects/) |
||||
|
||||
Thanks to everyone for their valuable feedback and contributions. |
||||
|
||||
|
||||
# Color [](https://godoc.org/github.com/fatih/color) |
||||
|
||||
Color lets you use colorized outputs in terms of [ANSI Escape |
||||
Codes](http://en.wikipedia.org/wiki/ANSI_escape_code#Colors) in Go (Golang). It |
||||
has support for Windows too! The API can be used in several ways, pick one that |
||||
suits you. |
||||
|
||||
|
||||
 |
||||
|
||||
|
||||
## Install |
||||
|
||||
```bash |
||||
go get github.com/fatih/color |
||||
``` |
||||
|
||||
## Examples |
||||
|
||||
### Standard colors |
||||
|
||||
```go |
||||
// Print with default helper functions |
||||
color.Cyan("Prints text in cyan.") |
||||
|
||||
// A newline will be appended automatically |
||||
color.Blue("Prints %s in blue.", "text") |
||||
|
||||
// These are using the default foreground colors |
||||
color.Red("We have red") |
||||
color.Magenta("And many others ..") |
||||
|
||||
``` |
||||
|
||||
### Mix and reuse colors |
||||
|
||||
```go |
||||
// Create a new color object |
||||
c := color.New(color.FgCyan).Add(color.Underline) |
||||
c.Println("Prints cyan text with an underline.") |
||||
|
||||
// Or just add them to New() |
||||
d := color.New(color.FgCyan, color.Bold) |
||||
d.Printf("This prints bold cyan %s\n", "too!.") |
||||
|
||||
// Mix up foreground and background colors, create new mixes! |
||||
red := color.New(color.FgRed) |
||||
|
||||
boldRed := red.Add(color.Bold) |
||||
boldRed.Println("This will print text in bold red.") |
||||
|
||||
whiteBackground := red.Add(color.BgWhite) |
||||
whiteBackground.Println("Red text with white background.") |
||||
``` |
||||
|
||||
### Use your own output (io.Writer) |
||||
|
||||
```go |
||||
// Use your own io.Writer output |
||||
color.New(color.FgBlue).Fprintln(myWriter, "blue color!") |
||||
|
||||
blue := color.New(color.FgBlue) |
||||
blue.Fprint(writer, "This will print text in blue.") |
||||
``` |
||||
|
||||
### Custom print functions (PrintFunc) |
||||
|
||||
```go |
||||
// Create a custom print function for convenience |
||||
red := color.New(color.FgRed).PrintfFunc() |
||||
red("Warning") |
||||
red("Error: %s", err) |
||||
|
||||
// Mix up multiple attributes |
||||
notice := color.New(color.Bold, color.FgGreen).PrintlnFunc() |
||||
notice("Don't forget this...") |
||||
``` |
||||
|
||||
### Custom fprint functions (FprintFunc) |
||||
|
||||
```go |
||||
blue := color.New(FgBlue).FprintfFunc() |
||||
blue(myWriter, "important notice: %s", stars) |
||||
|
||||
// Mix up with multiple attributes |
||||
success := color.New(color.Bold, color.FgGreen).FprintlnFunc() |
||||
success(myWriter, "Don't forget this...") |
||||
``` |
||||
|
||||
### Insert into noncolor strings (SprintFunc) |
||||
|
||||
```go |
||||
// Create SprintXxx functions to mix strings with other non-colorized strings: |
||||
yellow := color.New(color.FgYellow).SprintFunc() |
||||
red := color.New(color.FgRed).SprintFunc() |
||||
fmt.Printf("This is a %s and this is %s.\n", yellow("warning"), red("error")) |
||||
|
||||
info := color.New(color.FgWhite, color.BgGreen).SprintFunc() |
||||
fmt.Printf("This %s rocks!\n", info("package")) |
||||
|
||||
// Use helper functions |
||||
fmt.Println("This", color.RedString("warning"), "should be not neglected.") |
||||
fmt.Printf("%v %v\n", color.GreenString("Info:"), "an important message.") |
||||
|
||||
// Windows supported too! Just don't forget to change the output to color.Output |
||||
fmt.Fprintf(color.Output, "Windows support: %s", color.GreenString("PASS")) |
||||
``` |
||||
|
||||
### Plug into existing code |
||||
|
||||
```go |
||||
// Use handy standard colors |
||||
color.Set(color.FgYellow) |
||||
|
||||
fmt.Println("Existing text will now be in yellow") |
||||
fmt.Printf("This one %s\n", "too") |
||||
|
||||
color.Unset() // Don't forget to unset |
||||
|
||||
// You can mix up parameters |
||||
color.Set(color.FgMagenta, color.Bold) |
||||
defer color.Unset() // Use it in your function |
||||
|
||||
fmt.Println("All text will now be bold magenta.") |
||||
``` |
||||
|
||||
### Disable/Enable color |
||||
|
||||
There might be a case where you want to explicitly disable/enable color output. the |
||||
`go-isatty` package will automatically disable color output for non-tty output streams |
||||
(for example if the output were piped directly to `less`) |
||||
|
||||
`Color` has support to disable/enable colors both globally and for single color |
||||
definitions. For example suppose you have a CLI app and a `--no-color` bool flag. You |
||||
can easily disable the color output with: |
||||
|
||||
```go |
||||
|
||||
var flagNoColor = flag.Bool("no-color", false, "Disable color output") |
||||
|
||||
if *flagNoColor { |
||||
color.NoColor = true // disables colorized output |
||||
} |
||||
``` |
||||
|
||||
It also has support for single color definitions (local). You can |
||||
disable/enable color output on the fly: |
||||
|
||||
```go |
||||
c := color.New(color.FgCyan) |
||||
c.Println("Prints cyan text") |
||||
|
||||
c.DisableColor() |
||||
c.Println("This is printed without any color") |
||||
|
||||
c.EnableColor() |
||||
c.Println("This prints again cyan...") |
||||
``` |
||||
|
||||
## Todo |
||||
|
||||
* Save/Return previous values |
||||
* Evaluate fmt.Formatter interface |
||||
|
||||
|
||||
## Credits |
||||
|
||||
* [Fatih Arslan](https://github.com/fatih) |
||||
* Windows support via @mattn: [colorable](https://github.com/mattn/go-colorable) |
||||
|
||||
## License |
||||
|
||||
The MIT License (MIT) - see [`LICENSE.md`](https://github.com/fatih/color/blob/master/LICENSE.md) for more details |
||||
|
||||
@ -0,0 +1,603 @@ |
||||
package color |
||||
|
||||
import ( |
||||
"fmt" |
||||
"io" |
||||
"os" |
||||
"strconv" |
||||
"strings" |
||||
"sync" |
||||
|
||||
"github.com/mattn/go-colorable" |
||||
"github.com/mattn/go-isatty" |
||||
) |
||||
|
||||
var ( |
||||
// NoColor defines if the output is colorized or not. It's dynamically set to
|
||||
// false or true based on the stdout's file descriptor referring to a terminal
|
||||
// or not. This is a global option and affects all colors. For more control
|
||||
// over each color block use the methods DisableColor() individually.
|
||||
NoColor = os.Getenv("TERM") == "dumb" || |
||||
(!isatty.IsTerminal(os.Stdout.Fd()) && !isatty.IsCygwinTerminal(os.Stdout.Fd())) |
||||
|
||||
// Output defines the standard output of the print functions. By default
|
||||
// os.Stdout is used.
|
||||
Output = colorable.NewColorableStdout() |
||||
|
||||
// Error defines a color supporting writer for os.Stderr.
|
||||
Error = colorable.NewColorableStderr() |
||||
|
||||
// colorsCache is used to reduce the count of created Color objects and
|
||||
// allows to reuse already created objects with required Attribute.
|
||||
colorsCache = make(map[Attribute]*Color) |
||||
colorsCacheMu sync.Mutex // protects colorsCache
|
||||
) |
||||
|
||||
// Color defines a custom color object which is defined by SGR parameters.
|
||||
type Color struct { |
||||
params []Attribute |
||||
noColor *bool |
||||
} |
||||
|
||||
// Attribute defines a single SGR Code
|
||||
type Attribute int |
||||
|
||||
const escape = "\x1b" |
||||
|
||||
// Base attributes
|
||||
const ( |
||||
Reset Attribute = iota |
||||
Bold |
||||
Faint |
||||
Italic |
||||
Underline |
||||
BlinkSlow |
||||
BlinkRapid |
||||
ReverseVideo |
||||
Concealed |
||||
CrossedOut |
||||
) |
||||
|
||||
// Foreground text colors
|
||||
const ( |
||||
FgBlack Attribute = iota + 30 |
||||
FgRed |
||||
FgGreen |
||||
FgYellow |
||||
FgBlue |
||||
FgMagenta |
||||
FgCyan |
||||
FgWhite |
||||
) |
||||
|
||||
// Foreground Hi-Intensity text colors
|
||||
const ( |
||||
FgHiBlack Attribute = iota + 90 |
||||
FgHiRed |
||||
FgHiGreen |
||||
FgHiYellow |
||||
FgHiBlue |
||||
FgHiMagenta |
||||
FgHiCyan |
||||
FgHiWhite |
||||
) |
||||
|
||||
// Background text colors
|
||||
const ( |
||||
BgBlack Attribute = iota + 40 |
||||
BgRed |
||||
BgGreen |
||||
BgYellow |
||||
BgBlue |
||||
BgMagenta |
||||
BgCyan |
||||
BgWhite |
||||
) |
||||
|
||||
// Background Hi-Intensity text colors
|
||||
const ( |
||||
BgHiBlack Attribute = iota + 100 |
||||
BgHiRed |
||||
BgHiGreen |
||||
BgHiYellow |
||||
BgHiBlue |
||||
BgHiMagenta |
||||
BgHiCyan |
||||
BgHiWhite |
||||
) |
||||
|
||||
// New returns a newly created color object.
|
||||
func New(value ...Attribute) *Color { |
||||
c := &Color{params: make([]Attribute, 0)} |
||||
c.Add(value...) |
||||
return c |
||||
} |
||||
|
||||
// Set sets the given parameters immediately. It will change the color of
|
||||
// output with the given SGR parameters until color.Unset() is called.
|
||||
func Set(p ...Attribute) *Color { |
||||
c := New(p...) |
||||
c.Set() |
||||
return c |
||||
} |
||||
|
||||
// Unset resets all escape attributes and clears the output. Usually should
|
||||
// be called after Set().
|
||||
func Unset() { |
||||
if NoColor { |
||||
return |
||||
} |
||||
|
||||
fmt.Fprintf(Output, "%s[%dm", escape, Reset) |
||||
} |
||||
|
||||
// Set sets the SGR sequence.
|
||||
func (c *Color) Set() *Color { |
||||
if c.isNoColorSet() { |
||||
return c |
||||
} |
||||
|
||||
fmt.Fprintf(Output, c.format()) |
||||
return c |
||||
} |
||||
|
||||
func (c *Color) unset() { |
||||
if c.isNoColorSet() { |
||||
return |
||||
} |
||||
|
||||
Unset() |
||||
} |
||||
|
||||
func (c *Color) setWriter(w io.Writer) *Color { |
||||
if c.isNoColorSet() { |
||||
return c |
||||
} |
||||
|
||||
fmt.Fprintf(w, c.format()) |
||||
return c |
||||
} |
||||
|
||||
func (c *Color) unsetWriter(w io.Writer) { |
||||
if c.isNoColorSet() { |
||||
return |
||||
} |
||||
|
||||
if NoColor { |
||||
return |
||||
} |
||||
|
||||
fmt.Fprintf(w, "%s[%dm", escape, Reset) |
||||
} |
||||
|
||||
// Add is used to chain SGR parameters. Use as many as parameters to combine
|
||||
// and create custom color objects. Example: Add(color.FgRed, color.Underline).
|
||||
func (c *Color) Add(value ...Attribute) *Color { |
||||
c.params = append(c.params, value...) |
||||
return c |
||||
} |
||||
|
||||
func (c *Color) prepend(value Attribute) { |
||||
c.params = append(c.params, 0) |
||||
copy(c.params[1:], c.params[0:]) |
||||
c.params[0] = value |
||||
} |
||||
|
||||
// Fprint formats using the default formats for its operands and writes to w.
|
||||
// Spaces are added between operands when neither is a string.
|
||||
// It returns the number of bytes written and any write error encountered.
|
||||
// On Windows, users should wrap w with colorable.NewColorable() if w is of
|
||||
// type *os.File.
|
||||
func (c *Color) Fprint(w io.Writer, a ...interface{}) (n int, err error) { |
||||
c.setWriter(w) |
||||
defer c.unsetWriter(w) |
||||
|
||||
return fmt.Fprint(w, a...) |
||||
} |
||||
|
||||
// Print formats using the default formats for its operands and writes to
|
||||
// standard output. Spaces are added between operands when neither is a
|
||||
// string. It returns the number of bytes written and any write error
|
||||
// encountered. This is the standard fmt.Print() method wrapped with the given
|
||||
// color.
|
||||
func (c *Color) Print(a ...interface{}) (n int, err error) { |
||||
c.Set() |
||||
defer c.unset() |
||||
|
||||
return fmt.Fprint(Output, a...) |
||||
} |
||||
|
||||
// Fprintf formats according to a format specifier and writes to w.
|
||||
// It returns the number of bytes written and any write error encountered.
|
||||
// On Windows, users should wrap w with colorable.NewColorable() if w is of
|
||||
// type *os.File.
|
||||
func (c *Color) Fprintf(w io.Writer, format string, a ...interface{}) (n int, err error) { |
||||
c.setWriter(w) |
||||
defer c.unsetWriter(w) |
||||
|
||||
return fmt.Fprintf(w, format, a...) |
||||
} |
||||
|
||||
// Printf formats according to a format specifier and writes to standard output.
|
||||
// It returns the number of bytes written and any write error encountered.
|
||||
// This is the standard fmt.Printf() method wrapped with the given color.
|
||||
func (c *Color) Printf(format string, a ...interface{}) (n int, err error) { |
||||
c.Set() |
||||
defer c.unset() |
||||
|
||||
return fmt.Fprintf(Output, format, a...) |
||||
} |
||||
|
||||
// Fprintln formats using the default formats for its operands and writes to w.
|
||||
// Spaces are always added between operands and a newline is appended.
|
||||
// On Windows, users should wrap w with colorable.NewColorable() if w is of
|
||||
// type *os.File.
|
||||
func (c *Color) Fprintln(w io.Writer, a ...interface{}) (n int, err error) { |
||||
c.setWriter(w) |
||||
defer c.unsetWriter(w) |
||||
|
||||
return fmt.Fprintln(w, a...) |
||||
} |
||||
|
||||
// Println formats using the default formats for its operands and writes to
|
||||
// standard output. Spaces are always added between operands and a newline is
|
||||
// appended. It returns the number of bytes written and any write error
|
||||
// encountered. This is the standard fmt.Print() method wrapped with the given
|
||||
// color.
|
||||
func (c *Color) Println(a ...interface{}) (n int, err error) { |
||||
c.Set() |
||||
defer c.unset() |
||||
|
||||
return fmt.Fprintln(Output, a...) |
||||
} |
||||
|
||||
// Sprint is just like Print, but returns a string instead of printing it.
|
||||
func (c *Color) Sprint(a ...interface{}) string { |
||||
return c.wrap(fmt.Sprint(a...)) |
||||
} |
||||
|
||||
// Sprintln is just like Println, but returns a string instead of printing it.
|
||||
func (c *Color) Sprintln(a ...interface{}) string { |
||||
return c.wrap(fmt.Sprintln(a...)) |
||||
} |
||||
|
||||
// Sprintf is just like Printf, but returns a string instead of printing it.
|
||||
func (c *Color) Sprintf(format string, a ...interface{}) string { |
||||
return c.wrap(fmt.Sprintf(format, a...)) |
||||
} |
||||
|
||||
// FprintFunc returns a new function that prints the passed arguments as
|
||||
// colorized with color.Fprint().
|
||||
func (c *Color) FprintFunc() func(w io.Writer, a ...interface{}) { |
||||
return func(w io.Writer, a ...interface{}) { |
||||
c.Fprint(w, a...) |
||||
} |
||||
} |
||||
|
||||
// PrintFunc returns a new function that prints the passed arguments as
|
||||
// colorized with color.Print().
|
||||
func (c *Color) PrintFunc() func(a ...interface{}) { |
||||
return func(a ...interface{}) { |
||||
c.Print(a...) |
||||
} |
||||
} |
||||
|
||||
// FprintfFunc returns a new function that prints the passed arguments as
|
||||
// colorized with color.Fprintf().
|
||||
func (c *Color) FprintfFunc() func(w io.Writer, format string, a ...interface{}) { |
||||
return func(w io.Writer, format string, a ...interface{}) { |
||||
c.Fprintf(w, format, a...) |
||||
} |
||||
} |
||||
|
||||
// PrintfFunc returns a new function that prints the passed arguments as
|
||||
// colorized with color.Printf().
|
||||
func (c *Color) PrintfFunc() func(format string, a ...interface{}) { |
||||
return func(format string, a ...interface{}) { |
||||
c.Printf(format, a...) |
||||
} |
||||
} |
||||
|
||||
// FprintlnFunc returns a new function that prints the passed arguments as
|
||||
// colorized with color.Fprintln().
|
||||
func (c *Color) FprintlnFunc() func(w io.Writer, a ...interface{}) { |
||||
return func(w io.Writer, a ...interface{}) { |
||||
c.Fprintln(w, a...) |
||||
} |
||||
} |
||||
|
||||
// PrintlnFunc returns a new function that prints the passed arguments as
|
||||
// colorized with color.Println().
|
||||
func (c *Color) PrintlnFunc() func(a ...interface{}) { |
||||
return func(a ...interface{}) { |
||||
c.Println(a...) |
||||
} |
||||
} |
||||
|
||||
// SprintFunc returns a new function that returns colorized strings for the
|
||||
// given arguments with fmt.Sprint(). Useful to put into or mix into other
|
||||
// string. Windows users should use this in conjunction with color.Output, example:
|
||||
//
|
||||
// put := New(FgYellow).SprintFunc()
|
||||
// fmt.Fprintf(color.Output, "This is a %s", put("warning"))
|
||||
func (c *Color) SprintFunc() func(a ...interface{}) string { |
||||
return func(a ...interface{}) string { |
||||
return c.wrap(fmt.Sprint(a...)) |
||||
} |
||||
} |
||||
|
||||
// SprintfFunc returns a new function that returns colorized strings for the
|
||||
// given arguments with fmt.Sprintf(). Useful to put into or mix into other
|
||||
// string. Windows users should use this in conjunction with color.Output.
|
||||
func (c *Color) SprintfFunc() func(format string, a ...interface{}) string { |
||||
return func(format string, a ...interface{}) string { |
||||
return c.wrap(fmt.Sprintf(format, a...)) |
||||
} |
||||
} |
||||
|
||||
// SprintlnFunc returns a new function that returns colorized strings for the
|
||||
// given arguments with fmt.Sprintln(). Useful to put into or mix into other
|
||||
// string. Windows users should use this in conjunction with color.Output.
|
||||
func (c *Color) SprintlnFunc() func(a ...interface{}) string { |
||||
return func(a ...interface{}) string { |
||||
return c.wrap(fmt.Sprintln(a...)) |
||||
} |
||||
} |
||||
|
||||
// sequence returns a formatted SGR sequence to be plugged into a "\x1b[...m"
|
||||
// an example output might be: "1;36" -> bold cyan
|
||||
func (c *Color) sequence() string { |
||||
format := make([]string, len(c.params)) |
||||
for i, v := range c.params { |
||||
format[i] = strconv.Itoa(int(v)) |
||||
} |
||||
|
||||
return strings.Join(format, ";") |
||||
} |
||||
|
||||
// wrap wraps the s string with the colors attributes. The string is ready to
|
||||
// be printed.
|
||||
func (c *Color) wrap(s string) string { |
||||
if c.isNoColorSet() { |
||||
return s |
||||
} |
||||
|
||||
return c.format() + s + c.unformat() |
||||
} |
||||
|
||||
func (c *Color) format() string { |
||||
return fmt.Sprintf("%s[%sm", escape, c.sequence()) |
||||
} |
||||
|
||||
func (c *Color) unformat() string { |
||||
return fmt.Sprintf("%s[%dm", escape, Reset) |
||||
} |
||||
|
||||
// DisableColor disables the color output. Useful to not change any existing
|
||||
// code and still being able to output. Can be used for flags like
|
||||
// "--no-color". To enable back use EnableColor() method.
|
||||
func (c *Color) DisableColor() { |
||||
c.noColor = boolPtr(true) |
||||
} |
||||
|
||||
// EnableColor enables the color output. Use it in conjunction with
|
||||
// DisableColor(). Otherwise this method has no side effects.
|
||||
func (c *Color) EnableColor() { |
||||
c.noColor = boolPtr(false) |
||||
} |
||||
|
||||
func (c *Color) isNoColorSet() bool { |
||||
// check first if we have user setted action
|
||||
if c.noColor != nil { |
||||
return *c.noColor |
||||
} |
||||
|
||||
// if not return the global option, which is disabled by default
|
||||
return NoColor |
||||
} |
||||
|
||||
// Equals returns a boolean value indicating whether two colors are equal.
|
||||
func (c *Color) Equals(c2 *Color) bool { |
||||
if len(c.params) != len(c2.params) { |
||||
return false |
||||
} |
||||
|
||||
for _, attr := range c.params { |
||||
if !c2.attrExists(attr) { |
||||
return false |
||||
} |
||||
} |
||||
|
||||
return true |
||||
} |
||||
|
||||
func (c *Color) attrExists(a Attribute) bool { |
||||
for _, attr := range c.params { |
||||
if attr == a { |
||||
return true |
||||
} |
||||
} |
||||
|
||||
return false |
||||
} |
||||
|
||||
func boolPtr(v bool) *bool { |
||||
return &v |
||||
} |
||||
|
||||
func getCachedColor(p Attribute) *Color { |
||||
colorsCacheMu.Lock() |
||||
defer colorsCacheMu.Unlock() |
||||
|
||||
c, ok := colorsCache[p] |
||||
if !ok { |
||||
c = New(p) |
||||
colorsCache[p] = c |
||||
} |
||||
|
||||
return c |
||||
} |
||||
|
||||
func colorPrint(format string, p Attribute, a ...interface{}) { |
||||
c := getCachedColor(p) |
||||
|
||||
if !strings.HasSuffix(format, "\n") { |
||||
format += "\n" |
||||
} |
||||
|
||||
if len(a) == 0 { |
||||
c.Print(format) |
||||
} else { |
||||
c.Printf(format, a...) |
||||
} |
||||
} |
||||
|
||||
func colorString(format string, p Attribute, a ...interface{}) string { |
||||
c := getCachedColor(p) |
||||
|
||||
if len(a) == 0 { |
||||
return c.SprintFunc()(format) |
||||
} |
||||
|
||||
return c.SprintfFunc()(format, a...) |
||||
} |
||||
|
||||
// Black is a convenient helper function to print with black foreground. A
|
||||
// newline is appended to format by default.
|
||||
func Black(format string, a ...interface{}) { colorPrint(format, FgBlack, a...) } |
||||
|
||||
// Red is a convenient helper function to print with red foreground. A
|
||||
// newline is appended to format by default.
|
||||
func Red(format string, a ...interface{}) { colorPrint(format, FgRed, a...) } |
||||
|
||||
// Green is a convenient helper function to print with green foreground. A
|
||||
// newline is appended to format by default.
|
||||
func Green(format string, a ...interface{}) { colorPrint(format, FgGreen, a...) } |
||||
|
||||
// Yellow is a convenient helper function to print with yellow foreground.
|
||||
// A newline is appended to format by default.
|
||||
func Yellow(format string, a ...interface{}) { colorPrint(format, FgYellow, a...) } |
||||
|
||||
// Blue is a convenient helper function to print with blue foreground. A
|
||||
// newline is appended to format by default.
|
||||
func Blue(format string, a ...interface{}) { colorPrint(format, FgBlue, a...) } |
||||
|
||||
// Magenta is a convenient helper function to print with magenta foreground.
|
||||
// A newline is appended to format by default.
|
||||
func Magenta(format string, a ...interface{}) { colorPrint(format, FgMagenta, a...) } |
||||
|
||||
// Cyan is a convenient helper function to print with cyan foreground. A
|
||||
// newline is appended to format by default.
|
||||
func Cyan(format string, a ...interface{}) { colorPrint(format, FgCyan, a...) } |
||||
|
||||
// White is a convenient helper function to print with white foreground. A
|
||||
// newline is appended to format by default.
|
||||
func White(format string, a ...interface{}) { colorPrint(format, FgWhite, a...) } |
||||
|
||||
// BlackString is a convenient helper function to return a string with black
|
||||
// foreground.
|
||||
func BlackString(format string, a ...interface{}) string { return colorString(format, FgBlack, a...) } |
||||
|
||||
// RedString is a convenient helper function to return a string with red
|
||||
// foreground.
|
||||
func RedString(format string, a ...interface{}) string { return colorString(format, FgRed, a...) } |
||||
|
||||
// GreenString is a convenient helper function to return a string with green
|
||||
// foreground.
|
||||
func GreenString(format string, a ...interface{}) string { return colorString(format, FgGreen, a...) } |
||||
|
||||
// YellowString is a convenient helper function to return a string with yellow
|
||||
// foreground.
|
||||
func YellowString(format string, a ...interface{}) string { return colorString(format, FgYellow, a...) } |
||||
|
||||
// BlueString is a convenient helper function to return a string with blue
|
||||
// foreground.
|
||||
func BlueString(format string, a ...interface{}) string { return colorString(format, FgBlue, a...) } |
||||
|
||||
// MagentaString is a convenient helper function to return a string with magenta
|
||||
// foreground.
|
||||
func MagentaString(format string, a ...interface{}) string { |
||||
return colorString(format, FgMagenta, a...) |
||||
} |
||||
|
||||
// CyanString is a convenient helper function to return a string with cyan
|
||||
// foreground.
|
||||
func CyanString(format string, a ...interface{}) string { return colorString(format, FgCyan, a...) } |
||||
|
||||
// WhiteString is a convenient helper function to return a string with white
|
||||
// foreground.
|
||||
func WhiteString(format string, a ...interface{}) string { return colorString(format, FgWhite, a...) } |
||||
|
||||
// HiBlack is a convenient helper function to print with hi-intensity black foreground. A
|
||||
// newline is appended to format by default.
|
||||
func HiBlack(format string, a ...interface{}) { colorPrint(format, FgHiBlack, a...) } |
||||
|
||||
// HiRed is a convenient helper function to print with hi-intensity red foreground. A
|
||||
// newline is appended to format by default.
|
||||
func HiRed(format string, a ...interface{}) { colorPrint(format, FgHiRed, a...) } |
||||
|
||||
// HiGreen is a convenient helper function to print with hi-intensity green foreground. A
|
||||
// newline is appended to format by default.
|
||||
func HiGreen(format string, a ...interface{}) { colorPrint(format, FgHiGreen, a...) } |
||||
|
||||
// HiYellow is a convenient helper function to print with hi-intensity yellow foreground.
|
||||
// A newline is appended to format by default.
|
||||
func HiYellow(format string, a ...interface{}) { colorPrint(format, FgHiYellow, a...) } |
||||
|
||||
// HiBlue is a convenient helper function to print with hi-intensity blue foreground. A
|
||||
// newline is appended to format by default.
|
||||
func HiBlue(format string, a ...interface{}) { colorPrint(format, FgHiBlue, a...) } |
||||
|
||||
// HiMagenta is a convenient helper function to print with hi-intensity magenta foreground.
|
||||
// A newline is appended to format by default.
|
||||
func HiMagenta(format string, a ...interface{}) { colorPrint(format, FgHiMagenta, a...) } |
||||
|
||||
// HiCyan is a convenient helper function to print with hi-intensity cyan foreground. A
|
||||
// newline is appended to format by default.
|
||||
func HiCyan(format string, a ...interface{}) { colorPrint(format, FgHiCyan, a...) } |
||||
|
||||
// HiWhite is a convenient helper function to print with hi-intensity white foreground. A
|
||||
// newline is appended to format by default.
|
||||
func HiWhite(format string, a ...interface{}) { colorPrint(format, FgHiWhite, a...) } |
||||
|
||||
// HiBlackString is a convenient helper function to return a string with hi-intensity black
|
||||
// foreground.
|
||||
func HiBlackString(format string, a ...interface{}) string { |
||||
return colorString(format, FgHiBlack, a...) |
||||
} |
||||
|
||||
// HiRedString is a convenient helper function to return a string with hi-intensity red
|
||||
// foreground.
|
||||
func HiRedString(format string, a ...interface{}) string { return colorString(format, FgHiRed, a...) } |
||||
|
||||
// HiGreenString is a convenient helper function to return a string with hi-intensity green
|
||||
// foreground.
|
||||
func HiGreenString(format string, a ...interface{}) string { |
||||
return colorString(format, FgHiGreen, a...) |
||||
} |
||||
|
||||
// HiYellowString is a convenient helper function to return a string with hi-intensity yellow
|
||||
// foreground.
|
||||
func HiYellowString(format string, a ...interface{}) string { |
||||
return colorString(format, FgHiYellow, a...) |
||||
} |
||||
|
||||
// HiBlueString is a convenient helper function to return a string with hi-intensity blue
|
||||
// foreground.
|
||||
func HiBlueString(format string, a ...interface{}) string { return colorString(format, FgHiBlue, a...) } |
||||
|
||||
// HiMagentaString is a convenient helper function to return a string with hi-intensity magenta
|
||||
// foreground.
|
||||
func HiMagentaString(format string, a ...interface{}) string { |
||||
return colorString(format, FgHiMagenta, a...) |
||||
} |
||||
|
||||
// HiCyanString is a convenient helper function to return a string with hi-intensity cyan
|
||||
// foreground.
|
||||
func HiCyanString(format string, a ...interface{}) string { return colorString(format, FgHiCyan, a...) } |
||||
|
||||
// HiWhiteString is a convenient helper function to return a string with hi-intensity white
|
||||
// foreground.
|
||||
func HiWhiteString(format string, a ...interface{}) string { |
||||
return colorString(format, FgHiWhite, a...) |
||||
} |
||||
@ -0,0 +1,133 @@ |
||||
/* |
||||
Package color is an ANSI color package to output colorized or SGR defined |
||||
output to the standard output. The API can be used in several way, pick one |
||||
that suits you. |
||||
|
||||
Use simple and default helper functions with predefined foreground colors: |
||||
|
||||
color.Cyan("Prints text in cyan.") |
||||
|
||||
// a newline will be appended automatically
|
||||
color.Blue("Prints %s in blue.", "text") |
||||
|
||||
// More default foreground colors..
|
||||
color.Red("We have red") |
||||
color.Yellow("Yellow color too!") |
||||
color.Magenta("And many others ..") |
||||
|
||||
// Hi-intensity colors
|
||||
color.HiGreen("Bright green color.") |
||||
color.HiBlack("Bright black means gray..") |
||||
color.HiWhite("Shiny white color!") |
||||
|
||||
However there are times where custom color mixes are required. Below are some |
||||
examples to create custom color objects and use the print functions of each |
||||
separate color object. |
||||
|
||||
// Create a new color object
|
||||
c := color.New(color.FgCyan).Add(color.Underline) |
||||
c.Println("Prints cyan text with an underline.") |
||||
|
||||
// Or just add them to New()
|
||||
d := color.New(color.FgCyan, color.Bold) |
||||
d.Printf("This prints bold cyan %s\n", "too!.") |
||||
|
||||
|
||||
// Mix up foreground and background colors, create new mixes!
|
||||
red := color.New(color.FgRed) |
||||
|
||||
boldRed := red.Add(color.Bold) |
||||
boldRed.Println("This will print text in bold red.") |
||||
|
||||
whiteBackground := red.Add(color.BgWhite) |
||||
whiteBackground.Println("Red text with White background.") |
||||
|
||||
// Use your own io.Writer output
|
||||
color.New(color.FgBlue).Fprintln(myWriter, "blue color!") |
||||
|
||||
blue := color.New(color.FgBlue) |
||||
blue.Fprint(myWriter, "This will print text in blue.") |
||||
|
||||
You can create PrintXxx functions to simplify even more: |
||||
|
||||
// Create a custom print function for convenient
|
||||
red := color.New(color.FgRed).PrintfFunc() |
||||
red("warning") |
||||
red("error: %s", err) |
||||
|
||||
// Mix up multiple attributes
|
||||
notice := color.New(color.Bold, color.FgGreen).PrintlnFunc() |
||||
notice("don't forget this...") |
||||
|
||||
You can also FprintXxx functions to pass your own io.Writer: |
||||
|
||||
blue := color.New(FgBlue).FprintfFunc() |
||||
blue(myWriter, "important notice: %s", stars) |
||||
|
||||
// Mix up with multiple attributes
|
||||
success := color.New(color.Bold, color.FgGreen).FprintlnFunc() |
||||
success(myWriter, don't forget this...") |
||||
|
||||
|
||||
Or create SprintXxx functions to mix strings with other non-colorized strings: |
||||
|
||||
yellow := New(FgYellow).SprintFunc() |
||||
red := New(FgRed).SprintFunc() |
||||
|
||||
fmt.Printf("this is a %s and this is %s.\n", yellow("warning"), red("error")) |
||||
|
||||
info := New(FgWhite, BgGreen).SprintFunc() |
||||
fmt.Printf("this %s rocks!\n", info("package")) |
||||
|
||||
Windows support is enabled by default. All Print functions work as intended. |
||||
However only for color.SprintXXX functions, user should use fmt.FprintXXX and |
||||
set the output to color.Output: |
||||
|
||||
fmt.Fprintf(color.Output, "Windows support: %s", color.GreenString("PASS")) |
||||
|
||||
info := New(FgWhite, BgGreen).SprintFunc() |
||||
fmt.Fprintf(color.Output, "this %s rocks!\n", info("package")) |
||||
|
||||
Using with existing code is possible. Just use the Set() method to set the |
||||
standard output to the given parameters. That way a rewrite of an existing |
||||
code is not required. |
||||
|
||||
// Use handy standard colors.
|
||||
color.Set(color.FgYellow) |
||||
|
||||
fmt.Println("Existing text will be now in Yellow") |
||||
fmt.Printf("This one %s\n", "too") |
||||
|
||||
color.Unset() // don't forget to unset
|
||||
|
||||
// You can mix up parameters
|
||||
color.Set(color.FgMagenta, color.Bold) |
||||
defer color.Unset() // use it in your function
|
||||
|
||||
fmt.Println("All text will be now bold magenta.") |
||||
|
||||
There might be a case where you want to disable color output (for example to |
||||
pipe the standard output of your app to somewhere else). `Color` has support to |
||||
disable colors both globally and for single color definition. For example |
||||
suppose you have a CLI app and a `--no-color` bool flag. You can easily disable |
||||
the color output with: |
||||
|
||||
var flagNoColor = flag.Bool("no-color", false, "Disable color output") |
||||
|
||||
if *flagNoColor { |
||||
color.NoColor = true // disables colorized output
|
||||
} |
||||
|
||||
It also has support for single color definitions (local). You can |
||||
disable/enable color output on the fly: |
||||
|
||||
c := color.New(color.FgCyan) |
||||
c.Println("Prints cyan text") |
||||
|
||||
c.DisableColor() |
||||
c.Println("This is printed without any color") |
||||
|
||||
c.EnableColor() |
||||
c.Println("This prints again cyan...") |
||||
*/ |
||||
package color |
||||
@ -0,0 +1,8 @@ |
||||
module github.com/fatih/color |
||||
|
||||
go 1.13 |
||||
|
||||
require ( |
||||
github.com/mattn/go-colorable v0.1.4 |
||||
github.com/mattn/go-isatty v0.0.11 |
||||
) |
||||
@ -0,0 +1,8 @@ |
||||
github.com/mattn/go-colorable v0.1.4 h1:snbPLB8fVfU9iwbbo30TPtbLRzwWu6aJS6Xh4eaaviA= |
||||
github.com/mattn/go-colorable v0.1.4/go.mod h1:U0ppj6V5qS13XJ6of8GYAs25YV2eR4EVcfRqFIhoBtE= |
||||
github.com/mattn/go-isatty v0.0.8/go.mod h1:Iq45c/XA43vh69/j3iqttzPXn0bhXyGjM0Hdxcsrc5s= |
||||
github.com/mattn/go-isatty v0.0.11 h1:FxPOTFNqGkuDUGi3H/qkUbQO4ZiBa2brKq5r0l8TGeM= |
||||
github.com/mattn/go-isatty v0.0.11/go.mod h1:PhnuNfih5lzO57/f3n+odYbM4JtupLOxQOAqxQCu2WE= |
||||
golang.org/x/sys v0.0.0-20190222072716-a9d3bda3a223/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= |
||||
golang.org/x/sys v0.0.0-20191026070338-33540a1f6037 h1:YyJpGZS1sBuBCzLAR1VEpK193GlqGZbnPFnPV/5Rsb4= |
||||
golang.org/x/sys v0.0.0-20191026070338-33540a1f6037/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= |
||||
@ -0,0 +1,60 @@ |
||||
Copyright (c) 2017, Fatih Arslan |
||||
All rights reserved. |
||||
|
||||
Redistribution and use in source and binary forms, with or without |
||||
modification, are permitted provided that the following conditions are met: |
||||
|
||||
* Redistributions of source code must retain the above copyright notice, this |
||||
list of conditions and the following disclaimer. |
||||
|
||||
* Redistributions in binary form must reproduce the above copyright notice, |
||||
this list of conditions and the following disclaimer in the documentation |
||||
and/or other materials provided with the distribution. |
||||
|
||||
* Neither the name of structtag nor the names of its |
||||
contributors may be used to endorse or promote products derived from |
||||
this software without specific prior written permission. |
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
||||
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
||||
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE |
||||
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE |
||||
FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
||||
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR |
||||
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER |
||||
CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, |
||||
OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
|
||||
This software includes some portions from Go. Go is used under the terms of the |
||||
BSD like license. |
||||
|
||||
Copyright (c) 2012 The Go Authors. All rights reserved. |
||||
|
||||
Redistribution and use in source and binary forms, with or without |
||||
modification, are permitted provided that the following conditions are |
||||
met: |
||||
|
||||
* Redistributions of source code must retain the above copyright |
||||
notice, this list of conditions and the following disclaimer. |
||||
* Redistributions in binary form must reproduce the above |
||||
copyright notice, this list of conditions and the following disclaimer |
||||
in the documentation and/or other materials provided with the |
||||
distribution. |
||||
* Neither the name of Google Inc. nor the names of its |
||||
contributors may be used to endorse or promote products derived from |
||||
this software without specific prior written permission. |
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
|
||||
The Go gopher was designed by Renee French. http://reneefrench.blogspot.com/ The design is licensed under the Creative Commons 3.0 Attributions license. Read this article for more details: https://blog.golang.org/gopher |
||||
@ -0,0 +1,73 @@ |
||||
# structtag [](http://godoc.org/github.com/fatih/structtag) |
||||
|
||||
structtag provides an easy way of parsing and manipulating struct tag fields. |
||||
Please vendor the library as it might change in future versions. |
||||
|
||||
# Install |
||||
|
||||
```bash |
||||
go get github.com/fatih/structtag |
||||
``` |
||||
|
||||
# Example |
||||
|
||||
```go |
||||
package main |
||||
|
||||
import ( |
||||
"fmt" |
||||
"reflect" |
||||
"sort" |
||||
|
||||
"github.com/fatih/structtag" |
||||
) |
||||
|
||||
func main() { |
||||
type t struct { |
||||
t string `json:"foo,omitempty,string" xml:"foo"` |
||||
} |
||||
|
||||
// get field tag |
||||
tag := reflect.TypeOf(t{}).Field(0).Tag |
||||
|
||||
// ... and start using structtag by parsing the tag |
||||
tags, err := structtag.Parse(string(tag)) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
|
||||
// iterate over all tags |
||||
for _, t := range tags.Tags() { |
||||
fmt.Printf("tag: %+v\n", t) |
||||
} |
||||
|
||||
// get a single tag |
||||
jsonTag, err := tags.Get("json") |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
fmt.Println(jsonTag) // Output: json:"foo,omitempty,string" |
||||
fmt.Println(jsonTag.Key) // Output: json |
||||
fmt.Println(jsonTag.Name) // Output: foo |
||||
fmt.Println(jsonTag.Options) // Output: [omitempty string] |
||||
|
||||
// change existing tag |
||||
jsonTag.Name = "foo_bar" |
||||
jsonTag.Options = nil |
||||
tags.Set(jsonTag) |
||||
|
||||
// add new tag |
||||
tags.Set(&structtag.Tag{ |
||||
Key: "hcl", |
||||
Name: "foo", |
||||
Options: []string{"squash"}, |
||||
}) |
||||
|
||||
// print the tags |
||||
fmt.Println(tags) // Output: json:"foo_bar" xml:"foo" hcl:"foo,squash" |
||||
|
||||
// sort tags according to keys |
||||
sort.Sort(tags) |
||||
fmt.Println(tags) // Output: hcl:"foo,squash" json:"foo_bar" xml:"foo" |
||||
} |
||||
``` |
||||
@ -0,0 +1,3 @@ |
||||
module github.com/fatih/structtag |
||||
|
||||
go 1.12 |
||||
@ -0,0 +1,315 @@ |
||||
package structtag |
||||
|
||||
import ( |
||||
"bytes" |
||||
"errors" |
||||
"fmt" |
||||
"strconv" |
||||
"strings" |
||||
) |
||||
|
||||
var ( |
||||
errTagSyntax = errors.New("bad syntax for struct tag pair") |
||||
errTagKeySyntax = errors.New("bad syntax for struct tag key") |
||||
errTagValueSyntax = errors.New("bad syntax for struct tag value") |
||||
|
||||
errKeyNotSet = errors.New("tag key does not exist") |
||||
errTagNotExist = errors.New("tag does not exist") |
||||
errTagKeyMismatch = errors.New("mismatch between key and tag.key") |
||||
) |
||||
|
||||
// Tags represent a set of tags from a single struct field
|
||||
type Tags struct { |
||||
tags []*Tag |
||||
} |
||||
|
||||
// Tag defines a single struct's string literal tag
|
||||
type Tag struct { |
||||
// Key is the tag key, such as json, xml, etc..
|
||||
// i.e: `json:"foo,omitempty". Here key is: "json"
|
||||
Key string |
||||
|
||||
// Name is a part of the value
|
||||
// i.e: `json:"foo,omitempty". Here name is: "foo"
|
||||
Name string |
||||
|
||||
// Options is a part of the value. It contains a slice of tag options i.e:
|
||||
// `json:"foo,omitempty". Here options is: ["omitempty"]
|
||||
Options []string |
||||
} |
||||
|
||||
// Parse parses a single struct field tag and returns the set of tags.
|
||||
func Parse(tag string) (*Tags, error) { |
||||
var tags []*Tag |
||||
|
||||
hasTag := tag != "" |
||||
|
||||
// NOTE(arslan) following code is from reflect and vet package with some
|
||||
// modifications to collect all necessary information and extend it with
|
||||
// usable methods
|
||||
for tag != "" { |
||||
// Skip leading space.
|
||||
i := 0 |
||||
for i < len(tag) && tag[i] == ' ' { |
||||
i++ |
||||
} |
||||
tag = tag[i:] |
||||
if tag == "" { |
||||
break |
||||
} |
||||
|
||||
// Scan to colon. A space, a quote or a control character is a syntax
|
||||
// error. Strictly speaking, control chars include the range [0x7f,
|
||||
// 0x9f], not just [0x00, 0x1f], but in practice, we ignore the
|
||||
// multi-byte control characters as it is simpler to inspect the tag's
|
||||
// bytes than the tag's runes.
|
||||
i = 0 |
||||
for i < len(tag) && tag[i] > ' ' && tag[i] != ':' && tag[i] != '"' && tag[i] != 0x7f { |
||||
i++ |
||||
} |
||||
|
||||
if i == 0 { |
||||
return nil, errTagKeySyntax |
||||
} |
||||
if i+1 >= len(tag) || tag[i] != ':' { |
||||
return nil, errTagSyntax |
||||
} |
||||
if tag[i+1] != '"' { |
||||
return nil, errTagValueSyntax |
||||
} |
||||
|
||||
key := string(tag[:i]) |
||||
tag = tag[i+1:] |
||||
|
||||
// Scan quoted string to find value.
|
||||
i = 1 |
||||
for i < len(tag) && tag[i] != '"' { |
||||
if tag[i] == '\\' { |
||||
i++ |
||||
} |
||||
i++ |
||||
} |
||||
if i >= len(tag) { |
||||
return nil, errTagValueSyntax |
||||
} |
||||
|
||||
qvalue := string(tag[:i+1]) |
||||
tag = tag[i+1:] |
||||
|
||||
value, err := strconv.Unquote(qvalue) |
||||
if err != nil { |
||||
return nil, errTagValueSyntax |
||||
} |
||||
|
||||
res := strings.Split(value, ",") |
||||
name := res[0] |
||||
options := res[1:] |
||||
if len(options) == 0 { |
||||
options = nil |
||||
} |
||||
|
||||
tags = append(tags, &Tag{ |
||||
Key: key, |
||||
Name: name, |
||||
Options: options, |
||||
}) |
||||
} |
||||
|
||||
if hasTag && len(tags) == 0 { |
||||
return nil, nil |
||||
} |
||||
|
||||
return &Tags{ |
||||
tags: tags, |
||||
}, nil |
||||
} |
||||
|
||||
// Get returns the tag associated with the given key. If the key is present
|
||||
// in the tag the value (which may be empty) is returned. Otherwise the
|
||||
// returned value will be the empty string. The ok return value reports whether
|
||||
// the tag exists or not (which the return value is nil).
|
||||
func (t *Tags) Get(key string) (*Tag, error) { |
||||
for _, tag := range t.tags { |
||||
if tag.Key == key { |
||||
return tag, nil |
||||
} |
||||
} |
||||
|
||||
return nil, errTagNotExist |
||||
} |
||||
|
||||
// Set sets the given tag. If the tag key already exists it'll override it
|
||||
func (t *Tags) Set(tag *Tag) error { |
||||
if tag.Key == "" { |
||||
return errKeyNotSet |
||||
} |
||||
|
||||
added := false |
||||
for i, tg := range t.tags { |
||||
if tg.Key == tag.Key { |
||||
added = true |
||||
t.tags[i] = tag |
||||
} |
||||
} |
||||
|
||||
if !added { |
||||
// this means this is a new tag, add it
|
||||
t.tags = append(t.tags, tag) |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
// AddOptions adds the given option for the given key. If the option already
|
||||
// exists it doesn't add it again.
|
||||
func (t *Tags) AddOptions(key string, options ...string) { |
||||
for i, tag := range t.tags { |
||||
if tag.Key != key { |
||||
continue |
||||
} |
||||
|
||||
for _, opt := range options { |
||||
if !tag.HasOption(opt) { |
||||
tag.Options = append(tag.Options, opt) |
||||
} |
||||
} |
||||
|
||||
t.tags[i] = tag |
||||
} |
||||
} |
||||
|
||||
// DeleteOptions deletes the given options for the given key
|
||||
func (t *Tags) DeleteOptions(key string, options ...string) { |
||||
hasOption := func(option string) bool { |
||||
for _, opt := range options { |
||||
if opt == option { |
||||
return true |
||||
} |
||||
} |
||||
return false |
||||
} |
||||
|
||||
for i, tag := range t.tags { |
||||
if tag.Key != key { |
||||
continue |
||||
} |
||||
|
||||
var updated []string |
||||
for _, opt := range tag.Options { |
||||
if !hasOption(opt) { |
||||
updated = append(updated, opt) |
||||
} |
||||
} |
||||
|
||||
tag.Options = updated |
||||
t.tags[i] = tag |
||||
} |
||||
} |
||||
|
||||
// Delete deletes the tag for the given keys
|
||||
func (t *Tags) Delete(keys ...string) { |
||||
hasKey := func(key string) bool { |
||||
for _, k := range keys { |
||||
if k == key { |
||||
return true |
||||
} |
||||
} |
||||
return false |
||||
} |
||||
|
||||
var updated []*Tag |
||||
for _, tag := range t.tags { |
||||
if !hasKey(tag.Key) { |
||||
updated = append(updated, tag) |
||||
} |
||||
} |
||||
|
||||
t.tags = updated |
||||
} |
||||
|
||||
// Tags returns a slice of tags. The order is the original tag order unless it
|
||||
// was changed.
|
||||
func (t *Tags) Tags() []*Tag { |
||||
return t.tags |
||||
} |
||||
|
||||
// Tags returns a slice of tags. The order is the original tag order unless it
|
||||
// was changed.
|
||||
func (t *Tags) Keys() []string { |
||||
var keys []string |
||||
for _, tag := range t.tags { |
||||
keys = append(keys, tag.Key) |
||||
} |
||||
return keys |
||||
} |
||||
|
||||
// String reassembles the tags into a valid literal tag field representation
|
||||
func (t *Tags) String() string { |
||||
tags := t.Tags() |
||||
if len(tags) == 0 { |
||||
return "" |
||||
} |
||||
|
||||
var buf bytes.Buffer |
||||
for i, tag := range t.Tags() { |
||||
buf.WriteString(tag.String()) |
||||
if i != len(tags)-1 { |
||||
buf.WriteString(" ") |
||||
} |
||||
} |
||||
return buf.String() |
||||
} |
||||
|
||||
// HasOption returns true if the given option is available in options
|
||||
func (t *Tag) HasOption(opt string) bool { |
||||
for _, tagOpt := range t.Options { |
||||
if tagOpt == opt { |
||||
return true |
||||
} |
||||
} |
||||
|
||||
return false |
||||
} |
||||
|
||||
// Value returns the raw value of the tag, i.e. if the tag is
|
||||
// `json:"foo,omitempty", the Value is "foo,omitempty"
|
||||
func (t *Tag) Value() string { |
||||
options := strings.Join(t.Options, ",") |
||||
if options != "" { |
||||
return fmt.Sprintf(`%s,%s`, t.Name, options) |
||||
} |
||||
return t.Name |
||||
} |
||||
|
||||
// String reassembles the tag into a valid tag field representation
|
||||
func (t *Tag) String() string { |
||||
return fmt.Sprintf(`%s:%q`, t.Key, t.Value()) |
||||
} |
||||
|
||||
// GoString implements the fmt.GoStringer interface
|
||||
func (t *Tag) GoString() string { |
||||
template := `{ |
||||
Key: '%s', |
||||
Name: '%s', |
||||
Option: '%s', |
||||
}` |
||||
|
||||
if t.Options == nil { |
||||
return fmt.Sprintf(template, t.Key, t.Name, "nil") |
||||
} |
||||
|
||||
options := strings.Join(t.Options, ",") |
||||
return fmt.Sprintf(template, t.Key, t.Name, options) |
||||
} |
||||
|
||||
func (t *Tags) Len() int { |
||||
return len(t.tags) |
||||
} |
||||
|
||||
func (t *Tags) Less(i int, j int) bool { |
||||
return t.tags[i].Key < t.tags[j].Key |
||||
} |
||||
|
||||
func (t *Tags) Swap(i int, j int) { |
||||
t.tags[i], t.tags[j] = t.tags[j], t.tags[i] |
||||
} |
||||
@ -0,0 +1,9 @@ |
||||
language: go |
||||
go: |
||||
- tip |
||||
|
||||
before_install: |
||||
- go get github.com/mattn/goveralls |
||||
- go get golang.org/x/tools/cmd/cover |
||||
script: |
||||
- $HOME/gopath/bin/goveralls -repotoken xnXqRGwgW3SXIguzxf90ZSK1GPYZPaGrw |
||||
@ -0,0 +1,21 @@ |
||||
The MIT License (MIT) |
||||
|
||||
Copyright (c) 2016 Yasuhiro Matsumoto |
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy |
||||
of this software and associated documentation files (the "Software"), to deal |
||||
in the Software without restriction, including without limitation the rights |
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
||||
copies of the Software, and to permit persons to whom the Software is |
||||
furnished to do so, subject to the following conditions: |
||||
|
||||
The above copyright notice and this permission notice shall be included in all |
||||
copies or substantial portions of the Software. |
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
||||
SOFTWARE. |
||||
@ -0,0 +1,48 @@ |
||||
# go-colorable |
||||
|
||||
[](http://godoc.org/github.com/mattn/go-colorable) |
||||
[](https://travis-ci.org/mattn/go-colorable) |
||||
[](https://coveralls.io/github/mattn/go-colorable?branch=master) |
||||
[](https://goreportcard.com/report/mattn/go-colorable) |
||||
|
||||
Colorable writer for windows. |
||||
|
||||
For example, most of logger packages doesn't show colors on windows. (I know we can do it with ansicon. But I don't want.) |
||||
This package is possible to handle escape sequence for ansi color on windows. |
||||
|
||||
## Too Bad! |
||||
|
||||
 |
||||
|
||||
|
||||
## So Good! |
||||
|
||||
 |
||||
|
||||
## Usage |
||||
|
||||
```go |
||||
logrus.SetFormatter(&logrus.TextFormatter{ForceColors: true}) |
||||
logrus.SetOutput(colorable.NewColorableStdout()) |
||||
|
||||
logrus.Info("succeeded") |
||||
logrus.Warn("not correct") |
||||
logrus.Error("something error") |
||||
logrus.Fatal("panic") |
||||
``` |
||||
|
||||
You can compile above code on non-windows OSs. |
||||
|
||||
## Installation |
||||
|
||||
``` |
||||
$ go get github.com/mattn/go-colorable |
||||
``` |
||||
|
||||
# License |
||||
|
||||
MIT |
||||
|
||||
# Author |
||||
|
||||
Yasuhiro Matsumoto (a.k.a mattn) |
||||
@ -0,0 +1,29 @@ |
||||
// +build appengine
|
||||
|
||||
package colorable |
||||
|
||||
import ( |
||||
"io" |
||||
"os" |
||||
|
||||
_ "github.com/mattn/go-isatty" |
||||
) |
||||
|
||||
// NewColorable returns new instance of Writer which handles escape sequence.
|
||||
func NewColorable(file *os.File) io.Writer { |
||||
if file == nil { |
||||
panic("nil passed instead of *os.File to NewColorable()") |
||||
} |
||||
|
||||
return file |
||||
} |
||||
|
||||
// NewColorableStdout returns new instance of Writer which handles escape sequence for stdout.
|
||||
func NewColorableStdout() io.Writer { |
||||
return os.Stdout |
||||
} |
||||
|
||||
// NewColorableStderr returns new instance of Writer which handles escape sequence for stderr.
|
||||
func NewColorableStderr() io.Writer { |
||||
return os.Stderr |
||||
} |
||||
@ -0,0 +1,30 @@ |
||||
// +build !windows
|
||||
// +build !appengine
|
||||
|
||||
package colorable |
||||
|
||||
import ( |
||||
"io" |
||||
"os" |
||||
|
||||
_ "github.com/mattn/go-isatty" |
||||
) |
||||
|
||||
// NewColorable returns new instance of Writer which handles escape sequence.
|
||||
func NewColorable(file *os.File) io.Writer { |
||||
if file == nil { |
||||
panic("nil passed instead of *os.File to NewColorable()") |
||||
} |
||||
|
||||
return file |
||||
} |
||||
|
||||
// NewColorableStdout returns new instance of Writer which handles escape sequence for stdout.
|
||||
func NewColorableStdout() io.Writer { |
||||
return os.Stdout |
||||
} |
||||
|
||||
// NewColorableStderr returns new instance of Writer which handles escape sequence for stderr.
|
||||
func NewColorableStderr() io.Writer { |
||||
return os.Stderr |
||||
} |
||||
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,3 @@ |
||||
module github.com/mattn/go-colorable |
||||
|
||||
require github.com/mattn/go-isatty v0.0.8 |
||||
@ -0,0 +1,4 @@ |
||||
github.com/mattn/go-isatty v0.0.5 h1:tHXDdz1cpzGaovsTB+TVB8q90WEokoVmfMqoVcrLUgw= |
||||
github.com/mattn/go-isatty v0.0.5/go.mod h1:Iq45c/XA43vh69/j3iqttzPXn0bhXyGjM0Hdxcsrc5s= |
||||
golang.org/x/sys v0.0.0-20190222072716-a9d3bda3a223 h1:DH4skfRX4EBpamg7iV4ZlCpblAHI6s6TDM39bFZumv8= |
||||
golang.org/x/sys v0.0.0-20190222072716-a9d3bda3a223/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= |
||||
@ -0,0 +1,55 @@ |
||||
package colorable |
||||
|
||||
import ( |
||||
"bytes" |
||||
"io" |
||||
) |
||||
|
||||
// NonColorable holds writer but removes escape sequence.
|
||||
type NonColorable struct { |
||||
out io.Writer |
||||
} |
||||
|
||||
// NewNonColorable returns new instance of Writer which removes escape sequence from Writer.
|
||||
func NewNonColorable(w io.Writer) io.Writer { |
||||
return &NonColorable{out: w} |
||||
} |
||||
|
||||
// Write writes data on console
|
||||
func (w *NonColorable) Write(data []byte) (n int, err error) { |
||||
er := bytes.NewReader(data) |
||||
var bw [1]byte |
||||
loop: |
||||
for { |
||||
c1, err := er.ReadByte() |
||||
if err != nil { |
||||
break loop |
||||
} |
||||
if c1 != 0x1b { |
||||
bw[0] = c1 |
||||
w.out.Write(bw[:]) |
||||
continue |
||||
} |
||||
c2, err := er.ReadByte() |
||||
if err != nil { |
||||
break loop |
||||
} |
||||
if c2 != 0x5b { |
||||
continue |
||||
} |
||||
|
||||
var buf bytes.Buffer |
||||
for { |
||||
c, err := er.ReadByte() |
||||
if err != nil { |
||||
break loop |
||||
} |
||||
if ('a' <= c && c <= 'z') || ('A' <= c && c <= 'Z') || c == '@' { |
||||
break |
||||
} |
||||
buf.Write([]byte(string(c))) |
||||
} |
||||
} |
||||
|
||||
return len(data), nil |
||||
} |
||||
@ -1,3 +1,5 @@ |
||||
module github.com/mattn/go-isatty |
||||
|
||||
require golang.org/x/sys v0.0.0-20190222072716-a9d3bda3a223 |
||||
go 1.12 |
||||
|
||||
require golang.org/x/sys v0.0.0-20191026070338-33540a1f6037 |
||||
|
||||
@ -1,2 +1,2 @@ |
||||
golang.org/x/sys v0.0.0-20190222072716-a9d3bda3a223 h1:DH4skfRX4EBpamg7iV4ZlCpblAHI6s6TDM39bFZumv8= |
||||
golang.org/x/sys v0.0.0-20190222072716-a9d3bda3a223/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= |
||||
golang.org/x/sys v0.0.0-20191026070338-33540a1f6037 h1:YyJpGZS1sBuBCzLAR1VEpK193GlqGZbnPFnPV/5Rsb4= |
||||
golang.org/x/sys v0.0.0-20191026070338-33540a1f6037/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= |
||||
|
||||
@ -0,0 +1,22 @@ |
||||
// +build plan9
|
||||
|
||||
package isatty |
||||
|
||||
import ( |
||||
"syscall" |
||||
) |
||||
|
||||
// IsTerminal returns true if the given file descriptor is a terminal.
|
||||
func IsTerminal(fd uintptr) bool { |
||||
path, err := syscall.Fd2path(int(fd)) |
||||
if err != nil { |
||||
return false |
||||
} |
||||
return path == "/dev/cons" || path == "/mnt/term/dev/cons" |
||||
} |
||||
|
||||
// IsCygwinTerminal return true if the file descriptor is a cygwin or msys2
|
||||
// terminal. This is also always false on this environment.
|
||||
func IsCygwinTerminal(fd uintptr) bool { |
||||
return false |
||||
} |
||||
@ -1,4 +1,4 @@ |
||||
// +build linux
|
||||
// +build linux aix
|
||||
// +build !appengine
|
||||
// +build !android
|
||||
|
||||
@ -0,0 +1,8 @@ |
||||
language: go |
||||
go: |
||||
- tip |
||||
before_install: |
||||
- go get github.com/mattn/goveralls |
||||
- go get golang.org/x/tools/cmd/cover |
||||
script: |
||||
- $HOME/gopath/bin/goveralls -repotoken lAKAWPzcGsD3A8yBX3BGGtRUdJ6CaGERL |
||||
@ -0,0 +1,21 @@ |
||||
The MIT License (MIT) |
||||
|
||||
Copyright (c) 2016 Yasuhiro Matsumoto |
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy |
||||
of this software and associated documentation files (the "Software"), to deal |
||||
in the Software without restriction, including without limitation the rights |
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
||||
copies of the Software, and to permit persons to whom the Software is |
||||
furnished to do so, subject to the following conditions: |
||||
|
||||
The above copyright notice and this permission notice shall be included in all |
||||
copies or substantial portions of the Software. |
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
||||
SOFTWARE. |
||||
@ -0,0 +1,27 @@ |
||||
go-runewidth |
||||
============ |
||||
|
||||
[](https://travis-ci.org/mattn/go-runewidth) |
||||
[](https://coveralls.io/r/mattn/go-runewidth?branch=HEAD) |
||||
[](http://godoc.org/github.com/mattn/go-runewidth) |
||||
[](https://goreportcard.com/report/github.com/mattn/go-runewidth) |
||||
|
||||
Provides functions to get fixed width of the character or string. |
||||
|
||||
Usage |
||||
----- |
||||
|
||||
```go |
||||
runewidth.StringWidth("つのだ☆HIRO") == 12 |
||||
``` |
||||
|
||||
|
||||
Author |
||||
------ |
||||
|
||||
Yasuhiro Matsumoto |
||||
|
||||
License |
||||
------- |
||||
|
||||
under the MIT License: http://mattn.mit-license.org/2013 |
||||
@ -0,0 +1,3 @@ |
||||
module github.com/mattn/go-runewidth |
||||
|
||||
go 1.9 |
||||
@ -0,0 +1,258 @@ |
||||
package runewidth |
||||
|
||||
import ( |
||||
"os" |
||||
) |
||||
|
||||
//go:generate go run script/generate.go
|
||||
|
||||
var ( |
||||
// EastAsianWidth will be set true if the current locale is CJK
|
||||
EastAsianWidth bool |
||||
|
||||
// ZeroWidthJoiner is flag to set to use UTR#51 ZWJ
|
||||
ZeroWidthJoiner bool |
||||
|
||||
// DefaultCondition is a condition in current locale
|
||||
DefaultCondition = &Condition{} |
||||
) |
||||
|
||||
func init() { |
||||
handleEnv() |
||||
} |
||||
|
||||
func handleEnv() { |
||||
env := os.Getenv("RUNEWIDTH_EASTASIAN") |
||||
if env == "" { |
||||
EastAsianWidth = IsEastAsian() |
||||
} else { |
||||
EastAsianWidth = env == "1" |
||||
} |
||||
// update DefaultCondition
|
||||
DefaultCondition.EastAsianWidth = EastAsianWidth |
||||
DefaultCondition.ZeroWidthJoiner = ZeroWidthJoiner |
||||
} |
||||
|
||||
type interval struct { |
||||
first rune |
||||
last rune |
||||
} |
||||
|
||||
type table []interval |
||||
|
||||
func inTables(r rune, ts ...table) bool { |
||||
for _, t := range ts { |
||||
if inTable(r, t) { |
||||
return true |
||||
} |
||||
} |
||||
return false |
||||
} |
||||
|
||||
func inTable(r rune, t table) bool { |
||||
// func (t table) IncludesRune(r rune) bool {
|
||||
if r < t[0].first { |
||||
return false |
||||
} |
||||
|
||||
bot := 0 |
||||
top := len(t) - 1 |
||||
for top >= bot { |
||||
mid := (bot + top) >> 1 |
||||
|
||||
switch { |
||||
case t[mid].last < r: |
||||
bot = mid + 1 |
||||
case t[mid].first > r: |
||||
top = mid - 1 |
||||
default: |
||||
return true |
||||
} |
||||
} |
||||
|
||||
return false |
||||
} |
||||
|
||||
var private = table{ |
||||
{0x00E000, 0x00F8FF}, {0x0F0000, 0x0FFFFD}, {0x100000, 0x10FFFD}, |
||||
} |
||||
|
||||
var nonprint = table{ |
||||
{0x0000, 0x001F}, {0x007F, 0x009F}, {0x00AD, 0x00AD}, |
||||
{0x070F, 0x070F}, {0x180B, 0x180E}, {0x200B, 0x200F}, |
||||
{0x2028, 0x202E}, {0x206A, 0x206F}, {0xD800, 0xDFFF}, |
||||
{0xFEFF, 0xFEFF}, {0xFFF9, 0xFFFB}, {0xFFFE, 0xFFFF}, |
||||
} |
||||
|
||||
// Condition have flag EastAsianWidth whether the current locale is CJK or not.
|
||||
type Condition struct { |
||||
EastAsianWidth bool |
||||
ZeroWidthJoiner bool |
||||
} |
||||
|
||||
// NewCondition return new instance of Condition which is current locale.
|
||||
func NewCondition() *Condition { |
||||
return &Condition{ |
||||
EastAsianWidth: EastAsianWidth, |
||||
ZeroWidthJoiner: ZeroWidthJoiner, |
||||
} |
||||
} |
||||
|
||||
// RuneWidth returns the number of cells in r.
|
||||
// See http://www.unicode.org/reports/tr11/
|
||||
func (c *Condition) RuneWidth(r rune) int { |
||||
switch { |
||||
case r < 0 || r > 0x10FFFF || inTables(r, nonprint, combining, notassigned): |
||||
return 0 |
||||
case (c.EastAsianWidth && IsAmbiguousWidth(r)) || inTables(r, doublewidth): |
||||
return 2 |
||||
default: |
||||
return 1 |
||||
} |
||||
} |
||||
|
||||
func (c *Condition) stringWidth(s string) (width int) { |
||||
for _, r := range []rune(s) { |
||||
width += c.RuneWidth(r) |
||||
} |
||||
return width |
||||
} |
||||
|
||||
func (c *Condition) stringWidthZeroJoiner(s string) (width int) { |
||||
r1, r2 := rune(0), rune(0) |
||||
for _, r := range []rune(s) { |
||||
if r == 0xFE0E || r == 0xFE0F { |
||||
continue |
||||
} |
||||
w := c.RuneWidth(r) |
||||
if r2 == 0x200D && inTables(r, emoji) && inTables(r1, emoji) { |
||||
if width < w { |
||||
width = w |
||||
} |
||||
} else { |
||||
width += w |
||||
} |
||||
r1, r2 = r2, r |
||||
} |
||||
return width |
||||
} |
||||
|
||||
// StringWidth return width as you can see
|
||||
func (c *Condition) StringWidth(s string) (width int) { |
||||
if c.ZeroWidthJoiner { |
||||
return c.stringWidthZeroJoiner(s) |
||||
} |
||||
return c.stringWidth(s) |
||||
} |
||||
|
||||
// Truncate return string truncated with w cells
|
||||
func (c *Condition) Truncate(s string, w int, tail string) string { |
||||
if c.StringWidth(s) <= w { |
||||
return s |
||||
} |
||||
r := []rune(s) |
||||
tw := c.StringWidth(tail) |
||||
w -= tw |
||||
width := 0 |
||||
i := 0 |
||||
for ; i < len(r); i++ { |
||||
cw := c.RuneWidth(r[i]) |
||||
if width+cw > w { |
||||
break |
||||
} |
||||
width += cw |
||||
} |
||||
return string(r[0:i]) + tail |
||||
} |
||||
|
||||
// Wrap return string wrapped with w cells
|
||||
func (c *Condition) Wrap(s string, w int) string { |
||||
width := 0 |
||||
out := "" |
||||
for _, r := range []rune(s) { |
||||
cw := RuneWidth(r) |
||||
if r == '\n' { |
||||
out += string(r) |
||||
width = 0 |
||||
continue |
||||
} else if width+cw > w { |
||||
out += "\n" |
||||
width = 0 |
||||
out += string(r) |
||||
width += cw |
||||
continue |
||||
} |
||||
out += string(r) |
||||
width += cw |
||||
} |
||||
return out |
||||
} |
||||
|
||||
// FillLeft return string filled in left by spaces in w cells
|
||||
func (c *Condition) FillLeft(s string, w int) string { |
||||
width := c.StringWidth(s) |
||||
count := w - width |
||||
if count > 0 { |
||||
b := make([]byte, count) |
||||
for i := range b { |
||||
b[i] = ' ' |
||||
} |
||||
return string(b) + s |
||||
} |
||||
return s |
||||
} |
||||
|
||||
// FillRight return string filled in left by spaces in w cells
|
||||
func (c *Condition) FillRight(s string, w int) string { |
||||
width := c.StringWidth(s) |
||||
count := w - width |
||||
if count > 0 { |
||||
b := make([]byte, count) |
||||
for i := range b { |
||||
b[i] = ' ' |
||||
} |
||||
return s + string(b) |
||||
} |
||||
return s |
||||
} |
||||
|
||||
// RuneWidth returns the number of cells in r.
|
||||
// See http://www.unicode.org/reports/tr11/
|
||||
func RuneWidth(r rune) int { |
||||
return DefaultCondition.RuneWidth(r) |
||||
} |
||||
|
||||
// IsAmbiguousWidth returns whether is ambiguous width or not.
|
||||
func IsAmbiguousWidth(r rune) bool { |
||||
return inTables(r, private, ambiguous) |
||||
} |
||||
|
||||
// IsNeutralWidth returns whether is neutral width or not.
|
||||
func IsNeutralWidth(r rune) bool { |
||||
return inTable(r, neutral) |
||||
} |
||||
|
||||
// StringWidth return width as you can see
|
||||
func StringWidth(s string) (width int) { |
||||
return DefaultCondition.StringWidth(s) |
||||
} |
||||
|
||||
// Truncate return string truncated with w cells
|
||||
func Truncate(s string, w int, tail string) string { |
||||
return DefaultCondition.Truncate(s, w, tail) |
||||
} |
||||
|
||||
// Wrap return string wrapped with w cells
|
||||
func Wrap(s string, w int) string { |
||||
return DefaultCondition.Wrap(s, w) |
||||
} |
||||
|
||||
// FillLeft return string filled in left by spaces in w cells
|
||||
func FillLeft(s string, w int) string { |
||||
return DefaultCondition.FillLeft(s, w) |
||||
} |
||||
|
||||
// FillRight return string filled in left by spaces in w cells
|
||||
func FillRight(s string, w int) string { |
||||
return DefaultCondition.FillRight(s, w) |
||||
} |
||||
@ -0,0 +1,8 @@ |
||||
// +build appengine
|
||||
|
||||
package runewidth |
||||
|
||||
// IsEastAsian return true if the current locale is CJK
|
||||
func IsEastAsian() bool { |
||||
return false |
||||
} |
||||
@ -0,0 +1,9 @@ |
||||
// +build js
|
||||
// +build !appengine
|
||||
|
||||
package runewidth |
||||
|
||||
func IsEastAsian() bool { |
||||
// TODO: Implement this for the web. Detect east asian in a compatible way, and return true.
|
||||
return false |
||||
} |
||||
@ -0,0 +1,79 @@ |
||||
// +build !windows
|
||||
// +build !js
|
||||
// +build !appengine
|
||||
|
||||
package runewidth |
||||
|
||||
import ( |
||||
"os" |
||||
"regexp" |
||||
"strings" |
||||
) |
||||
|
||||
var reLoc = regexp.MustCompile(`^[a-z][a-z][a-z]?(?:_[A-Z][A-Z])?\.(.+)`) |
||||
|
||||
var mblenTable = map[string]int{ |
||||
"utf-8": 6, |
||||
"utf8": 6, |
||||
"jis": 8, |
||||
"eucjp": 3, |
||||
"euckr": 2, |
||||
"euccn": 2, |
||||
"sjis": 2, |
||||
"cp932": 2, |
||||
"cp51932": 2, |
||||
"cp936": 2, |
||||
"cp949": 2, |
||||
"cp950": 2, |
||||
"big5": 2, |
||||
"gbk": 2, |
||||
"gb2312": 2, |
||||
} |
||||
|
||||
func isEastAsian(locale string) bool { |
||||
charset := strings.ToLower(locale) |
||||
r := reLoc.FindStringSubmatch(locale) |
||||
if len(r) == 2 { |
||||
charset = strings.ToLower(r[1]) |
||||
} |
||||
|
||||
if strings.HasSuffix(charset, "@cjk_narrow") { |
||||
return false |
||||
} |
||||
|
||||
for pos, b := range []byte(charset) { |
||||
if b == '@' { |
||||
charset = charset[:pos] |
||||
break |
||||
} |
||||
} |
||||
max := 1 |
||||
if m, ok := mblenTable[charset]; ok { |
||||
max = m |
||||
} |
||||
if max > 1 && (charset[0] != 'u' || |
||||
strings.HasPrefix(locale, "ja") || |
||||
strings.HasPrefix(locale, "ko") || |
||||
strings.HasPrefix(locale, "zh")) { |
||||
return true |
||||
} |
||||
return false |
||||
} |
||||
|
||||
// IsEastAsian return true if the current locale is CJK
|
||||
func IsEastAsian() bool { |
||||
locale := os.Getenv("LC_CTYPE") |
||||
if locale == "" { |
||||
locale = os.Getenv("LANG") |
||||
} |
||||
|
||||
// ignore C locale
|
||||
if locale == "POSIX" || locale == "C" { |
||||
return false |
||||
} |
||||
if len(locale) > 1 && locale[0] == 'C' && (locale[1] == '.' || locale[1] == '-') { |
||||
return false |
||||
} |
||||
|
||||
return isEastAsian(locale) |
||||
} |
||||
@ -0,0 +1,427 @@ |
||||
package runewidth |
||||
|
||||
var combining = table{ |
||||
{0x0300, 0x036F}, {0x0483, 0x0489}, {0x07EB, 0x07F3}, |
||||
{0x0C00, 0x0C00}, {0x0C04, 0x0C04}, {0x0D00, 0x0D01}, |
||||
{0x135D, 0x135F}, {0x1A7F, 0x1A7F}, {0x1AB0, 0x1ABE}, |
||||
{0x1B6B, 0x1B73}, {0x1DC0, 0x1DF9}, {0x1DFB, 0x1DFF}, |
||||
{0x20D0, 0x20F0}, {0x2CEF, 0x2CF1}, {0x2DE0, 0x2DFF}, |
||||
{0x3099, 0x309A}, {0xA66F, 0xA672}, {0xA674, 0xA67D}, |
||||
{0xA69E, 0xA69F}, {0xA6F0, 0xA6F1}, {0xA8E0, 0xA8F1}, |
||||
{0xFE20, 0xFE2F}, {0x101FD, 0x101FD}, {0x10376, 0x1037A}, |
||||
{0x10F46, 0x10F50}, {0x11300, 0x11301}, {0x1133B, 0x1133C}, |
||||
{0x11366, 0x1136C}, {0x11370, 0x11374}, {0x16AF0, 0x16AF4}, |
||||
{0x1D165, 0x1D169}, {0x1D16D, 0x1D172}, {0x1D17B, 0x1D182}, |
||||
{0x1D185, 0x1D18B}, {0x1D1AA, 0x1D1AD}, {0x1D242, 0x1D244}, |
||||
{0x1E000, 0x1E006}, {0x1E008, 0x1E018}, {0x1E01B, 0x1E021}, |
||||
{0x1E023, 0x1E024}, {0x1E026, 0x1E02A}, {0x1E8D0, 0x1E8D6}, |
||||
} |
||||
|
||||
var doublewidth = table{ |
||||
{0x1100, 0x115F}, {0x231A, 0x231B}, {0x2329, 0x232A}, |
||||
{0x23E9, 0x23EC}, {0x23F0, 0x23F0}, {0x23F3, 0x23F3}, |
||||
{0x25FD, 0x25FE}, {0x2614, 0x2615}, {0x2648, 0x2653}, |
||||
{0x267F, 0x267F}, {0x2693, 0x2693}, {0x26A1, 0x26A1}, |
||||
{0x26AA, 0x26AB}, {0x26BD, 0x26BE}, {0x26C4, 0x26C5}, |
||||
{0x26CE, 0x26CE}, {0x26D4, 0x26D4}, {0x26EA, 0x26EA}, |
||||
{0x26F2, 0x26F3}, {0x26F5, 0x26F5}, {0x26FA, 0x26FA}, |
||||
{0x26FD, 0x26FD}, {0x2705, 0x2705}, {0x270A, 0x270B}, |
||||
{0x2728, 0x2728}, {0x274C, 0x274C}, {0x274E, 0x274E}, |
||||
{0x2753, 0x2755}, {0x2757, 0x2757}, {0x2795, 0x2797}, |
||||
{0x27B0, 0x27B0}, {0x27BF, 0x27BF}, {0x2B1B, 0x2B1C}, |
||||
{0x2B50, 0x2B50}, {0x2B55, 0x2B55}, {0x2E80, 0x2E99}, |
||||
{0x2E9B, 0x2EF3}, {0x2F00, 0x2FD5}, {0x2FF0, 0x2FFB}, |
||||
{0x3000, 0x303E}, {0x3041, 0x3096}, {0x3099, 0x30FF}, |
||||
{0x3105, 0x312F}, {0x3131, 0x318E}, {0x3190, 0x31BA}, |
||||
{0x31C0, 0x31E3}, {0x31F0, 0x321E}, {0x3220, 0x3247}, |
||||
{0x3250, 0x4DBF}, {0x4E00, 0xA48C}, {0xA490, 0xA4C6}, |
||||
{0xA960, 0xA97C}, {0xAC00, 0xD7A3}, {0xF900, 0xFAFF}, |
||||
{0xFE10, 0xFE19}, {0xFE30, 0xFE52}, {0xFE54, 0xFE66}, |
||||
{0xFE68, 0xFE6B}, {0xFF01, 0xFF60}, {0xFFE0, 0xFFE6}, |
||||
{0x16FE0, 0x16FE3}, {0x17000, 0x187F7}, {0x18800, 0x18AF2}, |
||||
{0x1B000, 0x1B11E}, {0x1B150, 0x1B152}, {0x1B164, 0x1B167}, |
||||
{0x1B170, 0x1B2FB}, {0x1F004, 0x1F004}, {0x1F0CF, 0x1F0CF}, |
||||
{0x1F18E, 0x1F18E}, {0x1F191, 0x1F19A}, {0x1F200, 0x1F202}, |
||||
{0x1F210, 0x1F23B}, {0x1F240, 0x1F248}, {0x1F250, 0x1F251}, |
||||
{0x1F260, 0x1F265}, {0x1F300, 0x1F320}, {0x1F32D, 0x1F335}, |
||||
{0x1F337, 0x1F37C}, {0x1F37E, 0x1F393}, {0x1F3A0, 0x1F3CA}, |
||||
{0x1F3CF, 0x1F3D3}, {0x1F3E0, 0x1F3F0}, {0x1F3F4, 0x1F3F4}, |
||||
{0x1F3F8, 0x1F43E}, {0x1F440, 0x1F440}, {0x1F442, 0x1F4FC}, |
||||
{0x1F4FF, 0x1F53D}, {0x1F54B, 0x1F54E}, {0x1F550, 0x1F567}, |
||||
{0x1F57A, 0x1F57A}, {0x1F595, 0x1F596}, {0x1F5A4, 0x1F5A4}, |
||||
{0x1F5FB, 0x1F64F}, {0x1F680, 0x1F6C5}, {0x1F6CC, 0x1F6CC}, |
||||
{0x1F6D0, 0x1F6D2}, {0x1F6D5, 0x1F6D5}, {0x1F6EB, 0x1F6EC}, |
||||
{0x1F6F4, 0x1F6FA}, {0x1F7E0, 0x1F7EB}, {0x1F90D, 0x1F971}, |
||||
{0x1F973, 0x1F976}, {0x1F97A, 0x1F9A2}, {0x1F9A5, 0x1F9AA}, |
||||
{0x1F9AE, 0x1F9CA}, {0x1F9CD, 0x1F9FF}, {0x1FA70, 0x1FA73}, |
||||
{0x1FA78, 0x1FA7A}, {0x1FA80, 0x1FA82}, {0x1FA90, 0x1FA95}, |
||||
{0x20000, 0x2FFFD}, {0x30000, 0x3FFFD}, |
||||
} |
||||
|
||||
var ambiguous = table{ |
||||
{0x00A1, 0x00A1}, {0x00A4, 0x00A4}, {0x00A7, 0x00A8}, |
||||
{0x00AA, 0x00AA}, {0x00AD, 0x00AE}, {0x00B0, 0x00B4}, |
||||
{0x00B6, 0x00BA}, {0x00BC, 0x00BF}, {0x00C6, 0x00C6}, |
||||
{0x00D0, 0x00D0}, {0x00D7, 0x00D8}, {0x00DE, 0x00E1}, |
||||
{0x00E6, 0x00E6}, {0x00E8, 0x00EA}, {0x00EC, 0x00ED}, |
||||
{0x00F0, 0x00F0}, {0x00F2, 0x00F3}, {0x00F7, 0x00FA}, |
||||
{0x00FC, 0x00FC}, {0x00FE, 0x00FE}, {0x0101, 0x0101}, |
||||
{0x0111, 0x0111}, {0x0113, 0x0113}, {0x011B, 0x011B}, |
||||
{0x0126, 0x0127}, {0x012B, 0x012B}, {0x0131, 0x0133}, |
||||
{0x0138, 0x0138}, {0x013F, 0x0142}, {0x0144, 0x0144}, |
||||
{0x0148, 0x014B}, {0x014D, 0x014D}, {0x0152, 0x0153}, |
||||
{0x0166, 0x0167}, {0x016B, 0x016B}, {0x01CE, 0x01CE}, |
||||
{0x01D0, 0x01D0}, {0x01D2, 0x01D2}, {0x01D4, 0x01D4}, |
||||
{0x01D6, 0x01D6}, {0x01D8, 0x01D8}, {0x01DA, 0x01DA}, |
||||
{0x01DC, 0x01DC}, {0x0251, 0x0251}, {0x0261, 0x0261}, |
||||
{0x02C4, 0x02C4}, {0x02C7, 0x02C7}, {0x02C9, 0x02CB}, |
||||
{0x02CD, 0x02CD}, {0x02D0, 0x02D0}, {0x02D8, 0x02DB}, |
||||
{0x02DD, 0x02DD}, {0x02DF, 0x02DF}, {0x0300, 0x036F}, |
||||
{0x0391, 0x03A1}, {0x03A3, 0x03A9}, {0x03B1, 0x03C1}, |
||||
{0x03C3, 0x03C9}, {0x0401, 0x0401}, {0x0410, 0x044F}, |
||||
{0x0451, 0x0451}, {0x2010, 0x2010}, {0x2013, 0x2016}, |
||||
{0x2018, 0x2019}, {0x201C, 0x201D}, {0x2020, 0x2022}, |
||||
{0x2024, 0x2027}, {0x2030, 0x2030}, {0x2032, 0x2033}, |
||||
{0x2035, 0x2035}, {0x203B, 0x203B}, {0x203E, 0x203E}, |
||||
{0x2074, 0x2074}, {0x207F, 0x207F}, {0x2081, 0x2084}, |
||||
{0x20AC, 0x20AC}, {0x2103, 0x2103}, {0x2105, 0x2105}, |
||||
{0x2109, 0x2109}, {0x2113, 0x2113}, {0x2116, 0x2116}, |
||||
{0x2121, 0x2122}, {0x2126, 0x2126}, {0x212B, 0x212B}, |
||||
{0x2153, 0x2154}, {0x215B, 0x215E}, {0x2160, 0x216B}, |
||||
{0x2170, 0x2179}, {0x2189, 0x2189}, {0x2190, 0x2199}, |
||||
{0x21B8, 0x21B9}, {0x21D2, 0x21D2}, {0x21D4, 0x21D4}, |
||||
{0x21E7, 0x21E7}, {0x2200, 0x2200}, {0x2202, 0x2203}, |
||||
{0x2207, 0x2208}, {0x220B, 0x220B}, {0x220F, 0x220F}, |
||||
{0x2211, 0x2211}, {0x2215, 0x2215}, {0x221A, 0x221A}, |
||||
{0x221D, 0x2220}, {0x2223, 0x2223}, {0x2225, 0x2225}, |
||||
{0x2227, 0x222C}, {0x222E, 0x222E}, {0x2234, 0x2237}, |
||||
{0x223C, 0x223D}, {0x2248, 0x2248}, {0x224C, 0x224C}, |
||||
{0x2252, 0x2252}, {0x2260, 0x2261}, {0x2264, 0x2267}, |
||||
{0x226A, 0x226B}, {0x226E, 0x226F}, {0x2282, 0x2283}, |
||||
{0x2286, 0x2287}, {0x2295, 0x2295}, {0x2299, 0x2299}, |
||||
{0x22A5, 0x22A5}, {0x22BF, 0x22BF}, {0x2312, 0x2312}, |
||||
{0x2460, 0x24E9}, {0x24EB, 0x254B}, {0x2550, 0x2573}, |
||||
{0x2580, 0x258F}, {0x2592, 0x2595}, {0x25A0, 0x25A1}, |
||||
{0x25A3, 0x25A9}, {0x25B2, 0x25B3}, {0x25B6, 0x25B7}, |
||||
{0x25BC, 0x25BD}, {0x25C0, 0x25C1}, {0x25C6, 0x25C8}, |
||||
{0x25CB, 0x25CB}, {0x25CE, 0x25D1}, {0x25E2, 0x25E5}, |
||||
{0x25EF, 0x25EF}, {0x2605, 0x2606}, {0x2609, 0x2609}, |
||||
{0x260E, 0x260F}, {0x261C, 0x261C}, {0x261E, 0x261E}, |
||||
{0x2640, 0x2640}, {0x2642, 0x2642}, {0x2660, 0x2661}, |
||||
{0x2663, 0x2665}, {0x2667, 0x266A}, {0x266C, 0x266D}, |
||||
{0x266F, 0x266F}, {0x269E, 0x269F}, {0x26BF, 0x26BF}, |
||||
{0x26C6, 0x26CD}, {0x26CF, 0x26D3}, {0x26D5, 0x26E1}, |
||||
{0x26E3, 0x26E3}, {0x26E8, 0x26E9}, {0x26EB, 0x26F1}, |
||||
{0x26F4, 0x26F4}, {0x26F6, 0x26F9}, {0x26FB, 0x26FC}, |
||||
{0x26FE, 0x26FF}, {0x273D, 0x273D}, {0x2776, 0x277F}, |
||||
{0x2B56, 0x2B59}, {0x3248, 0x324F}, {0xE000, 0xF8FF}, |
||||
{0xFE00, 0xFE0F}, {0xFFFD, 0xFFFD}, {0x1F100, 0x1F10A}, |
||||
{0x1F110, 0x1F12D}, {0x1F130, 0x1F169}, {0x1F170, 0x1F18D}, |
||||
{0x1F18F, 0x1F190}, {0x1F19B, 0x1F1AC}, {0xE0100, 0xE01EF}, |
||||
{0xF0000, 0xFFFFD}, {0x100000, 0x10FFFD}, |
||||
} |
||||
var notassigned = table{ |
||||
{0x27E6, 0x27ED}, {0x2985, 0x2986}, |
||||
} |
||||
|
||||
var neutral = table{ |
||||
{0x0000, 0x001F}, {0x007F, 0x00A0}, {0x00A9, 0x00A9}, |
||||
{0x00AB, 0x00AB}, {0x00B5, 0x00B5}, {0x00BB, 0x00BB}, |
||||
{0x00C0, 0x00C5}, {0x00C7, 0x00CF}, {0x00D1, 0x00D6}, |
||||
{0x00D9, 0x00DD}, {0x00E2, 0x00E5}, {0x00E7, 0x00E7}, |
||||
{0x00EB, 0x00EB}, {0x00EE, 0x00EF}, {0x00F1, 0x00F1}, |
||||
{0x00F4, 0x00F6}, {0x00FB, 0x00FB}, {0x00FD, 0x00FD}, |
||||
{0x00FF, 0x0100}, {0x0102, 0x0110}, {0x0112, 0x0112}, |
||||
{0x0114, 0x011A}, {0x011C, 0x0125}, {0x0128, 0x012A}, |
||||
{0x012C, 0x0130}, {0x0134, 0x0137}, {0x0139, 0x013E}, |
||||
{0x0143, 0x0143}, {0x0145, 0x0147}, {0x014C, 0x014C}, |
||||
{0x014E, 0x0151}, {0x0154, 0x0165}, {0x0168, 0x016A}, |
||||
{0x016C, 0x01CD}, {0x01CF, 0x01CF}, {0x01D1, 0x01D1}, |
||||
{0x01D3, 0x01D3}, {0x01D5, 0x01D5}, {0x01D7, 0x01D7}, |
||||
{0x01D9, 0x01D9}, {0x01DB, 0x01DB}, {0x01DD, 0x0250}, |
||||
{0x0252, 0x0260}, {0x0262, 0x02C3}, {0x02C5, 0x02C6}, |
||||
{0x02C8, 0x02C8}, {0x02CC, 0x02CC}, {0x02CE, 0x02CF}, |
||||
{0x02D1, 0x02D7}, {0x02DC, 0x02DC}, {0x02DE, 0x02DE}, |
||||
{0x02E0, 0x02FF}, {0x0370, 0x0377}, {0x037A, 0x037F}, |
||||
{0x0384, 0x038A}, {0x038C, 0x038C}, {0x038E, 0x0390}, |
||||
{0x03AA, 0x03B0}, {0x03C2, 0x03C2}, {0x03CA, 0x0400}, |
||||
{0x0402, 0x040F}, {0x0450, 0x0450}, {0x0452, 0x052F}, |
||||
{0x0531, 0x0556}, {0x0559, 0x058A}, {0x058D, 0x058F}, |
||||
{0x0591, 0x05C7}, {0x05D0, 0x05EA}, {0x05EF, 0x05F4}, |
||||
{0x0600, 0x061C}, {0x061E, 0x070D}, {0x070F, 0x074A}, |
||||
{0x074D, 0x07B1}, {0x07C0, 0x07FA}, {0x07FD, 0x082D}, |
||||
{0x0830, 0x083E}, {0x0840, 0x085B}, {0x085E, 0x085E}, |
||||
{0x0860, 0x086A}, {0x08A0, 0x08B4}, {0x08B6, 0x08BD}, |
||||
{0x08D3, 0x0983}, {0x0985, 0x098C}, {0x098F, 0x0990}, |
||||
{0x0993, 0x09A8}, {0x09AA, 0x09B0}, {0x09B2, 0x09B2}, |
||||
{0x09B6, 0x09B9}, {0x09BC, 0x09C4}, {0x09C7, 0x09C8}, |
||||
{0x09CB, 0x09CE}, {0x09D7, 0x09D7}, {0x09DC, 0x09DD}, |
||||
{0x09DF, 0x09E3}, {0x09E6, 0x09FE}, {0x0A01, 0x0A03}, |
||||
{0x0A05, 0x0A0A}, {0x0A0F, 0x0A10}, {0x0A13, 0x0A28}, |
||||
{0x0A2A, 0x0A30}, {0x0A32, 0x0A33}, {0x0A35, 0x0A36}, |
||||
{0x0A38, 0x0A39}, {0x0A3C, 0x0A3C}, {0x0A3E, 0x0A42}, |
||||
{0x0A47, 0x0A48}, {0x0A4B, 0x0A4D}, {0x0A51, 0x0A51}, |
||||
{0x0A59, 0x0A5C}, {0x0A5E, 0x0A5E}, {0x0A66, 0x0A76}, |
||||
{0x0A81, 0x0A83}, {0x0A85, 0x0A8D}, {0x0A8F, 0x0A91}, |
||||
{0x0A93, 0x0AA8}, {0x0AAA, 0x0AB0}, {0x0AB2, 0x0AB3}, |
||||
{0x0AB5, 0x0AB9}, {0x0ABC, 0x0AC5}, {0x0AC7, 0x0AC9}, |
||||
{0x0ACB, 0x0ACD}, {0x0AD0, 0x0AD0}, {0x0AE0, 0x0AE3}, |
||||
{0x0AE6, 0x0AF1}, {0x0AF9, 0x0AFF}, {0x0B01, 0x0B03}, |
||||
{0x0B05, 0x0B0C}, {0x0B0F, 0x0B10}, {0x0B13, 0x0B28}, |
||||
{0x0B2A, 0x0B30}, {0x0B32, 0x0B33}, {0x0B35, 0x0B39}, |
||||
{0x0B3C, 0x0B44}, {0x0B47, 0x0B48}, {0x0B4B, 0x0B4D}, |
||||
{0x0B56, 0x0B57}, {0x0B5C, 0x0B5D}, {0x0B5F, 0x0B63}, |
||||
{0x0B66, 0x0B77}, {0x0B82, 0x0B83}, {0x0B85, 0x0B8A}, |
||||
{0x0B8E, 0x0B90}, {0x0B92, 0x0B95}, {0x0B99, 0x0B9A}, |
||||
{0x0B9C, 0x0B9C}, {0x0B9E, 0x0B9F}, {0x0BA3, 0x0BA4}, |
||||
{0x0BA8, 0x0BAA}, {0x0BAE, 0x0BB9}, {0x0BBE, 0x0BC2}, |
||||
{0x0BC6, 0x0BC8}, {0x0BCA, 0x0BCD}, {0x0BD0, 0x0BD0}, |
||||
{0x0BD7, 0x0BD7}, {0x0BE6, 0x0BFA}, {0x0C00, 0x0C0C}, |
||||
{0x0C0E, 0x0C10}, {0x0C12, 0x0C28}, {0x0C2A, 0x0C39}, |
||||
{0x0C3D, 0x0C44}, {0x0C46, 0x0C48}, {0x0C4A, 0x0C4D}, |
||||
{0x0C55, 0x0C56}, {0x0C58, 0x0C5A}, {0x0C60, 0x0C63}, |
||||
{0x0C66, 0x0C6F}, {0x0C77, 0x0C8C}, {0x0C8E, 0x0C90}, |
||||
{0x0C92, 0x0CA8}, {0x0CAA, 0x0CB3}, {0x0CB5, 0x0CB9}, |
||||
{0x0CBC, 0x0CC4}, {0x0CC6, 0x0CC8}, {0x0CCA, 0x0CCD}, |
||||
{0x0CD5, 0x0CD6}, {0x0CDE, 0x0CDE}, {0x0CE0, 0x0CE3}, |
||||
{0x0CE6, 0x0CEF}, {0x0CF1, 0x0CF2}, {0x0D00, 0x0D03}, |
||||
{0x0D05, 0x0D0C}, {0x0D0E, 0x0D10}, {0x0D12, 0x0D44}, |
||||
{0x0D46, 0x0D48}, {0x0D4A, 0x0D4F}, {0x0D54, 0x0D63}, |
||||
{0x0D66, 0x0D7F}, {0x0D82, 0x0D83}, {0x0D85, 0x0D96}, |
||||
{0x0D9A, 0x0DB1}, {0x0DB3, 0x0DBB}, {0x0DBD, 0x0DBD}, |
||||
{0x0DC0, 0x0DC6}, {0x0DCA, 0x0DCA}, {0x0DCF, 0x0DD4}, |
||||
{0x0DD6, 0x0DD6}, {0x0DD8, 0x0DDF}, {0x0DE6, 0x0DEF}, |
||||
{0x0DF2, 0x0DF4}, {0x0E01, 0x0E3A}, {0x0E3F, 0x0E5B}, |
||||
{0x0E81, 0x0E82}, {0x0E84, 0x0E84}, {0x0E86, 0x0E8A}, |
||||
{0x0E8C, 0x0EA3}, {0x0EA5, 0x0EA5}, {0x0EA7, 0x0EBD}, |
||||
{0x0EC0, 0x0EC4}, {0x0EC6, 0x0EC6}, {0x0EC8, 0x0ECD}, |
||||
{0x0ED0, 0x0ED9}, {0x0EDC, 0x0EDF}, {0x0F00, 0x0F47}, |
||||
{0x0F49, 0x0F6C}, {0x0F71, 0x0F97}, {0x0F99, 0x0FBC}, |
||||
{0x0FBE, 0x0FCC}, {0x0FCE, 0x0FDA}, {0x1000, 0x10C5}, |
||||
{0x10C7, 0x10C7}, {0x10CD, 0x10CD}, {0x10D0, 0x10FF}, |
||||
{0x1160, 0x1248}, {0x124A, 0x124D}, {0x1250, 0x1256}, |
||||
{0x1258, 0x1258}, {0x125A, 0x125D}, {0x1260, 0x1288}, |
||||
{0x128A, 0x128D}, {0x1290, 0x12B0}, {0x12B2, 0x12B5}, |
||||
{0x12B8, 0x12BE}, {0x12C0, 0x12C0}, {0x12C2, 0x12C5}, |
||||
{0x12C8, 0x12D6}, {0x12D8, 0x1310}, {0x1312, 0x1315}, |
||||
{0x1318, 0x135A}, {0x135D, 0x137C}, {0x1380, 0x1399}, |
||||
{0x13A0, 0x13F5}, {0x13F8, 0x13FD}, {0x1400, 0x169C}, |
||||
{0x16A0, 0x16F8}, {0x1700, 0x170C}, {0x170E, 0x1714}, |
||||
{0x1720, 0x1736}, {0x1740, 0x1753}, {0x1760, 0x176C}, |
||||
{0x176E, 0x1770}, {0x1772, 0x1773}, {0x1780, 0x17DD}, |
||||
{0x17E0, 0x17E9}, {0x17F0, 0x17F9}, {0x1800, 0x180E}, |
||||
{0x1810, 0x1819}, {0x1820, 0x1878}, {0x1880, 0x18AA}, |
||||
{0x18B0, 0x18F5}, {0x1900, 0x191E}, {0x1920, 0x192B}, |
||||
{0x1930, 0x193B}, {0x1940, 0x1940}, {0x1944, 0x196D}, |
||||
{0x1970, 0x1974}, {0x1980, 0x19AB}, {0x19B0, 0x19C9}, |
||||
{0x19D0, 0x19DA}, {0x19DE, 0x1A1B}, {0x1A1E, 0x1A5E}, |
||||
{0x1A60, 0x1A7C}, {0x1A7F, 0x1A89}, {0x1A90, 0x1A99}, |
||||
{0x1AA0, 0x1AAD}, {0x1AB0, 0x1ABE}, {0x1B00, 0x1B4B}, |
||||
{0x1B50, 0x1B7C}, {0x1B80, 0x1BF3}, {0x1BFC, 0x1C37}, |
||||
{0x1C3B, 0x1C49}, {0x1C4D, 0x1C88}, {0x1C90, 0x1CBA}, |
||||
{0x1CBD, 0x1CC7}, {0x1CD0, 0x1CFA}, {0x1D00, 0x1DF9}, |
||||
{0x1DFB, 0x1F15}, {0x1F18, 0x1F1D}, {0x1F20, 0x1F45}, |
||||
{0x1F48, 0x1F4D}, {0x1F50, 0x1F57}, {0x1F59, 0x1F59}, |
||||
{0x1F5B, 0x1F5B}, {0x1F5D, 0x1F5D}, {0x1F5F, 0x1F7D}, |
||||
{0x1F80, 0x1FB4}, {0x1FB6, 0x1FC4}, {0x1FC6, 0x1FD3}, |
||||
{0x1FD6, 0x1FDB}, {0x1FDD, 0x1FEF}, {0x1FF2, 0x1FF4}, |
||||
{0x1FF6, 0x1FFE}, {0x2000, 0x200F}, {0x2011, 0x2012}, |
||||
{0x2017, 0x2017}, {0x201A, 0x201B}, {0x201E, 0x201F}, |
||||
{0x2023, 0x2023}, {0x2028, 0x202F}, {0x2031, 0x2031}, |
||||
{0x2034, 0x2034}, {0x2036, 0x203A}, {0x203C, 0x203D}, |
||||
{0x203F, 0x2064}, {0x2066, 0x2071}, {0x2075, 0x207E}, |
||||
{0x2080, 0x2080}, {0x2085, 0x208E}, {0x2090, 0x209C}, |
||||
{0x20A0, 0x20A8}, {0x20AA, 0x20AB}, {0x20AD, 0x20BF}, |
||||
{0x20D0, 0x20F0}, {0x2100, 0x2102}, {0x2104, 0x2104}, |
||||
{0x2106, 0x2108}, {0x210A, 0x2112}, {0x2114, 0x2115}, |
||||
{0x2117, 0x2120}, {0x2123, 0x2125}, {0x2127, 0x212A}, |
||||
{0x212C, 0x2152}, {0x2155, 0x215A}, {0x215F, 0x215F}, |
||||
{0x216C, 0x216F}, {0x217A, 0x2188}, {0x218A, 0x218B}, |
||||
{0x219A, 0x21B7}, {0x21BA, 0x21D1}, {0x21D3, 0x21D3}, |
||||
{0x21D5, 0x21E6}, {0x21E8, 0x21FF}, {0x2201, 0x2201}, |
||||
{0x2204, 0x2206}, {0x2209, 0x220A}, {0x220C, 0x220E}, |
||||
{0x2210, 0x2210}, {0x2212, 0x2214}, {0x2216, 0x2219}, |
||||
{0x221B, 0x221C}, {0x2221, 0x2222}, {0x2224, 0x2224}, |
||||
{0x2226, 0x2226}, {0x222D, 0x222D}, {0x222F, 0x2233}, |
||||
{0x2238, 0x223B}, {0x223E, 0x2247}, {0x2249, 0x224B}, |
||||
{0x224D, 0x2251}, {0x2253, 0x225F}, {0x2262, 0x2263}, |
||||
{0x2268, 0x2269}, {0x226C, 0x226D}, {0x2270, 0x2281}, |
||||
{0x2284, 0x2285}, {0x2288, 0x2294}, {0x2296, 0x2298}, |
||||
{0x229A, 0x22A4}, {0x22A6, 0x22BE}, {0x22C0, 0x2311}, |
||||
{0x2313, 0x2319}, {0x231C, 0x2328}, {0x232B, 0x23E8}, |
||||
{0x23ED, 0x23EF}, {0x23F1, 0x23F2}, {0x23F4, 0x2426}, |
||||
{0x2440, 0x244A}, {0x24EA, 0x24EA}, {0x254C, 0x254F}, |
||||
{0x2574, 0x257F}, {0x2590, 0x2591}, {0x2596, 0x259F}, |
||||
{0x25A2, 0x25A2}, {0x25AA, 0x25B1}, {0x25B4, 0x25B5}, |
||||
{0x25B8, 0x25BB}, {0x25BE, 0x25BF}, {0x25C2, 0x25C5}, |
||||
{0x25C9, 0x25CA}, {0x25CC, 0x25CD}, {0x25D2, 0x25E1}, |
||||
{0x25E6, 0x25EE}, {0x25F0, 0x25FC}, {0x25FF, 0x2604}, |
||||
{0x2607, 0x2608}, {0x260A, 0x260D}, {0x2610, 0x2613}, |
||||
{0x2616, 0x261B}, {0x261D, 0x261D}, {0x261F, 0x263F}, |
||||
{0x2641, 0x2641}, {0x2643, 0x2647}, {0x2654, 0x265F}, |
||||
{0x2662, 0x2662}, {0x2666, 0x2666}, {0x266B, 0x266B}, |
||||
{0x266E, 0x266E}, {0x2670, 0x267E}, {0x2680, 0x2692}, |
||||
{0x2694, 0x269D}, {0x26A0, 0x26A0}, {0x26A2, 0x26A9}, |
||||
{0x26AC, 0x26BC}, {0x26C0, 0x26C3}, {0x26E2, 0x26E2}, |
||||
{0x26E4, 0x26E7}, {0x2700, 0x2704}, {0x2706, 0x2709}, |
||||
{0x270C, 0x2727}, {0x2729, 0x273C}, {0x273E, 0x274B}, |
||||
{0x274D, 0x274D}, {0x274F, 0x2752}, {0x2756, 0x2756}, |
||||
{0x2758, 0x2775}, {0x2780, 0x2794}, {0x2798, 0x27AF}, |
||||
{0x27B1, 0x27BE}, {0x27C0, 0x27E5}, {0x27EE, 0x2984}, |
||||
{0x2987, 0x2B1A}, {0x2B1D, 0x2B4F}, {0x2B51, 0x2B54}, |
||||
{0x2B5A, 0x2B73}, {0x2B76, 0x2B95}, {0x2B98, 0x2C2E}, |
||||
{0x2C30, 0x2C5E}, {0x2C60, 0x2CF3}, {0x2CF9, 0x2D25}, |
||||
{0x2D27, 0x2D27}, {0x2D2D, 0x2D2D}, {0x2D30, 0x2D67}, |
||||
{0x2D6F, 0x2D70}, {0x2D7F, 0x2D96}, {0x2DA0, 0x2DA6}, |
||||
{0x2DA8, 0x2DAE}, {0x2DB0, 0x2DB6}, {0x2DB8, 0x2DBE}, |
||||
{0x2DC0, 0x2DC6}, {0x2DC8, 0x2DCE}, {0x2DD0, 0x2DD6}, |
||||
{0x2DD8, 0x2DDE}, {0x2DE0, 0x2E4F}, {0x303F, 0x303F}, |
||||
{0x4DC0, 0x4DFF}, {0xA4D0, 0xA62B}, {0xA640, 0xA6F7}, |
||||
{0xA700, 0xA7BF}, {0xA7C2, 0xA7C6}, {0xA7F7, 0xA82B}, |
||||
{0xA830, 0xA839}, {0xA840, 0xA877}, {0xA880, 0xA8C5}, |
||||
{0xA8CE, 0xA8D9}, {0xA8E0, 0xA953}, {0xA95F, 0xA95F}, |
||||
{0xA980, 0xA9CD}, {0xA9CF, 0xA9D9}, {0xA9DE, 0xA9FE}, |
||||
{0xAA00, 0xAA36}, {0xAA40, 0xAA4D}, {0xAA50, 0xAA59}, |
||||
{0xAA5C, 0xAAC2}, {0xAADB, 0xAAF6}, {0xAB01, 0xAB06}, |
||||
{0xAB09, 0xAB0E}, {0xAB11, 0xAB16}, {0xAB20, 0xAB26}, |
||||
{0xAB28, 0xAB2E}, {0xAB30, 0xAB67}, {0xAB70, 0xABED}, |
||||
{0xABF0, 0xABF9}, {0xD7B0, 0xD7C6}, {0xD7CB, 0xD7FB}, |
||||
{0xD800, 0xDFFF}, {0xFB00, 0xFB06}, {0xFB13, 0xFB17}, |
||||
{0xFB1D, 0xFB36}, {0xFB38, 0xFB3C}, {0xFB3E, 0xFB3E}, |
||||
{0xFB40, 0xFB41}, {0xFB43, 0xFB44}, {0xFB46, 0xFBC1}, |
||||
{0xFBD3, 0xFD3F}, {0xFD50, 0xFD8F}, {0xFD92, 0xFDC7}, |
||||
{0xFDF0, 0xFDFD}, {0xFE20, 0xFE2F}, {0xFE70, 0xFE74}, |
||||
{0xFE76, 0xFEFC}, {0xFEFF, 0xFEFF}, {0xFFF9, 0xFFFC}, |
||||
{0x10000, 0x1000B}, {0x1000D, 0x10026}, {0x10028, 0x1003A}, |
||||
{0x1003C, 0x1003D}, {0x1003F, 0x1004D}, {0x10050, 0x1005D}, |
||||
{0x10080, 0x100FA}, {0x10100, 0x10102}, {0x10107, 0x10133}, |
||||
{0x10137, 0x1018E}, {0x10190, 0x1019B}, {0x101A0, 0x101A0}, |
||||
{0x101D0, 0x101FD}, {0x10280, 0x1029C}, {0x102A0, 0x102D0}, |
||||
{0x102E0, 0x102FB}, {0x10300, 0x10323}, {0x1032D, 0x1034A}, |
||||
{0x10350, 0x1037A}, {0x10380, 0x1039D}, {0x1039F, 0x103C3}, |
||||
{0x103C8, 0x103D5}, {0x10400, 0x1049D}, {0x104A0, 0x104A9}, |
||||
{0x104B0, 0x104D3}, {0x104D8, 0x104FB}, {0x10500, 0x10527}, |
||||
{0x10530, 0x10563}, {0x1056F, 0x1056F}, {0x10600, 0x10736}, |
||||
{0x10740, 0x10755}, {0x10760, 0x10767}, {0x10800, 0x10805}, |
||||
{0x10808, 0x10808}, {0x1080A, 0x10835}, {0x10837, 0x10838}, |
||||
{0x1083C, 0x1083C}, {0x1083F, 0x10855}, {0x10857, 0x1089E}, |
||||
{0x108A7, 0x108AF}, {0x108E0, 0x108F2}, {0x108F4, 0x108F5}, |
||||
{0x108FB, 0x1091B}, {0x1091F, 0x10939}, {0x1093F, 0x1093F}, |
||||
{0x10980, 0x109B7}, {0x109BC, 0x109CF}, {0x109D2, 0x10A03}, |
||||
{0x10A05, 0x10A06}, {0x10A0C, 0x10A13}, {0x10A15, 0x10A17}, |
||||
{0x10A19, 0x10A35}, {0x10A38, 0x10A3A}, {0x10A3F, 0x10A48}, |
||||
{0x10A50, 0x10A58}, {0x10A60, 0x10A9F}, {0x10AC0, 0x10AE6}, |
||||
{0x10AEB, 0x10AF6}, {0x10B00, 0x10B35}, {0x10B39, 0x10B55}, |
||||
{0x10B58, 0x10B72}, {0x10B78, 0x10B91}, {0x10B99, 0x10B9C}, |
||||
{0x10BA9, 0x10BAF}, {0x10C00, 0x10C48}, {0x10C80, 0x10CB2}, |
||||
{0x10CC0, 0x10CF2}, {0x10CFA, 0x10D27}, {0x10D30, 0x10D39}, |
||||
{0x10E60, 0x10E7E}, {0x10F00, 0x10F27}, {0x10F30, 0x10F59}, |
||||
{0x10FE0, 0x10FF6}, {0x11000, 0x1104D}, {0x11052, 0x1106F}, |
||||
{0x1107F, 0x110C1}, {0x110CD, 0x110CD}, {0x110D0, 0x110E8}, |
||||
{0x110F0, 0x110F9}, {0x11100, 0x11134}, {0x11136, 0x11146}, |
||||
{0x11150, 0x11176}, {0x11180, 0x111CD}, {0x111D0, 0x111DF}, |
||||
{0x111E1, 0x111F4}, {0x11200, 0x11211}, {0x11213, 0x1123E}, |
||||
{0x11280, 0x11286}, {0x11288, 0x11288}, {0x1128A, 0x1128D}, |
||||
{0x1128F, 0x1129D}, {0x1129F, 0x112A9}, {0x112B0, 0x112EA}, |
||||
{0x112F0, 0x112F9}, {0x11300, 0x11303}, {0x11305, 0x1130C}, |
||||
{0x1130F, 0x11310}, {0x11313, 0x11328}, {0x1132A, 0x11330}, |
||||
{0x11332, 0x11333}, {0x11335, 0x11339}, {0x1133B, 0x11344}, |
||||
{0x11347, 0x11348}, {0x1134B, 0x1134D}, {0x11350, 0x11350}, |
||||
{0x11357, 0x11357}, {0x1135D, 0x11363}, {0x11366, 0x1136C}, |
||||
{0x11370, 0x11374}, {0x11400, 0x11459}, {0x1145B, 0x1145B}, |
||||
{0x1145D, 0x1145F}, {0x11480, 0x114C7}, {0x114D0, 0x114D9}, |
||||
{0x11580, 0x115B5}, {0x115B8, 0x115DD}, {0x11600, 0x11644}, |
||||
{0x11650, 0x11659}, {0x11660, 0x1166C}, {0x11680, 0x116B8}, |
||||
{0x116C0, 0x116C9}, {0x11700, 0x1171A}, {0x1171D, 0x1172B}, |
||||
{0x11730, 0x1173F}, {0x11800, 0x1183B}, {0x118A0, 0x118F2}, |
||||
{0x118FF, 0x118FF}, {0x119A0, 0x119A7}, {0x119AA, 0x119D7}, |
||||
{0x119DA, 0x119E4}, {0x11A00, 0x11A47}, {0x11A50, 0x11AA2}, |
||||
{0x11AC0, 0x11AF8}, {0x11C00, 0x11C08}, {0x11C0A, 0x11C36}, |
||||
{0x11C38, 0x11C45}, {0x11C50, 0x11C6C}, {0x11C70, 0x11C8F}, |
||||
{0x11C92, 0x11CA7}, {0x11CA9, 0x11CB6}, {0x11D00, 0x11D06}, |
||||
{0x11D08, 0x11D09}, {0x11D0B, 0x11D36}, {0x11D3A, 0x11D3A}, |
||||
{0x11D3C, 0x11D3D}, {0x11D3F, 0x11D47}, {0x11D50, 0x11D59}, |
||||
{0x11D60, 0x11D65}, {0x11D67, 0x11D68}, {0x11D6A, 0x11D8E}, |
||||
{0x11D90, 0x11D91}, {0x11D93, 0x11D98}, {0x11DA0, 0x11DA9}, |
||||
{0x11EE0, 0x11EF8}, {0x11FC0, 0x11FF1}, {0x11FFF, 0x12399}, |
||||
{0x12400, 0x1246E}, {0x12470, 0x12474}, {0x12480, 0x12543}, |
||||
{0x13000, 0x1342E}, {0x13430, 0x13438}, {0x14400, 0x14646}, |
||||
{0x16800, 0x16A38}, {0x16A40, 0x16A5E}, {0x16A60, 0x16A69}, |
||||
{0x16A6E, 0x16A6F}, {0x16AD0, 0x16AED}, {0x16AF0, 0x16AF5}, |
||||
{0x16B00, 0x16B45}, {0x16B50, 0x16B59}, {0x16B5B, 0x16B61}, |
||||
{0x16B63, 0x16B77}, {0x16B7D, 0x16B8F}, {0x16E40, 0x16E9A}, |
||||
{0x16F00, 0x16F4A}, {0x16F4F, 0x16F87}, {0x16F8F, 0x16F9F}, |
||||
{0x1BC00, 0x1BC6A}, {0x1BC70, 0x1BC7C}, {0x1BC80, 0x1BC88}, |
||||
{0x1BC90, 0x1BC99}, {0x1BC9C, 0x1BCA3}, {0x1D000, 0x1D0F5}, |
||||
{0x1D100, 0x1D126}, {0x1D129, 0x1D1E8}, {0x1D200, 0x1D245}, |
||||
{0x1D2E0, 0x1D2F3}, {0x1D300, 0x1D356}, {0x1D360, 0x1D378}, |
||||
{0x1D400, 0x1D454}, {0x1D456, 0x1D49C}, {0x1D49E, 0x1D49F}, |
||||
{0x1D4A2, 0x1D4A2}, {0x1D4A5, 0x1D4A6}, {0x1D4A9, 0x1D4AC}, |
||||
{0x1D4AE, 0x1D4B9}, {0x1D4BB, 0x1D4BB}, {0x1D4BD, 0x1D4C3}, |
||||
{0x1D4C5, 0x1D505}, {0x1D507, 0x1D50A}, {0x1D50D, 0x1D514}, |
||||
{0x1D516, 0x1D51C}, {0x1D51E, 0x1D539}, {0x1D53B, 0x1D53E}, |
||||
{0x1D540, 0x1D544}, {0x1D546, 0x1D546}, {0x1D54A, 0x1D550}, |
||||
{0x1D552, 0x1D6A5}, {0x1D6A8, 0x1D7CB}, {0x1D7CE, 0x1DA8B}, |
||||
{0x1DA9B, 0x1DA9F}, {0x1DAA1, 0x1DAAF}, {0x1E000, 0x1E006}, |
||||
{0x1E008, 0x1E018}, {0x1E01B, 0x1E021}, {0x1E023, 0x1E024}, |
||||
{0x1E026, 0x1E02A}, {0x1E100, 0x1E12C}, {0x1E130, 0x1E13D}, |
||||
{0x1E140, 0x1E149}, {0x1E14E, 0x1E14F}, {0x1E2C0, 0x1E2F9}, |
||||
{0x1E2FF, 0x1E2FF}, {0x1E800, 0x1E8C4}, {0x1E8C7, 0x1E8D6}, |
||||
{0x1E900, 0x1E94B}, {0x1E950, 0x1E959}, {0x1E95E, 0x1E95F}, |
||||
{0x1EC71, 0x1ECB4}, {0x1ED01, 0x1ED3D}, {0x1EE00, 0x1EE03}, |
||||
{0x1EE05, 0x1EE1F}, {0x1EE21, 0x1EE22}, {0x1EE24, 0x1EE24}, |
||||
{0x1EE27, 0x1EE27}, {0x1EE29, 0x1EE32}, {0x1EE34, 0x1EE37}, |
||||
{0x1EE39, 0x1EE39}, {0x1EE3B, 0x1EE3B}, {0x1EE42, 0x1EE42}, |
||||
{0x1EE47, 0x1EE47}, {0x1EE49, 0x1EE49}, {0x1EE4B, 0x1EE4B}, |
||||
{0x1EE4D, 0x1EE4F}, {0x1EE51, 0x1EE52}, {0x1EE54, 0x1EE54}, |
||||
{0x1EE57, 0x1EE57}, {0x1EE59, 0x1EE59}, {0x1EE5B, 0x1EE5B}, |
||||
{0x1EE5D, 0x1EE5D}, {0x1EE5F, 0x1EE5F}, {0x1EE61, 0x1EE62}, |
||||
{0x1EE64, 0x1EE64}, {0x1EE67, 0x1EE6A}, {0x1EE6C, 0x1EE72}, |
||||
{0x1EE74, 0x1EE77}, {0x1EE79, 0x1EE7C}, {0x1EE7E, 0x1EE7E}, |
||||
{0x1EE80, 0x1EE89}, {0x1EE8B, 0x1EE9B}, {0x1EEA1, 0x1EEA3}, |
||||
{0x1EEA5, 0x1EEA9}, {0x1EEAB, 0x1EEBB}, {0x1EEF0, 0x1EEF1}, |
||||
{0x1F000, 0x1F003}, {0x1F005, 0x1F02B}, {0x1F030, 0x1F093}, |
||||
{0x1F0A0, 0x1F0AE}, {0x1F0B1, 0x1F0BF}, {0x1F0C1, 0x1F0CE}, |
||||
{0x1F0D1, 0x1F0F5}, {0x1F10B, 0x1F10C}, {0x1F12E, 0x1F12F}, |
||||
{0x1F16A, 0x1F16C}, {0x1F1E6, 0x1F1FF}, {0x1F321, 0x1F32C}, |
||||
{0x1F336, 0x1F336}, {0x1F37D, 0x1F37D}, {0x1F394, 0x1F39F}, |
||||
{0x1F3CB, 0x1F3CE}, {0x1F3D4, 0x1F3DF}, {0x1F3F1, 0x1F3F3}, |
||||
{0x1F3F5, 0x1F3F7}, {0x1F43F, 0x1F43F}, {0x1F441, 0x1F441}, |
||||
{0x1F4FD, 0x1F4FE}, {0x1F53E, 0x1F54A}, {0x1F54F, 0x1F54F}, |
||||
{0x1F568, 0x1F579}, {0x1F57B, 0x1F594}, {0x1F597, 0x1F5A3}, |
||||
{0x1F5A5, 0x1F5FA}, {0x1F650, 0x1F67F}, {0x1F6C6, 0x1F6CB}, |
||||
{0x1F6CD, 0x1F6CF}, {0x1F6D3, 0x1F6D4}, {0x1F6E0, 0x1F6EA}, |
||||
{0x1F6F0, 0x1F6F3}, {0x1F700, 0x1F773}, {0x1F780, 0x1F7D8}, |
||||
{0x1F800, 0x1F80B}, {0x1F810, 0x1F847}, {0x1F850, 0x1F859}, |
||||
{0x1F860, 0x1F887}, {0x1F890, 0x1F8AD}, {0x1F900, 0x1F90B}, |
||||
{0x1FA00, 0x1FA53}, {0x1FA60, 0x1FA6D}, {0xE0001, 0xE0001}, |
||||
{0xE0020, 0xE007F}, |
||||
} |
||||
|
||||
var emoji = table{ |
||||
{0x203C, 0x203C}, {0x2049, 0x2049}, {0x2122, 0x2122}, |
||||
{0x2139, 0x2139}, {0x2194, 0x2199}, {0x21A9, 0x21AA}, |
||||
{0x231A, 0x231B}, {0x2328, 0x2328}, {0x2388, 0x2388}, |
||||
{0x23CF, 0x23CF}, {0x23E9, 0x23F3}, {0x23F8, 0x23FA}, |
||||
{0x24C2, 0x24C2}, {0x25AA, 0x25AB}, {0x25B6, 0x25B6}, |
||||
{0x25C0, 0x25C0}, {0x25FB, 0x25FE}, {0x2600, 0x2605}, |
||||
{0x2607, 0x2612}, {0x2614, 0x2685}, {0x2690, 0x2705}, |
||||
{0x2708, 0x2712}, {0x2714, 0x2714}, {0x2716, 0x2716}, |
||||
{0x271D, 0x271D}, {0x2721, 0x2721}, {0x2728, 0x2728}, |
||||
{0x2733, 0x2734}, {0x2744, 0x2744}, {0x2747, 0x2747}, |
||||
{0x274C, 0x274C}, {0x274E, 0x274E}, {0x2753, 0x2755}, |
||||
{0x2757, 0x2757}, {0x2763, 0x2767}, {0x2795, 0x2797}, |
||||
{0x27A1, 0x27A1}, {0x27B0, 0x27B0}, {0x27BF, 0x27BF}, |
||||
{0x2934, 0x2935}, {0x2B05, 0x2B07}, {0x2B1B, 0x2B1C}, |
||||
{0x2B50, 0x2B50}, {0x2B55, 0x2B55}, {0x3030, 0x3030}, |
||||
{0x303D, 0x303D}, {0x3297, 0x3297}, {0x3299, 0x3299}, |
||||
{0x1F000, 0x1F0FF}, {0x1F10D, 0x1F10F}, {0x1F12F, 0x1F12F}, |
||||
{0x1F16C, 0x1F171}, {0x1F17E, 0x1F17F}, {0x1F18E, 0x1F18E}, |
||||
{0x1F191, 0x1F19A}, {0x1F1AD, 0x1F1E5}, {0x1F201, 0x1F20F}, |
||||
{0x1F21A, 0x1F21A}, {0x1F22F, 0x1F22F}, {0x1F232, 0x1F23A}, |
||||
{0x1F23C, 0x1F23F}, {0x1F249, 0x1F3FA}, {0x1F400, 0x1F53D}, |
||||
{0x1F546, 0x1F64F}, {0x1F680, 0x1F6FF}, {0x1F774, 0x1F77F}, |
||||
{0x1F7D5, 0x1F7FF}, {0x1F80C, 0x1F80F}, {0x1F848, 0x1F84F}, |
||||
{0x1F85A, 0x1F85F}, {0x1F888, 0x1F88F}, {0x1F8AE, 0x1F8FF}, |
||||
{0x1F90C, 0x1F93A}, {0x1F93C, 0x1F945}, {0x1F947, 0x1FFFD}, |
||||
} |
||||
@ -0,0 +1,28 @@ |
||||
// +build windows
|
||||
// +build !appengine
|
||||
|
||||
package runewidth |
||||
|
||||
import ( |
||||
"syscall" |
||||
) |
||||
|
||||
var ( |
||||
kernel32 = syscall.NewLazyDLL("kernel32") |
||||
procGetConsoleOutputCP = kernel32.NewProc("GetConsoleOutputCP") |
||||
) |
||||
|
||||
// IsEastAsian return true if the current locale is CJK
|
||||
func IsEastAsian() bool { |
||||
r1, _, _ := procGetConsoleOutputCP.Call() |
||||
if r1 == 0 { |
||||
return false |
||||
} |
||||
|
||||
switch int(r1) { |
||||
case 932, 51932, 936, 949, 950: |
||||
return true |
||||
} |
||||
|
||||
return false |
||||
} |
||||
@ -0,0 +1,2 @@ |
||||
language: go |
||||
go: master |
||||
@ -0,0 +1,21 @@ |
||||
MIT License |
||||
|
||||
Copyright (c) 2018 Minko Gechev |
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy |
||||
of this software and associated documentation files (the "Software"), to deal |
||||
in the Software without restriction, including without limitation the rights |
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
||||
copies of the Software, and to permit persons to whom the Software is |
||||
furnished to do so, subject to the following conditions: |
||||
|
||||
The above copyright notice and this permission notice shall be included in all |
||||
copies or substantial portions of the Software. |
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
||||
SOFTWARE. |
||||
@ -0,0 +1,456 @@ |
||||
package dots |
||||
|
||||
import ( |
||||
"go/build" |
||||
"log" |
||||
"os" |
||||
"path" |
||||
"path/filepath" |
||||
"regexp" |
||||
"runtime" |
||||
"strings" |
||||
) |
||||
|
||||
var ( |
||||
buildContext = build.Default |
||||
goroot = filepath.Clean(runtime.GOROOT()) |
||||
gorootSrc = filepath.Join(goroot, "src") |
||||
) |
||||
|
||||
func flatten(arr [][]string) []string { |
||||
var res []string |
||||
for _, e := range arr { |
||||
res = append(res, e...) |
||||
} |
||||
return res |
||||
} |
||||
|
||||
// Resolve accepts a slice of paths with optional "..." placeholder and a slice with paths to be skipped.
|
||||
// The final result is the set of all files from the selected directories subtracted with
|
||||
// the files in the skip slice.
|
||||
func Resolve(includePatterns, skipPatterns []string) ([]string, error) { |
||||
skip, err := resolvePatterns(skipPatterns) |
||||
filter := newPathFilter(flatten(skip)) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
pathSet := map[string]bool{} |
||||
includePackages, err := resolvePatterns(includePatterns) |
||||
include := flatten(includePackages) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
var result []string |
||||
for _, i := range include { |
||||
if _, ok := pathSet[i]; !ok && !filter(i) { |
||||
pathSet[i] = true |
||||
result = append(result, i) |
||||
} |
||||
} |
||||
return result, err |
||||
} |
||||
|
||||
// ResolvePackages accepts a slice of paths with optional "..." placeholder and a slice with paths to be skipped.
|
||||
// The final result is the set of all files from the selected directories subtracted with
|
||||
// the files in the skip slice. The difference between `Resolve` and `ResolvePackages`
|
||||
// is that `ResolvePackages` preserves the package structure in the nested slices.
|
||||
func ResolvePackages(includePatterns, skipPatterns []string) ([][]string, error) { |
||||
skip, err := resolvePatterns(skipPatterns) |
||||
filter := newPathFilter(flatten(skip)) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
pathSet := map[string]bool{} |
||||
include, err := resolvePatterns(includePatterns) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
var result [][]string |
||||
for _, p := range include { |
||||
var packageFiles []string |
||||
for _, f := range p { |
||||
if _, ok := pathSet[f]; !ok && !filter(f) { |
||||
pathSet[f] = true |
||||
packageFiles = append(packageFiles, f) |
||||
} |
||||
} |
||||
result = append(result, packageFiles) |
||||
} |
||||
return result, err |
||||
} |
||||
|
||||
func isDir(filename string) bool { |
||||
fi, err := os.Stat(filename) |
||||
return err == nil && fi.IsDir() |
||||
} |
||||
|
||||
func exists(filename string) bool { |
||||
_, err := os.Stat(filename) |
||||
return err == nil |
||||
} |
||||
|
||||
func resolveDir(dirname string) ([]string, error) { |
||||
pkg, err := build.ImportDir(dirname, 0) |
||||
return resolveImportedPackage(pkg, err) |
||||
} |
||||
|
||||
func resolvePackage(pkgname string) ([]string, error) { |
||||
pkg, err := build.Import(pkgname, ".", 0) |
||||
return resolveImportedPackage(pkg, err) |
||||
} |
||||
|
||||
func resolveImportedPackage(pkg *build.Package, err error) ([]string, error) { |
||||
if err != nil { |
||||
if _, nogo := err.(*build.NoGoError); nogo { |
||||
// Don't complain if the failure is due to no Go source files.
|
||||
return nil, nil |
||||
} |
||||
return nil, err |
||||
} |
||||
|
||||
var files []string |
||||
files = append(files, pkg.GoFiles...) |
||||
files = append(files, pkg.CgoFiles...) |
||||
files = append(files, pkg.TestGoFiles...) |
||||
if pkg.Dir != "." { |
||||
for i, f := range files { |
||||
files[i] = filepath.Join(pkg.Dir, f) |
||||
} |
||||
} |
||||
return files, nil |
||||
} |
||||
|
||||
func resolvePatterns(patterns []string) ([][]string, error) { |
||||
var files [][]string |
||||
for _, pattern := range patterns { |
||||
f, err := resolvePattern(pattern) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
files = append(files, f...) |
||||
} |
||||
return files, nil |
||||
} |
||||
|
||||
func resolvePattern(pattern string) ([][]string, error) { |
||||
// dirsRun, filesRun, and pkgsRun indicate whether golint is applied to
|
||||
// directory, file or package targets. The distinction affects which
|
||||
// checks are run. It is no valid to mix target types.
|
||||
var dirsRun, filesRun, pkgsRun int |
||||
var matches []string |
||||
|
||||
if strings.HasSuffix(pattern, "/...") && isDir(pattern[:len(pattern)-len("/...")]) { |
||||
dirsRun = 1 |
||||
for _, dirname := range matchPackagesInFS(pattern) { |
||||
matches = append(matches, dirname) |
||||
} |
||||
} else if isDir(pattern) { |
||||
dirsRun = 1 |
||||
matches = append(matches, pattern) |
||||
} else if exists(pattern) { |
||||
filesRun = 1 |
||||
matches = append(matches, pattern) |
||||
} else { |
||||
pkgsRun = 1 |
||||
matches = append(matches, pattern) |
||||
} |
||||
|
||||
result := [][]string{} |
||||
switch { |
||||
case dirsRun == 1: |
||||
for _, dir := range matches { |
||||
res, err := resolveDir(dir) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
result = append(result, res) |
||||
} |
||||
case filesRun == 1: |
||||
return [][]string{matches}, nil |
||||
case pkgsRun == 1: |
||||
for _, pkg := range importPaths(matches) { |
||||
res, err := resolvePackage(pkg) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
result = append(result, res) |
||||
} |
||||
} |
||||
return result, nil |
||||
} |
||||
|
||||
func newPathFilter(skip []string) func(string) bool { |
||||
filter := map[string]bool{} |
||||
for _, name := range skip { |
||||
filter[name] = true |
||||
} |
||||
|
||||
return func(path string) bool { |
||||
base := filepath.Base(path) |
||||
if filter[base] || filter[path] { |
||||
return true |
||||
} |
||||
return base != "." && base != ".." && strings.ContainsAny(base[0:1], "_.") |
||||
} |
||||
} |
||||
|
||||
// importPathsNoDotExpansion returns the import paths to use for the given
|
||||
// command line, but it does no ... expansion.
|
||||
func importPathsNoDotExpansion(args []string) []string { |
||||
if len(args) == 0 { |
||||
return []string{"."} |
||||
} |
||||
var out []string |
||||
for _, a := range args { |
||||
// Arguments are supposed to be import paths, but
|
||||
// as a courtesy to Windows developers, rewrite \ to /
|
||||
// in command-line arguments. Handles .\... and so on.
|
||||
if filepath.Separator == '\\' { |
||||
a = strings.Replace(a, `\`, `/`, -1) |
||||
} |
||||
|
||||
// Put argument in canonical form, but preserve leading ./.
|
||||
if strings.HasPrefix(a, "./") { |
||||
a = "./" + path.Clean(a) |
||||
if a == "./." { |
||||
a = "." |
||||
} |
||||
} else { |
||||
a = path.Clean(a) |
||||
} |
||||
if a == "all" || a == "std" { |
||||
out = append(out, matchPackages(a)...) |
||||
continue |
||||
} |
||||
out = append(out, a) |
||||
} |
||||
return out |
||||
} |
||||
|
||||
// importPaths returns the import paths to use for the given command line.
|
||||
func importPaths(args []string) []string { |
||||
args = importPathsNoDotExpansion(args) |
||||
var out []string |
||||
for _, a := range args { |
||||
if strings.Contains(a, "...") { |
||||
if build.IsLocalImport(a) { |
||||
out = append(out, matchPackagesInFS(a)...) |
||||
} else { |
||||
out = append(out, matchPackages(a)...) |
||||
} |
||||
continue |
||||
} |
||||
out = append(out, a) |
||||
} |
||||
return out |
||||
} |
||||
|
||||
// matchPattern(pattern)(name) reports whether
|
||||
// name matches pattern. Pattern is a limited glob
|
||||
// pattern in which '...' means 'any string' and there
|
||||
// is no other special syntax.
|
||||
func matchPattern(pattern string) func(name string) bool { |
||||
re := regexp.QuoteMeta(pattern) |
||||
re = strings.Replace(re, `\.\.\.`, `.*`, -1) |
||||
// Special case: foo/... matches foo too.
|
||||
if strings.HasSuffix(re, `/.*`) { |
||||
re = re[:len(re)-len(`/.*`)] + `(/.*)?` |
||||
} |
||||
reg := regexp.MustCompile(`^` + re + `$`) |
||||
return func(name string) bool { |
||||
return reg.MatchString(name) |
||||
} |
||||
} |
||||
|
||||
// hasPathPrefix reports whether the path s begins with the
|
||||
// elements in prefix.
|
||||
func hasPathPrefix(s, prefix string) bool { |
||||
switch { |
||||
default: |
||||
return false |
||||
case len(s) == len(prefix): |
||||
return s == prefix |
||||
case len(s) > len(prefix): |
||||
if prefix != "" && prefix[len(prefix)-1] == '/' { |
||||
return strings.HasPrefix(s, prefix) |
||||
} |
||||
return s[len(prefix)] == '/' && s[:len(prefix)] == prefix |
||||
} |
||||
} |
||||
|
||||
// treeCanMatchPattern(pattern)(name) reports whether
|
||||
// name or children of name can possibly match pattern.
|
||||
// Pattern is the same limited glob accepted by matchPattern.
|
||||
func treeCanMatchPattern(pattern string) func(name string) bool { |
||||
wildCard := false |
||||
if i := strings.Index(pattern, "..."); i >= 0 { |
||||
wildCard = true |
||||
pattern = pattern[:i] |
||||
} |
||||
return func(name string) bool { |
||||
return len(name) <= len(pattern) && hasPathPrefix(pattern, name) || |
||||
wildCard && strings.HasPrefix(name, pattern) |
||||
} |
||||
} |
||||
|
||||
func matchPackages(pattern string) []string { |
||||
match := func(string) bool { return true } |
||||
treeCanMatch := func(string) bool { return true } |
||||
if pattern != "all" && pattern != "std" { |
||||
match = matchPattern(pattern) |
||||
treeCanMatch = treeCanMatchPattern(pattern) |
||||
} |
||||
|
||||
have := map[string]bool{ |
||||
"builtin": true, // ignore pseudo-package that exists only for documentation
|
||||
} |
||||
if !buildContext.CgoEnabled { |
||||
have["runtime/cgo"] = true // ignore during walk
|
||||
} |
||||
var pkgs []string |
||||
|
||||
// Commands
|
||||
cmd := filepath.Join(goroot, "src/cmd") + string(filepath.Separator) |
||||
filepath.Walk(cmd, func(path string, fi os.FileInfo, err error) error { |
||||
if err != nil || !fi.IsDir() || path == cmd { |
||||
return nil |
||||
} |
||||
name := path[len(cmd):] |
||||
if !treeCanMatch(name) { |
||||
return filepath.SkipDir |
||||
} |
||||
// Commands are all in cmd/, not in subdirectories.
|
||||
if strings.Contains(name, string(filepath.Separator)) { |
||||
return filepath.SkipDir |
||||
} |
||||
|
||||
// We use, e.g., cmd/gofmt as the pseudo import path for gofmt.
|
||||
name = "cmd/" + name |
||||
if have[name] { |
||||
return nil |
||||
} |
||||
have[name] = true |
||||
if !match(name) { |
||||
return nil |
||||
} |
||||
_, err = buildContext.ImportDir(path, 0) |
||||
if err != nil { |
||||
if _, noGo := err.(*build.NoGoError); !noGo { |
||||
log.Print(err) |
||||
} |
||||
return nil |
||||
} |
||||
pkgs = append(pkgs, name) |
||||
return nil |
||||
}) |
||||
|
||||
for _, src := range buildContext.SrcDirs() { |
||||
if (pattern == "std" || pattern == "cmd") && src != gorootSrc { |
||||
continue |
||||
} |
||||
src = filepath.Clean(src) + string(filepath.Separator) |
||||
root := src |
||||
if pattern == "cmd" { |
||||
root += "cmd" + string(filepath.Separator) |
||||
} |
||||
filepath.Walk(root, func(path string, fi os.FileInfo, err error) error { |
||||
if err != nil || !fi.IsDir() || path == src { |
||||
return nil |
||||
} |
||||
|
||||
// Avoid .foo, _foo, and testdata directory trees.
|
||||
_, elem := filepath.Split(path) |
||||
if strings.HasPrefix(elem, ".") || strings.HasPrefix(elem, "_") || elem == "testdata" { |
||||
return filepath.SkipDir |
||||
} |
||||
|
||||
name := filepath.ToSlash(path[len(src):]) |
||||
if pattern == "std" && (strings.Contains(name, ".") || name == "cmd") { |
||||
// The name "std" is only the standard library.
|
||||
// If the name is cmd, it's the root of the command tree.
|
||||
return filepath.SkipDir |
||||
} |
||||
if !treeCanMatch(name) { |
||||
return filepath.SkipDir |
||||
} |
||||
if have[name] { |
||||
return nil |
||||
} |
||||
have[name] = true |
||||
if !match(name) { |
||||
return nil |
||||
} |
||||
_, err = buildContext.ImportDir(path, 0) |
||||
if err != nil { |
||||
if _, noGo := err.(*build.NoGoError); noGo { |
||||
return nil |
||||
} |
||||
} |
||||
pkgs = append(pkgs, name) |
||||
return nil |
||||
}) |
||||
} |
||||
return pkgs |
||||
} |
||||
|
||||
func matchPackagesInFS(pattern string) []string { |
||||
// Find directory to begin the scan.
|
||||
// Could be smarter but this one optimization
|
||||
// is enough for now, since ... is usually at the
|
||||
// end of a path.
|
||||
i := strings.Index(pattern, "...") |
||||
dir, _ := path.Split(pattern[:i]) |
||||
|
||||
// pattern begins with ./ or ../.
|
||||
// path.Clean will discard the ./ but not the ../.
|
||||
// We need to preserve the ./ for pattern matching
|
||||
// and in the returned import paths.
|
||||
prefix := "" |
||||
if strings.HasPrefix(pattern, "./") { |
||||
prefix = "./" |
||||
} |
||||
match := matchPattern(pattern) |
||||
|
||||
var pkgs []string |
||||
filepath.Walk(dir, func(path string, fi os.FileInfo, err error) error { |
||||
if err != nil || !fi.IsDir() { |
||||
return nil |
||||
} |
||||
if path == dir { |
||||
// filepath.Walk starts at dir and recurses. For the recursive case,
|
||||
// the path is the result of filepath.Join, which calls filepath.Clean.
|
||||
// The initial case is not Cleaned, though, so we do this explicitly.
|
||||
//
|
||||
// This converts a path like "./io/" to "io". Without this step, running
|
||||
// "cd $GOROOT/src/pkg; go list ./io/..." would incorrectly skip the io
|
||||
// package, because prepending the prefix "./" to the unclean path would
|
||||
// result in "././io", and match("././io") returns false.
|
||||
path = filepath.Clean(path) |
||||
} |
||||
|
||||
// Avoid .foo, _foo, and testdata directory trees, but do not avoid "." or "..".
|
||||
_, elem := filepath.Split(path) |
||||
dot := strings.HasPrefix(elem, ".") && elem != "." && elem != ".." |
||||
if dot || strings.HasPrefix(elem, "_") || elem == "testdata" { |
||||
return filepath.SkipDir |
||||
} |
||||
|
||||
name := prefix + filepath.ToSlash(path) |
||||
if !match(name) { |
||||
return nil |
||||
} |
||||
if _, err = build.ImportDir(path, 0); err != nil { |
||||
if _, noGo := err.(*build.NoGoError); !noGo { |
||||
log.Print(err) |
||||
} |
||||
return nil |
||||
} |
||||
pkgs = append(pkgs, name) |
||||
return nil |
||||
}) |
||||
return pkgs |
||||
} |
||||
@ -0,0 +1,21 @@ |
||||
MIT License |
||||
|
||||
Copyright (c) 2018 Minko Gechev |
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy |
||||
of this software and associated documentation files (the "Software"), to deal |
||||
in the Software without restriction, including without limitation the rights |
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
||||
copies of the Software, and to permit persons to whom the Software is |
||||
furnished to do so, subject to the following conditions: |
||||
|
||||
The above copyright notice and this permission notice shall be included in all |
||||
copies or substantial portions of the Software. |
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
||||
SOFTWARE. |
||||
@ -0,0 +1,76 @@ |
||||
package formatter |
||||
|
||||
import ( |
||||
"bytes" |
||||
"encoding/xml" |
||||
"github.com/mgechev/revive/lint" |
||||
plainTemplate "text/template" |
||||
) |
||||
|
||||
// Checkstyle is an implementation of the Formatter interface
|
||||
// which formats the errors to Checkstyle-like format.
|
||||
type Checkstyle struct { |
||||
Metadata lint.FormatterMetadata |
||||
} |
||||
|
||||
// Name returns the name of the formatter
|
||||
func (f *Checkstyle) Name() string { |
||||
return "checkstyle" |
||||
} |
||||
|
||||
type issue struct { |
||||
Line int |
||||
Col int |
||||
What string |
||||
Confidence float64 |
||||
Severity lint.Severity |
||||
RuleName string |
||||
} |
||||
|
||||
// Format formats the failures gotten from the lint.
|
||||
func (f *Checkstyle) Format(failures <-chan lint.Failure, config lint.Config) (string, error) { |
||||
var issues = map[string][]issue{} |
||||
for failure := range failures { |
||||
buf := new(bytes.Buffer) |
||||
xml.Escape(buf, []byte(failure.Failure)) |
||||
what := buf.String() |
||||
iss := issue{ |
||||
Line: failure.Position.Start.Line, |
||||
Col: failure.Position.Start.Column, |
||||
What: what, |
||||
Confidence: failure.Confidence, |
||||
Severity: severity(config, failure), |
||||
RuleName: failure.RuleName, |
||||
} |
||||
fn := failure.GetFilename() |
||||
if issues[fn] == nil { |
||||
issues[fn] = make([]issue, 0) |
||||
} |
||||
issues[fn] = append(issues[fn], iss) |
||||
} |
||||
|
||||
t, err := plainTemplate.New("revive").Parse(checkstyleTemplate) |
||||
if err != nil { |
||||
return "", err |
||||
} |
||||
|
||||
buf := new(bytes.Buffer) |
||||
|
||||
err = t.Execute(buf, issues) |
||||
if err != nil { |
||||
return "", err |
||||
} |
||||
|
||||
return buf.String(), nil |
||||
} |
||||
|
||||
const checkstyleTemplate = `<?xml version='1.0' encoding='UTF-8'?> |
||||
<checkstyle version="5.0"> |
||||
{{- range $k, $v := . }} |
||||
<file name="{{ $k }}"> |
||||
{{- range $i, $issue := $v }} |
||||
<error line="{{ $issue.Line }}" column="{{ $issue.Col }}" message="{{ $issue.What }} (confidence {{ $issue.Confidence}})" severity="{{ $issue.Severity }}" source="revive/{{ $issue.RuleName }}"/> |
||||
{{- end }} |
||||
</file> |
||||
{{- end }} |
||||
</checkstyle>` |
||||
@ -0,0 +1,26 @@ |
||||
package formatter |
||||
|
||||
import ( |
||||
"fmt" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// Default is an implementation of the Formatter interface
|
||||
// which formats the errors to text.
|
||||
type Default struct { |
||||
Metadata lint.FormatterMetadata |
||||
} |
||||
|
||||
// Name returns the name of the formatter
|
||||
func (f *Default) Name() string { |
||||
return "default" |
||||
} |
||||
|
||||
// Format formats the failures gotten from the lint.
|
||||
func (f *Default) Format(failures <-chan lint.Failure, _ lint.Config) (string, error) { |
||||
for failure := range failures { |
||||
fmt.Printf("%v: %s\n", failure.Position.Start, failure.Failure) |
||||
} |
||||
return "", nil |
||||
} |
||||
@ -0,0 +1,146 @@ |
||||
package formatter |
||||
|
||||
import ( |
||||
"bytes" |
||||
"fmt" |
||||
"sort" |
||||
|
||||
"github.com/fatih/color" |
||||
"github.com/mgechev/revive/lint" |
||||
"github.com/olekukonko/tablewriter" |
||||
) |
||||
|
||||
var ( |
||||
errorEmoji = color.RedString("✘") |
||||
warningEmoji = color.YellowString("⚠") |
||||
) |
||||
|
||||
var newLines = map[rune]bool{ |
||||
0x000A: true, |
||||
0x000B: true, |
||||
0x000C: true, |
||||
0x000D: true, |
||||
0x0085: true, |
||||
0x2028: true, |
||||
0x2029: true, |
||||
} |
||||
|
||||
// Friendly is an implementation of the Formatter interface
|
||||
// which formats the errors to JSON.
|
||||
type Friendly struct { |
||||
Metadata lint.FormatterMetadata |
||||
} |
||||
|
||||
// Name returns the name of the formatter
|
||||
func (f *Friendly) Name() string { |
||||
return "friendly" |
||||
} |
||||
|
||||
// Format formats the failures gotten from the lint.
|
||||
func (f *Friendly) Format(failures <-chan lint.Failure, config lint.Config) (string, error) { |
||||
errorMap := map[string]int{} |
||||
warningMap := map[string]int{} |
||||
totalErrors := 0 |
||||
totalWarnings := 0 |
||||
for failure := range failures { |
||||
sev := severity(config, failure) |
||||
f.printFriendlyFailure(failure, sev) |
||||
if sev == lint.SeverityWarning { |
||||
warningMap[failure.RuleName] = warningMap[failure.RuleName] + 1 |
||||
totalWarnings++ |
||||
} |
||||
if sev == lint.SeverityError { |
||||
errorMap[failure.RuleName] = errorMap[failure.RuleName] + 1 |
||||
totalErrors++ |
||||
} |
||||
} |
||||
f.printSummary(totalErrors, totalWarnings) |
||||
f.printStatistics(color.RedString("Errors:"), errorMap) |
||||
f.printStatistics(color.YellowString("Warnings:"), warningMap) |
||||
return "", nil |
||||
} |
||||
|
||||
func (f *Friendly) printFriendlyFailure(failure lint.Failure, severity lint.Severity) { |
||||
f.printHeaderRow(failure, severity) |
||||
f.printFilePosition(failure) |
||||
fmt.Println() |
||||
fmt.Println() |
||||
} |
||||
|
||||
func (f *Friendly) printHeaderRow(failure lint.Failure, severity lint.Severity) { |
||||
emoji := warningEmoji |
||||
if severity == lint.SeverityError { |
||||
emoji = errorEmoji |
||||
} |
||||
fmt.Print(f.table([][]string{{emoji, "https://revive.run/r#" + failure.RuleName, color.GreenString(failure.Failure)}})) |
||||
} |
||||
|
||||
func (f *Friendly) printFilePosition(failure lint.Failure) { |
||||
fmt.Printf(" %s:%d:%d", failure.GetFilename(), failure.Position.Start.Line, failure.Position.Start.Column) |
||||
} |
||||
|
||||
type statEntry struct { |
||||
name string |
||||
failures int |
||||
} |
||||
|
||||
func (f *Friendly) printSummary(errors, warnings int) { |
||||
emoji := warningEmoji |
||||
if errors > 0 { |
||||
emoji = errorEmoji |
||||
} |
||||
problemsLabel := "problems" |
||||
if errors+warnings == 1 { |
||||
problemsLabel = "problem" |
||||
} |
||||
warningsLabel := "warnings" |
||||
if warnings == 1 { |
||||
warningsLabel = "warning" |
||||
} |
||||
errorsLabel := "errors" |
||||
if errors == 1 { |
||||
errorsLabel = "error" |
||||
} |
||||
str := fmt.Sprintf("%d %s (%d %s, %d %s)", errors+warnings, problemsLabel, errors, errorsLabel, warnings, warningsLabel) |
||||
if errors > 0 { |
||||
fmt.Printf("%s %s\n", emoji, color.RedString(str)) |
||||
fmt.Println() |
||||
return |
||||
} |
||||
if warnings > 0 { |
||||
fmt.Printf("%s %s\n", emoji, color.YellowString(str)) |
||||
fmt.Println() |
||||
return |
||||
} |
||||
} |
||||
|
||||
func (f *Friendly) printStatistics(header string, stats map[string]int) { |
||||
if len(stats) == 0 { |
||||
return |
||||
} |
||||
var data []statEntry |
||||
for name, total := range stats { |
||||
data = append(data, statEntry{name, total}) |
||||
} |
||||
sort.Slice(data, func(i, j int) bool { |
||||
return data[i].failures > data[j].failures |
||||
}) |
||||
formatted := [][]string{} |
||||
for _, entry := range data { |
||||
formatted = append(formatted, []string{color.GreenString(fmt.Sprintf("%d", entry.failures)), entry.name}) |
||||
} |
||||
fmt.Println(header) |
||||
fmt.Println(f.table(formatted)) |
||||
} |
||||
|
||||
func (f *Friendly) table(rows [][]string) string { |
||||
buf := new(bytes.Buffer) |
||||
table := tablewriter.NewWriter(buf) |
||||
table.SetBorder(false) |
||||
table.SetColumnSeparator("") |
||||
table.SetRowSeparator("") |
||||
table.SetAutoWrapText(false) |
||||
table.AppendBulk(rows) |
||||
table.Render() |
||||
return buf.String() |
||||
} |
||||
@ -0,0 +1,40 @@ |
||||
package formatter |
||||
|
||||
import ( |
||||
"encoding/json" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// JSON is an implementation of the Formatter interface
|
||||
// which formats the errors to JSON.
|
||||
type JSON struct { |
||||
Metadata lint.FormatterMetadata |
||||
} |
||||
|
||||
// Name returns the name of the formatter
|
||||
func (f *JSON) Name() string { |
||||
return "json" |
||||
} |
||||
|
||||
// jsonObject defines a JSON object of an failure
|
||||
type jsonObject struct { |
||||
Severity lint.Severity |
||||
lint.Failure `json:",inline"` |
||||
} |
||||
|
||||
// Format formats the failures gotten from the lint.
|
||||
func (f *JSON) Format(failures <-chan lint.Failure, config lint.Config) (string, error) { |
||||
var slice []jsonObject |
||||
for failure := range failures { |
||||
obj := jsonObject{} |
||||
obj.Severity = severity(config, failure) |
||||
obj.Failure = failure |
||||
slice = append(slice, obj) |
||||
} |
||||
result, err := json.Marshal(slice) |
||||
if err != nil { |
||||
return "", err |
||||
} |
||||
return string(result), err |
||||
} |
||||
@ -0,0 +1,34 @@ |
||||
package formatter |
||||
|
||||
import ( |
||||
"encoding/json" |
||||
"os" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// NDJSON is an implementation of the Formatter interface
|
||||
// which formats the errors to NDJSON stream.
|
||||
type NDJSON struct { |
||||
Metadata lint.FormatterMetadata |
||||
} |
||||
|
||||
// Name returns the name of the formatter
|
||||
func (f *NDJSON) Name() string { |
||||
return "ndjson" |
||||
} |
||||
|
||||
// Format formats the failures gotten from the lint.
|
||||
func (f *NDJSON) Format(failures <-chan lint.Failure, config lint.Config) (string, error) { |
||||
enc := json.NewEncoder(os.Stdout) |
||||
for failure := range failures { |
||||
obj := jsonObject{} |
||||
obj.Severity = severity(config, failure) |
||||
obj.Failure = failure |
||||
err := enc.Encode(obj) |
||||
if err != nil { |
||||
return "", err |
||||
} |
||||
} |
||||
return "", nil |
||||
} |
||||
@ -0,0 +1,26 @@ |
||||
package formatter |
||||
|
||||
import ( |
||||
"fmt" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// Plain is an implementation of the Formatter interface
|
||||
// which formats the errors to JSON.
|
||||
type Plain struct { |
||||
Metadata lint.FormatterMetadata |
||||
} |
||||
|
||||
// Name returns the name of the formatter
|
||||
func (f *Plain) Name() string { |
||||
return "plain" |
||||
} |
||||
|
||||
// Format formats the failures gotten from the lint.
|
||||
func (f *Plain) Format(failures <-chan lint.Failure, _ lint.Config) (string, error) { |
||||
for failure := range failures { |
||||
fmt.Printf("%v: %s %s\n", failure.Position.Start, failure.Failure, "https://revive.run/r#"+failure.RuleName) |
||||
} |
||||
return "", nil |
||||
} |
||||
@ -0,0 +1,13 @@ |
||||
package formatter |
||||
|
||||
import "github.com/mgechev/revive/lint" |
||||
|
||||
func severity(config lint.Config, failure lint.Failure) lint.Severity { |
||||
if config, ok := config.Rules[failure.RuleName]; ok && config.Severity == lint.SeverityError { |
||||
return lint.SeverityError |
||||
} |
||||
if config, ok := config.Directives[failure.RuleName]; ok && config.Severity == lint.SeverityError { |
||||
return lint.SeverityError |
||||
} |
||||
return lint.SeverityWarning |
||||
} |
||||
@ -0,0 +1,89 @@ |
||||
package formatter |
||||
|
||||
import ( |
||||
"bytes" |
||||
"fmt" |
||||
|
||||
"github.com/fatih/color" |
||||
"github.com/mgechev/revive/lint" |
||||
"github.com/olekukonko/tablewriter" |
||||
) |
||||
|
||||
// Stylish is an implementation of the Formatter interface
|
||||
// which formats the errors to JSON.
|
||||
type Stylish struct { |
||||
Metadata lint.FormatterMetadata |
||||
} |
||||
|
||||
// Name returns the name of the formatter
|
||||
func (f *Stylish) Name() string { |
||||
return "stylish" |
||||
} |
||||
|
||||
func formatFailure(failure lint.Failure, severity lint.Severity) []string { |
||||
fString := color.CyanString(failure.Failure) |
||||
fName := color.RedString("https://revive.run/r#" + failure.RuleName) |
||||
lineColumn := failure.Position |
||||
pos := fmt.Sprintf("(%d, %d)", lineColumn.Start.Line, lineColumn.Start.Column) |
||||
if severity == lint.SeverityWarning { |
||||
fName = color.YellowString("https://revive.run/r#" + failure.RuleName) |
||||
} |
||||
return []string{failure.GetFilename(), pos, fName, fString} |
||||
} |
||||
|
||||
// Format formats the failures gotten from the lint.
|
||||
func (f *Stylish) Format(failures <-chan lint.Failure, config lint.Config) (string, error) { |
||||
var result [][]string |
||||
var totalErrors = 0 |
||||
var total = 0 |
||||
|
||||
for f := range failures { |
||||
total++ |
||||
currentType := severity(config, f) |
||||
if currentType == lint.SeverityError { |
||||
totalErrors++ |
||||
} |
||||
result = append(result, formatFailure(f, lint.Severity(currentType))) |
||||
} |
||||
ps := "problems" |
||||
if total == 1 { |
||||
ps = "problem" |
||||
} |
||||
|
||||
fileReport := make(map[string][][]string) |
||||
|
||||
for _, row := range result { |
||||
if _, ok := fileReport[row[0]]; !ok { |
||||
fileReport[row[0]] = [][]string{} |
||||
} |
||||
|
||||
fileReport[row[0]] = append(fileReport[row[0]], []string{row[1], row[2], row[3]}) |
||||
} |
||||
|
||||
output := "" |
||||
for filename, val := range fileReport { |
||||
buf := new(bytes.Buffer) |
||||
table := tablewriter.NewWriter(buf) |
||||
table.SetBorder(false) |
||||
table.SetColumnSeparator("") |
||||
table.SetRowSeparator("") |
||||
table.SetAutoWrapText(false) |
||||
table.AppendBulk(val) |
||||
table.Render() |
||||
c := color.New(color.Underline) |
||||
output += c.SprintfFunc()(filename + "\n") |
||||
output += buf.String() + "\n" |
||||
} |
||||
|
||||
suffix := fmt.Sprintf(" %d %s (%d errors) (%d warnings)", total, ps, totalErrors, total-totalErrors) |
||||
|
||||
if total > 0 && totalErrors > 0 { |
||||
suffix = color.RedString("\n ✖" + suffix) |
||||
} else if total > 0 && totalErrors == 0 { |
||||
suffix = color.YellowString("\n ✖" + suffix) |
||||
} else { |
||||
suffix, output = "", "" |
||||
} |
||||
|
||||
return output + suffix, nil |
||||
} |
||||
@ -0,0 +1,27 @@ |
||||
package formatter |
||||
|
||||
import ( |
||||
"fmt" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// Unix is an implementation of the Formatter interface
|
||||
// which formats the errors to a simple line based error format
|
||||
// main.go:24:9: [errorf] should replace errors.New(fmt.Sprintf(...)) with fmt.Errorf(...)
|
||||
type Unix struct { |
||||
Metadata lint.FormatterMetadata |
||||
} |
||||
|
||||
// Name returns the name of the formatter
|
||||
func (f *Unix) Name() string { |
||||
return "unix" |
||||
} |
||||
|
||||
// Format formats the failures gotten from the lint.
|
||||
func (f *Unix) Format(failures <-chan lint.Failure, _ lint.Config) (string, error) { |
||||
for failure := range failures { |
||||
fmt.Printf("%v: [%s] %s\n", failure.Position.Start, failure.RuleName, failure.Failure) |
||||
} |
||||
return "", nil |
||||
} |
||||
@ -0,0 +1,32 @@ |
||||
package lint |
||||
|
||||
// Arguments is type used for the arguments of a rule.
|
||||
type Arguments = []interface{} |
||||
|
||||
// RuleConfig is type used for the rule configuration.
|
||||
type RuleConfig struct { |
||||
Arguments Arguments |
||||
Severity Severity |
||||
} |
||||
|
||||
// RulesConfig defines the config for all rules.
|
||||
type RulesConfig = map[string]RuleConfig |
||||
|
||||
// DirectiveConfig is type used for the linter directive configuration.
|
||||
type DirectiveConfig struct { |
||||
Severity Severity |
||||
} |
||||
|
||||
// DirectivesConfig defines the config for all directives.
|
||||
type DirectivesConfig = map[string]DirectiveConfig |
||||
|
||||
// Config defines the config of the linter.
|
||||
type Config struct { |
||||
IgnoreGeneratedHeader bool `toml:"ignoreGeneratedHeader"` |
||||
Confidence float64 |
||||
Severity Severity |
||||
Rules RulesConfig `toml:"rule"` |
||||
ErrorCode int `toml:"errorCode"` |
||||
WarningCode int `toml:"warningCode"` |
||||
Directives DirectivesConfig `toml:"directive"` |
||||
} |
||||
@ -0,0 +1,39 @@ |
||||
package lint |
||||
|
||||
import ( |
||||
"go/ast" |
||||
"go/token" |
||||
) |
||||
|
||||
const ( |
||||
// SeverityWarning declares failures of type warning
|
||||
SeverityWarning = "warning" |
||||
// SeverityError declares failures of type error.
|
||||
SeverityError = "error" |
||||
) |
||||
|
||||
// Severity is the type for the failure types.
|
||||
type Severity string |
||||
|
||||
// FailurePosition returns the failure position
|
||||
type FailurePosition struct { |
||||
Start token.Position |
||||
End token.Position |
||||
} |
||||
|
||||
// Failure defines a struct for a linting failure.
|
||||
type Failure struct { |
||||
Failure string |
||||
RuleName string |
||||
Category string |
||||
Position FailurePosition |
||||
Node ast.Node `json:"-"` |
||||
Confidence float64 |
||||
// For future use
|
||||
ReplacementLine string |
||||
} |
||||
|
||||
// GetFilename returns the filename.
|
||||
func (f *Failure) GetFilename() string { |
||||
return f.Position.Start.Filename |
||||
} |
||||
@ -0,0 +1,278 @@ |
||||
package lint |
||||
|
||||
import ( |
||||
"bytes" |
||||
"go/ast" |
||||
"go/parser" |
||||
"go/printer" |
||||
"go/token" |
||||
"go/types" |
||||
"math" |
||||
"regexp" |
||||
"strings" |
||||
) |
||||
|
||||
// File abstraction used for representing files.
|
||||
type File struct { |
||||
Name string |
||||
Pkg *Package |
||||
content []byte |
||||
AST *ast.File |
||||
} |
||||
|
||||
// IsTest returns if the file contains tests.
|
||||
func (f *File) IsTest() bool { return strings.HasSuffix(f.Name, "_test.go") } |
||||
|
||||
// Content returns the file's content.
|
||||
func (f *File) Content() []byte { |
||||
return f.content |
||||
} |
||||
|
||||
// NewFile creates a new file
|
||||
func NewFile(name string, content []byte, pkg *Package) (*File, error) { |
||||
f, err := parser.ParseFile(pkg.fset, name, content, parser.ParseComments) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
return &File{ |
||||
Name: name, |
||||
content: content, |
||||
Pkg: pkg, |
||||
AST: f, |
||||
}, nil |
||||
} |
||||
|
||||
// ToPosition returns line and column for given position.
|
||||
func (f *File) ToPosition(pos token.Pos) token.Position { |
||||
return f.Pkg.fset.Position(pos) |
||||
} |
||||
|
||||
// Render renters a node.
|
||||
func (f *File) Render(x interface{}) string { |
||||
var buf bytes.Buffer |
||||
if err := printer.Fprint(&buf, f.Pkg.fset, x); err != nil { |
||||
panic(err) |
||||
} |
||||
return buf.String() |
||||
} |
||||
|
||||
// CommentMap builds a comment map for the file.
|
||||
func (f *File) CommentMap() ast.CommentMap { |
||||
return ast.NewCommentMap(f.Pkg.fset, f.AST, f.AST.Comments) |
||||
} |
||||
|
||||
var basicTypeKinds = map[types.BasicKind]string{ |
||||
types.UntypedBool: "bool", |
||||
types.UntypedInt: "int", |
||||
types.UntypedRune: "rune", |
||||
types.UntypedFloat: "float64", |
||||
types.UntypedComplex: "complex128", |
||||
types.UntypedString: "string", |
||||
} |
||||
|
||||
// IsUntypedConst reports whether expr is an untyped constant,
|
||||
// and indicates what its default type is.
|
||||
// scope may be nil.
|
||||
func (f *File) IsUntypedConst(expr ast.Expr) (defType string, ok bool) { |
||||
// Re-evaluate expr outside of its context to see if it's untyped.
|
||||
// (An expr evaluated within, for example, an assignment context will get the type of the LHS.)
|
||||
exprStr := f.Render(expr) |
||||
tv, err := types.Eval(f.Pkg.fset, f.Pkg.TypesPkg, expr.Pos(), exprStr) |
||||
if err != nil { |
||||
return "", false |
||||
} |
||||
if b, ok := tv.Type.(*types.Basic); ok { |
||||
if dt, ok := basicTypeKinds[b.Kind()]; ok { |
||||
return dt, true |
||||
} |
||||
} |
||||
|
||||
return "", false |
||||
} |
||||
|
||||
func (f *File) isMain() bool { |
||||
if f.AST.Name.Name == "main" { |
||||
return true |
||||
} |
||||
return false |
||||
} |
||||
|
||||
const directiveSpecifyDisableReason = "specify-disable-reason" |
||||
|
||||
func (f *File) lint(rules []Rule, config Config, failures chan Failure) { |
||||
rulesConfig := config.Rules |
||||
_, mustSpecifyDisableReason := config.Directives[directiveSpecifyDisableReason] |
||||
disabledIntervals := f.disabledIntervals(rules, mustSpecifyDisableReason, failures) |
||||
for _, currentRule := range rules { |
||||
ruleConfig := rulesConfig[currentRule.Name()] |
||||
currentFailures := currentRule.Apply(f, ruleConfig.Arguments) |
||||
for idx, failure := range currentFailures { |
||||
if failure.RuleName == "" { |
||||
failure.RuleName = currentRule.Name() |
||||
} |
||||
if failure.Node != nil { |
||||
failure.Position = ToFailurePosition(failure.Node.Pos(), failure.Node.End(), f) |
||||
} |
||||
currentFailures[idx] = failure |
||||
} |
||||
currentFailures = f.filterFailures(currentFailures, disabledIntervals) |
||||
for _, failure := range currentFailures { |
||||
if failure.Confidence >= config.Confidence { |
||||
failures <- failure |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
type enableDisableConfig struct { |
||||
enabled bool |
||||
position int |
||||
} |
||||
|
||||
const directiveRE = `^//[\s]*revive:(enable|disable)(?:-(line|next-line))?(?::([^\s]+))?[\s]*(?: (.+))?$` |
||||
const directivePos = 1 |
||||
const modifierPos = 2 |
||||
const rulesPos = 3 |
||||
const reasonPos = 4 |
||||
|
||||
var re = regexp.MustCompile(directiveRE) |
||||
|
||||
func (f *File) disabledIntervals(rules []Rule, mustSpecifyDisableReason bool, failures chan Failure) disabledIntervalsMap { |
||||
enabledDisabledRulesMap := make(map[string][]enableDisableConfig) |
||||
|
||||
getEnabledDisabledIntervals := func() disabledIntervalsMap { |
||||
result := make(disabledIntervalsMap) |
||||
|
||||
for ruleName, disabledArr := range enabledDisabledRulesMap { |
||||
ruleResult := []DisabledInterval{} |
||||
for i := 0; i < len(disabledArr); i++ { |
||||
interval := DisabledInterval{ |
||||
RuleName: ruleName, |
||||
From: token.Position{ |
||||
Filename: f.Name, |
||||
Line: disabledArr[i].position, |
||||
}, |
||||
To: token.Position{ |
||||
Filename: f.Name, |
||||
Line: math.MaxInt32, |
||||
}, |
||||
} |
||||
if i%2 == 0 { |
||||
ruleResult = append(ruleResult, interval) |
||||
} else { |
||||
ruleResult[len(ruleResult)-1].To.Line = disabledArr[i].position |
||||
} |
||||
} |
||||
result[ruleName] = ruleResult |
||||
} |
||||
|
||||
return result |
||||
} |
||||
|
||||
handleConfig := func(isEnabled bool, line int, name string) { |
||||
existing, ok := enabledDisabledRulesMap[name] |
||||
if !ok { |
||||
existing = []enableDisableConfig{} |
||||
enabledDisabledRulesMap[name] = existing |
||||
} |
||||
if (len(existing) > 1 && existing[len(existing)-1].enabled == isEnabled) || |
||||
(len(existing) == 0 && isEnabled) { |
||||
return |
||||
} |
||||
existing = append(existing, enableDisableConfig{ |
||||
enabled: isEnabled, |
||||
position: line, |
||||
}) |
||||
enabledDisabledRulesMap[name] = existing |
||||
} |
||||
|
||||
handleRules := func(filename, modifier string, isEnabled bool, line int, ruleNames []string) []DisabledInterval { |
||||
var result []DisabledInterval |
||||
for _, name := range ruleNames { |
||||
if modifier == "line" { |
||||
handleConfig(isEnabled, line, name) |
||||
handleConfig(!isEnabled, line, name) |
||||
} else if modifier == "next-line" { |
||||
handleConfig(isEnabled, line+1, name) |
||||
handleConfig(!isEnabled, line+1, name) |
||||
} else { |
||||
handleConfig(isEnabled, line, name) |
||||
} |
||||
} |
||||
return result |
||||
} |
||||
|
||||
handleComment := func(filename string, c *ast.CommentGroup, line int) { |
||||
comments := c.List |
||||
for _, c := range comments { |
||||
match := re.FindStringSubmatch(c.Text) |
||||
if len(match) == 0 { |
||||
return |
||||
} |
||||
|
||||
ruleNames := []string{} |
||||
tempNames := strings.Split(match[rulesPos], ",") |
||||
for _, name := range tempNames { |
||||
name = strings.Trim(name, "\n") |
||||
if len(name) > 0 { |
||||
ruleNames = append(ruleNames, name) |
||||
} |
||||
} |
||||
|
||||
mustCheckDisablingReason := mustSpecifyDisableReason && match[directivePos] == "disable" |
||||
if mustCheckDisablingReason && strings.Trim(match[reasonPos], " ") == "" { |
||||
failures <- Failure{ |
||||
Confidence: 1, |
||||
RuleName: directiveSpecifyDisableReason, |
||||
Failure: "reason of lint disabling not found", |
||||
Position: ToFailurePosition(c.Pos(), c.End(), f), |
||||
Node: c, |
||||
} |
||||
continue // skip this linter disabling directive
|
||||
} |
||||
|
||||
// TODO: optimize
|
||||
if len(ruleNames) == 0 { |
||||
for _, rule := range rules { |
||||
ruleNames = append(ruleNames, rule.Name()) |
||||
} |
||||
} |
||||
|
||||
handleRules(filename, match[modifierPos], match[directivePos] == "enable", line, ruleNames) |
||||
} |
||||
} |
||||
|
||||
comments := f.AST.Comments |
||||
for _, c := range comments { |
||||
handleComment(f.Name, c, f.ToPosition(c.End()).Line) |
||||
} |
||||
|
||||
return getEnabledDisabledIntervals() |
||||
} |
||||
|
||||
func (f *File) filterFailures(failures []Failure, disabledIntervals disabledIntervalsMap) []Failure { |
||||
result := []Failure{} |
||||
for _, failure := range failures { |
||||
fStart := failure.Position.Start.Line |
||||
fEnd := failure.Position.End.Line |
||||
intervals, ok := disabledIntervals[failure.RuleName] |
||||
if !ok { |
||||
result = append(result, failure) |
||||
} else { |
||||
include := true |
||||
for _, interval := range intervals { |
||||
intStart := interval.From.Line |
||||
intEnd := interval.To.Line |
||||
if (fStart >= intStart && fStart <= intEnd) || |
||||
(fEnd >= intStart && fEnd <= intEnd) { |
||||
include = false |
||||
break |
||||
} |
||||
} |
||||
if include { |
||||
result = append(result, failure) |
||||
} |
||||
} |
||||
} |
||||
return result |
||||
} |
||||
@ -0,0 +1,14 @@ |
||||
package lint |
||||
|
||||
// FormatterMetadata configuration of a formatter
|
||||
type FormatterMetadata struct { |
||||
Name string |
||||
Description string |
||||
Sample string |
||||
} |
||||
|
||||
// Formatter defines an interface for failure formatters
|
||||
type Formatter interface { |
||||
Format(<-chan Failure, Config) (string, error) |
||||
Name() string |
||||
} |
||||
@ -0,0 +1,99 @@ |
||||
package lint |
||||
|
||||
import ( |
||||
"bufio" |
||||
"bytes" |
||||
"fmt" |
||||
"go/token" |
||||
"os" |
||||
"sync" |
||||
) |
||||
|
||||
// ReadFile defines an abstraction for reading files.
|
||||
type ReadFile func(path string) (result []byte, err error) |
||||
|
||||
type disabledIntervalsMap = map[string][]DisabledInterval |
||||
|
||||
// Linter is used for linting set of files.
|
||||
type Linter struct { |
||||
reader ReadFile |
||||
} |
||||
|
||||
// New creates a new Linter
|
||||
func New(reader ReadFile) Linter { |
||||
return Linter{reader: reader} |
||||
} |
||||
|
||||
var ( |
||||
genHdr = []byte("// Code generated ") |
||||
genFtr = []byte(" DO NOT EDIT.") |
||||
) |
||||
|
||||
// Lint lints a set of files with the specified rule.
|
||||
func (l *Linter) Lint(packages [][]string, ruleSet []Rule, config Config) (<-chan Failure, error) { |
||||
failures := make(chan Failure) |
||||
|
||||
var wg sync.WaitGroup |
||||
for _, pkg := range packages { |
||||
wg.Add(1) |
||||
go func(pkg []string) { |
||||
if err := l.lintPackage(pkg, ruleSet, config, failures); err != nil { |
||||
fmt.Fprintln(os.Stderr, err) |
||||
os.Exit(1) |
||||
} |
||||
defer wg.Done() |
||||
}(pkg) |
||||
} |
||||
|
||||
go func() { |
||||
wg.Wait() |
||||
close(failures) |
||||
}() |
||||
|
||||
return failures, nil |
||||
} |
||||
|
||||
func (l *Linter) lintPackage(filenames []string, ruleSet []Rule, config Config, failures chan Failure) error { |
||||
pkg := &Package{ |
||||
fset: token.NewFileSet(), |
||||
files: map[string]*File{}, |
||||
mu: sync.Mutex{}, |
||||
} |
||||
for _, filename := range filenames { |
||||
content, err := l.reader(filename) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
if isGenerated(content) && !config.IgnoreGeneratedHeader { |
||||
continue |
||||
} |
||||
|
||||
file, err := NewFile(filename, content, pkg) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
pkg.files[filename] = file |
||||
} |
||||
|
||||
if len(pkg.files) == 0 { |
||||
return nil |
||||
} |
||||
|
||||
pkg.lint(ruleSet, config, failures) |
||||
|
||||
return nil |
||||
} |
||||
|
||||
// isGenerated reports whether the source file is generated code
|
||||
// according the rules from https://golang.org/s/generatedcode.
|
||||
// This is inherited from the original go lint.
|
||||
func isGenerated(src []byte) bool { |
||||
sc := bufio.NewScanner(bytes.NewReader(src)) |
||||
for sc.Scan() { |
||||
b := sc.Bytes() |
||||
if bytes.HasPrefix(b, genHdr) && bytes.HasSuffix(b, genFtr) && len(b) >= len(genHdr)+len(genFtr) { |
||||
return true |
||||
} |
||||
} |
||||
return false |
||||
} |
||||
@ -0,0 +1,178 @@ |
||||
package lint |
||||
|
||||
import ( |
||||
"go/ast" |
||||
"go/token" |
||||
"go/types" |
||||
"sync" |
||||
|
||||
"golang.org/x/tools/go/gcexportdata" |
||||
) |
||||
|
||||
// Package represents a package in the project.
|
||||
type Package struct { |
||||
fset *token.FileSet |
||||
files map[string]*File |
||||
|
||||
TypesPkg *types.Package |
||||
TypesInfo *types.Info |
||||
|
||||
// sortable is the set of types in the package that implement sort.Interface.
|
||||
Sortable map[string]bool |
||||
// main is whether this is a "main" package.
|
||||
main int |
||||
mu sync.Mutex |
||||
} |
||||
|
||||
var newImporter = func(fset *token.FileSet) types.ImporterFrom { |
||||
return gcexportdata.NewImporter(fset, make(map[string]*types.Package)) |
||||
} |
||||
|
||||
var ( |
||||
trueValue = 1 |
||||
falseValue = 2 |
||||
notSet = 3 |
||||
) |
||||
|
||||
// IsMain returns if that's the main package.
|
||||
func (p *Package) IsMain() bool { |
||||
if p.main == trueValue { |
||||
return true |
||||
} else if p.main == falseValue { |
||||
return false |
||||
} |
||||
for _, f := range p.files { |
||||
if f.isMain() { |
||||
p.main = trueValue |
||||
return true |
||||
} |
||||
} |
||||
p.main = falseValue |
||||
return false |
||||
} |
||||
|
||||
// TypeCheck performs type checking for given package.
|
||||
func (p *Package) TypeCheck() error { |
||||
p.mu.Lock() |
||||
// If type checking has already been performed
|
||||
// skip it.
|
||||
if p.TypesInfo != nil || p.TypesPkg != nil { |
||||
p.mu.Unlock() |
||||
return nil |
||||
} |
||||
config := &types.Config{ |
||||
// By setting a no-op error reporter, the type checker does as much work as possible.
|
||||
Error: func(error) {}, |
||||
Importer: newImporter(p.fset), |
||||
} |
||||
info := &types.Info{ |
||||
Types: make(map[ast.Expr]types.TypeAndValue), |
||||
Defs: make(map[*ast.Ident]types.Object), |
||||
Uses: make(map[*ast.Ident]types.Object), |
||||
Scopes: make(map[ast.Node]*types.Scope), |
||||
} |
||||
var anyFile *File |
||||
var astFiles []*ast.File |
||||
for _, f := range p.files { |
||||
anyFile = f |
||||
astFiles = append(astFiles, f.AST) |
||||
} |
||||
|
||||
typesPkg, err := check(config, anyFile.AST.Name.Name, p.fset, astFiles, info) |
||||
|
||||
// Remember the typechecking info, even if config.Check failed,
|
||||
// since we will get partial information.
|
||||
p.TypesPkg = typesPkg |
||||
p.TypesInfo = info |
||||
p.mu.Unlock() |
||||
return err |
||||
} |
||||
|
||||
// check function encapsulates the call to go/types.Config.Check method and
|
||||
// recovers if the called method panics (see issue #59)
|
||||
func check(config *types.Config, n string, fset *token.FileSet, astFiles []*ast.File, info *types.Info) (p *types.Package, err error) { |
||||
defer func() { |
||||
if r := recover(); r != nil { |
||||
err, _ = r.(error) |
||||
p = nil |
||||
return |
||||
} |
||||
}() |
||||
|
||||
return config.Check(n, fset, astFiles, info) |
||||
} |
||||
|
||||
// TypeOf returns the type of an expression.
|
||||
func (p *Package) TypeOf(expr ast.Expr) types.Type { |
||||
if p.TypesInfo == nil { |
||||
return nil |
||||
} |
||||
return p.TypesInfo.TypeOf(expr) |
||||
} |
||||
|
||||
type walker struct { |
||||
nmap map[string]int |
||||
has map[string]int |
||||
} |
||||
|
||||
func (w *walker) Visit(n ast.Node) ast.Visitor { |
||||
fn, ok := n.(*ast.FuncDecl) |
||||
if !ok || fn.Recv == nil || len(fn.Recv.List) == 0 { |
||||
return w |
||||
} |
||||
// TODO(dsymonds): We could check the signature to be more precise.
|
||||
recv := receiverType(fn) |
||||
if i, ok := w.nmap[fn.Name.Name]; ok { |
||||
w.has[recv] |= i |
||||
} |
||||
return w |
||||
} |
||||
|
||||
func (p *Package) scanSortable() { |
||||
p.Sortable = make(map[string]bool) |
||||
|
||||
// bitfield for which methods exist on each type.
|
||||
const ( |
||||
Len = 1 << iota |
||||
Less |
||||
Swap |
||||
) |
||||
nmap := map[string]int{"Len": Len, "Less": Less, "Swap": Swap} |
||||
has := make(map[string]int) |
||||
for _, f := range p.files { |
||||
ast.Walk(&walker{nmap, has}, f.AST) |
||||
} |
||||
for typ, ms := range has { |
||||
if ms == Len|Less|Swap { |
||||
p.Sortable[typ] = true |
||||
} |
||||
} |
||||
} |
||||
|
||||
// receiverType returns the named type of the method receiver, sans "*",
|
||||
// or "invalid-type" if fn.Recv is ill formed.
|
||||
func receiverType(fn *ast.FuncDecl) string { |
||||
switch e := fn.Recv.List[0].Type.(type) { |
||||
case *ast.Ident: |
||||
return e.Name |
||||
case *ast.StarExpr: |
||||
if id, ok := e.X.(*ast.Ident); ok { |
||||
return id.Name |
||||
} |
||||
} |
||||
// The parser accepts much more than just the legal forms.
|
||||
return "invalid-type" |
||||
} |
||||
|
||||
func (p *Package) lint(rules []Rule, config Config, failures chan Failure) { |
||||
p.scanSortable() |
||||
var wg sync.WaitGroup |
||||
for _, file := range p.files { |
||||
wg.Add(1) |
||||
go (func(file *File) { |
||||
file.lint(rules, config, failures) |
||||
defer wg.Done() |
||||
})(file) |
||||
} |
||||
wg.Wait() |
||||
} |
||||
@ -0,0 +1,31 @@ |
||||
package lint |
||||
|
||||
import ( |
||||
"go/token" |
||||
) |
||||
|
||||
// DisabledInterval contains a single disabled interval and the associated rule name.
|
||||
type DisabledInterval struct { |
||||
From token.Position |
||||
To token.Position |
||||
RuleName string |
||||
} |
||||
|
||||
// Rule defines an abstract rule interaface
|
||||
type Rule interface { |
||||
Name() string |
||||
Apply(*File, Arguments) []Failure |
||||
} |
||||
|
||||
// AbstractRule defines an abstract rule.
|
||||
type AbstractRule struct { |
||||
Failures []Failure |
||||
} |
||||
|
||||
// ToFailurePosition returns the failure position.
|
||||
func ToFailurePosition(start token.Pos, end token.Pos, file *File) FailurePosition { |
||||
return FailurePosition{ |
||||
Start: file.ToPosition(start), |
||||
End: file.ToPosition(end), |
||||
} |
||||
} |
||||
@ -0,0 +1,128 @@ |
||||
package lint |
||||
|
||||
import ( |
||||
"strings" |
||||
"unicode" |
||||
) |
||||
|
||||
// Name returns a different name if it should be different.
|
||||
func Name(name string, whitelist, blacklist []string) (should string) { |
||||
// Fast path for simple cases: "_" and all lowercase.
|
||||
if name == "_" { |
||||
return name |
||||
} |
||||
allLower := true |
||||
for _, r := range name { |
||||
if !unicode.IsLower(r) { |
||||
allLower = false |
||||
break |
||||
} |
||||
} |
||||
if allLower { |
||||
return name |
||||
} |
||||
|
||||
// Split camelCase at any lower->upper transition, and split on underscores.
|
||||
// Check each word for common initialisms.
|
||||
runes := []rune(name) |
||||
w, i := 0, 0 // index of start of word, scan
|
||||
for i+1 <= len(runes) { |
||||
eow := false // whether we hit the end of a word
|
||||
if i+1 == len(runes) { |
||||
eow = true |
||||
} else if runes[i+1] == '_' { |
||||
// underscore; shift the remainder forward over any run of underscores
|
||||
eow = true |
||||
n := 1 |
||||
for i+n+1 < len(runes) && runes[i+n+1] == '_' { |
||||
n++ |
||||
} |
||||
|
||||
// Leave at most one underscore if the underscore is between two digits
|
||||
if i+n+1 < len(runes) && unicode.IsDigit(runes[i]) && unicode.IsDigit(runes[i+n+1]) { |
||||
n-- |
||||
} |
||||
|
||||
copy(runes[i+1:], runes[i+n+1:]) |
||||
runes = runes[:len(runes)-n] |
||||
} else if unicode.IsLower(runes[i]) && !unicode.IsLower(runes[i+1]) { |
||||
// lower->non-lower
|
||||
eow = true |
||||
} |
||||
i++ |
||||
if !eow { |
||||
continue |
||||
} |
||||
|
||||
// [w,i) is a word.
|
||||
word := string(runes[w:i]) |
||||
ignoreInitWarnings := map[string]bool{} |
||||
for _, i := range whitelist { |
||||
ignoreInitWarnings[i] = true |
||||
} |
||||
|
||||
extraInits := map[string]bool{} |
||||
for _, i := range blacklist { |
||||
extraInits[i] = true |
||||
} |
||||
|
||||
if u := strings.ToUpper(word); (commonInitialisms[u] || extraInits[u]) && !ignoreInitWarnings[u] { |
||||
// Keep consistent case, which is lowercase only at the start.
|
||||
if w == 0 && unicode.IsLower(runes[w]) { |
||||
u = strings.ToLower(u) |
||||
} |
||||
// All the common initialisms are ASCII,
|
||||
// so we can replace the bytes exactly.
|
||||
copy(runes[w:], []rune(u)) |
||||
} else if w > 0 && strings.ToLower(word) == word { |
||||
// already all lowercase, and not the first word, so uppercase the first character.
|
||||
runes[w] = unicode.ToUpper(runes[w]) |
||||
} |
||||
w = i |
||||
} |
||||
return string(runes) |
||||
} |
||||
|
||||
// commonInitialisms is a set of common initialisms.
|
||||
// Only add entries that are highly unlikely to be non-initialisms.
|
||||
// For instance, "ID" is fine (Freudian code is rare), but "AND" is not.
|
||||
var commonInitialisms = map[string]bool{ |
||||
"ACL": true, |
||||
"API": true, |
||||
"ASCII": true, |
||||
"CPU": true, |
||||
"CSS": true, |
||||
"DNS": true, |
||||
"EOF": true, |
||||
"GUID": true, |
||||
"HTML": true, |
||||
"HTTP": true, |
||||
"HTTPS": true, |
||||
"ID": true, |
||||
"IP": true, |
||||
"JSON": true, |
||||
"LHS": true, |
||||
"QPS": true, |
||||
"RAM": true, |
||||
"RHS": true, |
||||
"RPC": true, |
||||
"SLA": true, |
||||
"SMTP": true, |
||||
"SQL": true, |
||||
"SSH": true, |
||||
"TCP": true, |
||||
"TLS": true, |
||||
"TTL": true, |
||||
"UDP": true, |
||||
"UI": true, |
||||
"UID": true, |
||||
"UUID": true, |
||||
"URI": true, |
||||
"URL": true, |
||||
"UTF8": true, |
||||
"VM": true, |
||||
"XML": true, |
||||
"XMPP": true, |
||||
"XSRF": true, |
||||
"XSS": true, |
||||
} |
||||
@ -0,0 +1,151 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"fmt" |
||||
"github.com/mgechev/revive/lint" |
||||
"go/ast" |
||||
"strconv" |
||||
"strings" |
||||
) |
||||
|
||||
const ( |
||||
defaultStrLitLimit = 2 |
||||
kindFLOAT = "FLOAT" |
||||
kindINT = "INT" |
||||
kindSTRING = "STRING" |
||||
) |
||||
|
||||
type whiteList map[string]map[string]bool |
||||
|
||||
func newWhiteList() whiteList { |
||||
return map[string]map[string]bool{kindINT: map[string]bool{}, kindFLOAT: map[string]bool{}, kindSTRING: map[string]bool{}} |
||||
} |
||||
|
||||
func (wl whiteList) add(kind string, list string) { |
||||
elems := strings.Split(list, ",") |
||||
for _, e := range elems { |
||||
wl[kind][e] = true |
||||
} |
||||
} |
||||
|
||||
// AddConstantRule lints unused params in functions.
|
||||
type AddConstantRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *AddConstantRule) Apply(file *lint.File, arguments lint.Arguments) []lint.Failure { |
||||
strLitLimit := defaultStrLitLimit |
||||
var whiteList = newWhiteList() |
||||
if len(arguments) > 0 { |
||||
args, ok := arguments[0].(map[string]interface{}) |
||||
if !ok { |
||||
panic(fmt.Sprintf("Invalid argument to the add-constant rule. Expecting a k,v map, got %T", arguments[0])) |
||||
} |
||||
for k, v := range args { |
||||
kind := "" |
||||
switch k { |
||||
case "allowFloats": |
||||
kind = kindFLOAT |
||||
fallthrough |
||||
case "allowInts": |
||||
if kind == "" { |
||||
kind = kindINT |
||||
} |
||||
fallthrough |
||||
case "allowStrs": |
||||
if kind == "" { |
||||
kind = kindSTRING |
||||
} |
||||
list, ok := v.(string) |
||||
if !ok { |
||||
panic(fmt.Sprintf("Invalid argument to the add-constant rule, string expected. Got '%v' (%T)", v, v)) |
||||
} |
||||
whiteList.add(kind, list) |
||||
case "maxLitCount": |
||||
sl, ok := v.(string) |
||||
if !ok { |
||||
panic(fmt.Sprintf("Invalid argument to the add-constant rule, expecting string representation of an integer. Got '%v' (%T)", v, v)) |
||||
} |
||||
|
||||
limit, err := strconv.Atoi(sl) |
||||
if err != nil { |
||||
panic(fmt.Sprintf("Invalid argument to the add-constant rule, expecting string representation of an integer. Got '%v'", v)) |
||||
} |
||||
strLitLimit = limit |
||||
} |
||||
} |
||||
} |
||||
|
||||
var failures []lint.Failure |
||||
|
||||
onFailure := func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
} |
||||
|
||||
w := lintAddConstantRule{onFailure: onFailure, strLits: make(map[string]int, 0), strLitLimit: strLitLimit, whiteLst: whiteList} |
||||
|
||||
ast.Walk(w, file.AST) |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *AddConstantRule) Name() string { |
||||
return "add-constant" |
||||
} |
||||
|
||||
type lintAddConstantRule struct { |
||||
onFailure func(lint.Failure) |
||||
strLits map[string]int |
||||
strLitLimit int |
||||
whiteLst whiteList |
||||
} |
||||
|
||||
func (w lintAddConstantRule) Visit(node ast.Node) ast.Visitor { |
||||
switch n := node.(type) { |
||||
case *ast.GenDecl: |
||||
return nil // skip declarations
|
||||
case *ast.BasicLit: |
||||
switch kind := n.Kind.String(); kind { |
||||
case kindFLOAT, kindINT: |
||||
w.checkNumLit(kind, n) |
||||
case kindSTRING: |
||||
w.checkStrLit(n) |
||||
} |
||||
} |
||||
|
||||
return w |
||||
|
||||
} |
||||
|
||||
func (w lintAddConstantRule) checkStrLit(n *ast.BasicLit) { |
||||
if w.whiteLst[kindSTRING][n.Value] { |
||||
return |
||||
} |
||||
|
||||
count := w.strLits[n.Value] |
||||
if count >= 0 { |
||||
w.strLits[n.Value] = count + 1 |
||||
if w.strLits[n.Value] > w.strLitLimit { |
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Node: n, |
||||
Category: "style", |
||||
Failure: fmt.Sprintf("string literal %s appears, at least, %d times, create a named constant for it", n.Value, w.strLits[n.Value]), |
||||
}) |
||||
w.strLits[n.Value] = -1 // mark it to avoid failing again on the same literal
|
||||
} |
||||
} |
||||
} |
||||
|
||||
func (w lintAddConstantRule) checkNumLit(kind string, n *ast.BasicLit) { |
||||
if w.whiteLst[kind][n.Value] { |
||||
return |
||||
} |
||||
|
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Node: n, |
||||
Category: "style", |
||||
Failure: fmt.Sprintf("avoid magic numbers like '%s', create a named constant for it", n.Value), |
||||
}) |
||||
} |
||||
@ -0,0 +1,67 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"fmt" |
||||
"go/ast" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// ArgumentsLimitRule lints given else constructs.
|
||||
type ArgumentsLimitRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *ArgumentsLimitRule) Apply(file *lint.File, arguments lint.Arguments) []lint.Failure { |
||||
if len(arguments) != 1 { |
||||
panic(`invalid configuration for "argument-limit"`) |
||||
} |
||||
|
||||
total, ok := arguments[0].(int64) // Alt. non panicking version
|
||||
if !ok { |
||||
panic(`invalid value passed as argument number to the "argument-list" rule`) |
||||
} |
||||
|
||||
var failures []lint.Failure |
||||
|
||||
walker := lintArgsNum{ |
||||
total: int(total), |
||||
onFailure: func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
}, |
||||
} |
||||
|
||||
ast.Walk(walker, file.AST) |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *ArgumentsLimitRule) Name() string { |
||||
return "argument-limit" |
||||
} |
||||
|
||||
type lintArgsNum struct { |
||||
total int |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w lintArgsNum) Visit(n ast.Node) ast.Visitor { |
||||
node, ok := n.(*ast.FuncDecl) |
||||
if ok { |
||||
num := 0 |
||||
for _, l := range node.Type.Params.List { |
||||
for range l.Names { |
||||
num++ |
||||
} |
||||
} |
||||
if num > w.total { |
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Failure: fmt.Sprintf("maximum number of arguments per function exceeded; max %d but got %d", w.total, num), |
||||
Node: node.Type, |
||||
}) |
||||
return w |
||||
} |
||||
} |
||||
return w |
||||
} |
||||
@ -0,0 +1,94 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"go/ast" |
||||
"go/token" |
||||
"go/types" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// AtomicRule lints given else constructs.
|
||||
type AtomicRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *AtomicRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
walker := atomic{ |
||||
pkgTypesInfo: file.Pkg.TypesInfo, |
||||
onFailure: func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
}, |
||||
} |
||||
|
||||
ast.Walk(walker, file.AST) |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *AtomicRule) Name() string { |
||||
return "atomic" |
||||
} |
||||
|
||||
type atomic struct { |
||||
pkgTypesInfo *types.Info |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w atomic) Visit(node ast.Node) ast.Visitor { |
||||
n, ok := node.(*ast.AssignStmt) |
||||
if !ok { |
||||
return w |
||||
} |
||||
|
||||
if len(n.Lhs) != len(n.Rhs) { |
||||
return nil // skip assignment sub-tree
|
||||
} |
||||
if len(n.Lhs) == 1 && n.Tok == token.DEFINE { |
||||
return nil // skip assignment sub-tree
|
||||
} |
||||
|
||||
for i, right := range n.Rhs { |
||||
call, ok := right.(*ast.CallExpr) |
||||
if !ok { |
||||
continue |
||||
} |
||||
sel, ok := call.Fun.(*ast.SelectorExpr) |
||||
if !ok { |
||||
continue |
||||
} |
||||
pkgIdent, _ := sel.X.(*ast.Ident) |
||||
if w.pkgTypesInfo != nil { |
||||
pkgName, ok := w.pkgTypesInfo.Uses[pkgIdent].(*types.PkgName) |
||||
if !ok || pkgName.Imported().Path() != "sync/atomic" { |
||||
continue |
||||
} |
||||
} |
||||
|
||||
switch sel.Sel.Name { |
||||
case "AddInt32", "AddInt64", "AddUint32", "AddUint64", "AddUintptr": |
||||
left := n.Lhs[i] |
||||
if len(call.Args) != 2 { |
||||
continue |
||||
} |
||||
arg := call.Args[0] |
||||
broken := false |
||||
|
||||
if uarg, ok := arg.(*ast.UnaryExpr); ok && uarg.Op == token.AND { |
||||
broken = gofmt(left) == gofmt(uarg.X) |
||||
} else if star, ok := left.(*ast.StarExpr); ok { |
||||
broken = gofmt(star.X) == gofmt(arg) |
||||
} |
||||
|
||||
if broken { |
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Failure: "direct assignment to atomic value", |
||||
Node: n, |
||||
}) |
||||
} |
||||
} |
||||
} |
||||
return w |
||||
} |
||||
@ -0,0 +1,84 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"go/ast" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// BareReturnRule lints given else constructs.
|
||||
type BareReturnRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *BareReturnRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
onFailure := func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
} |
||||
|
||||
w := lintBareReturnRule{onFailure: onFailure} |
||||
ast.Walk(w, file.AST) |
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *BareReturnRule) Name() string { |
||||
return "bare-return" |
||||
} |
||||
|
||||
type lintBareReturnRule struct { |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w lintBareReturnRule) Visit(node ast.Node) ast.Visitor { |
||||
switch n := node.(type) { |
||||
case *ast.FuncDecl: |
||||
w.checkFunc(n.Type.Results, n.Body) |
||||
case *ast.FuncLit: // to cope with deferred functions and go-routines
|
||||
w.checkFunc(n.Type.Results, n.Body) |
||||
} |
||||
|
||||
return w |
||||
} |
||||
|
||||
// checkFunc will verify if the given function has named result and bare returns
|
||||
func (w lintBareReturnRule) checkFunc(results *ast.FieldList, body *ast.BlockStmt) { |
||||
hasNamedResults := results != nil && len(results.List) > 0 && results.List[0].Names != nil |
||||
if !hasNamedResults || body == nil { |
||||
return // nothing to do
|
||||
} |
||||
|
||||
brf := bareReturnFinder{w.onFailure} |
||||
ast.Walk(brf, body) |
||||
} |
||||
|
||||
type bareReturnFinder struct { |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w bareReturnFinder) Visit(node ast.Node) ast.Visitor { |
||||
_, ok := node.(*ast.FuncLit) |
||||
if ok { |
||||
// skip analysing function literals
|
||||
// they will analyzed by the lintBareReturnRule.Visit method
|
||||
return nil |
||||
} |
||||
|
||||
rs, ok := node.(*ast.ReturnStmt) |
||||
if !ok { |
||||
return w |
||||
} |
||||
|
||||
if len(rs.Results) > 0 { |
||||
return w |
||||
} |
||||
|
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Node: rs, |
||||
Failure: "avoid using bare returns, please add return expressions", |
||||
}) |
||||
|
||||
return w |
||||
} |
||||
@ -0,0 +1,74 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"go/ast" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// BlankImportsRule lints given else constructs.
|
||||
type BlankImportsRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *BlankImportsRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
fileAst := file.AST |
||||
walker := lintBlankImports{ |
||||
file: file, |
||||
fileAst: fileAst, |
||||
onFailure: func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
}, |
||||
} |
||||
|
||||
ast.Walk(walker, fileAst) |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *BlankImportsRule) Name() string { |
||||
return "blank-imports" |
||||
} |
||||
|
||||
type lintBlankImports struct { |
||||
fileAst *ast.File |
||||
file *lint.File |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w lintBlankImports) Visit(_ ast.Node) ast.Visitor { |
||||
// In package main and in tests, we don't complain about blank imports.
|
||||
if w.file.Pkg.IsMain() || w.file.IsTest() { |
||||
return nil |
||||
} |
||||
|
||||
// The first element of each contiguous group of blank imports should have
|
||||
// an explanatory comment of some kind.
|
||||
for i, imp := range w.fileAst.Imports { |
||||
pos := w.file.ToPosition(imp.Pos()) |
||||
|
||||
if !isBlank(imp.Name) { |
||||
continue // Ignore non-blank imports.
|
||||
} |
||||
if i > 0 { |
||||
prev := w.fileAst.Imports[i-1] |
||||
prevPos := w.file.ToPosition(prev.Pos()) |
||||
if isBlank(prev.Name) && prevPos.Line+1 == pos.Line { |
||||
continue // A subsequent blank in a group.
|
||||
} |
||||
} |
||||
|
||||
// This is the first blank import of a group.
|
||||
if imp.Doc == nil && imp.Comment == nil { |
||||
w.onFailure(lint.Failure{ |
||||
Node: imp, |
||||
Failure: "a blank import should be only in a main or test package, or have a comment justifying it", |
||||
Confidence: 1, |
||||
Category: "imports", |
||||
}) |
||||
} |
||||
} |
||||
return nil |
||||
} |
||||
@ -0,0 +1,73 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"go/ast" |
||||
"go/token" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// BoolLiteralRule warns when logic expressions contains Boolean literals.
|
||||
type BoolLiteralRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *BoolLiteralRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
onFailure := func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
} |
||||
|
||||
astFile := file.AST |
||||
w := &lintBoolLiteral{astFile, onFailure} |
||||
ast.Walk(w, astFile) |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *BoolLiteralRule) Name() string { |
||||
return "bool-literal-in-expr" |
||||
} |
||||
|
||||
type lintBoolLiteral struct { |
||||
file *ast.File |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w *lintBoolLiteral) Visit(node ast.Node) ast.Visitor { |
||||
switch n := node.(type) { |
||||
case *ast.BinaryExpr: |
||||
if !isBoolOp(n.Op) { |
||||
return w |
||||
} |
||||
|
||||
lexeme, ok := isExprABooleanLit(n.X) |
||||
if !ok { |
||||
lexeme, ok = isExprABooleanLit(n.Y) |
||||
|
||||
if !ok { |
||||
return w |
||||
} |
||||
} |
||||
|
||||
isConstant := (n.Op == token.LAND && lexeme == "false") || (n.Op == token.LOR && lexeme == "true") |
||||
|
||||
if isConstant { |
||||
w.addFailure(n, "Boolean expression seems to always evaluate to "+lexeme, "logic") |
||||
} else { |
||||
w.addFailure(n, "omit Boolean literal in expression", "style") |
||||
} |
||||
} |
||||
|
||||
return w |
||||
} |
||||
|
||||
func (w lintBoolLiteral) addFailure(node ast.Node, msg string, cat string) { |
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Node: node, |
||||
Category: cat, |
||||
Failure: msg, |
||||
}) |
||||
} |
||||
@ -0,0 +1,70 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"go/ast" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// CallToGCRule lints calls to the garbage collector.
|
||||
type CallToGCRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *CallToGCRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
onFailure := func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
} |
||||
|
||||
var gcTriggeringFunctions = map[string]map[string]bool{ |
||||
"runtime": map[string]bool{"GC": true}, |
||||
} |
||||
|
||||
w := lintCallToGC{onFailure, gcTriggeringFunctions} |
||||
ast.Walk(w, file.AST) |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *CallToGCRule) Name() string { |
||||
return "call-to-gc" |
||||
} |
||||
|
||||
type lintCallToGC struct { |
||||
onFailure func(lint.Failure) |
||||
gcTriggeringFunctions map[string]map[string]bool |
||||
} |
||||
|
||||
func (w lintCallToGC) Visit(node ast.Node) ast.Visitor { |
||||
ce, ok := node.(*ast.CallExpr) |
||||
if !ok { |
||||
return w // nothing to do, the node is not a call
|
||||
} |
||||
|
||||
fc, ok := ce.Fun.(*ast.SelectorExpr) |
||||
if !ok { |
||||
return nil // nothing to do, the call is not of the form pkg.func(...)
|
||||
} |
||||
|
||||
id, ok := fc.X.(*ast.Ident) |
||||
|
||||
if !ok { |
||||
return nil // in case X is not an id (it should be!)
|
||||
} |
||||
|
||||
fn := fc.Sel.Name |
||||
pkg := id.Name |
||||
if !w.gcTriggeringFunctions[pkg][fn] { |
||||
return nil // it isn't a call to a GC triggering function
|
||||
} |
||||
|
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Node: node, |
||||
Category: "bad practice", |
||||
Failure: "explicit call to the garbage collector", |
||||
}) |
||||
|
||||
return w |
||||
} |
||||
@ -0,0 +1,195 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"fmt" |
||||
"go/ast" |
||||
"go/token" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
"golang.org/x/tools/go/ast/astutil" |
||||
) |
||||
|
||||
// CognitiveComplexityRule lints given else constructs.
|
||||
type CognitiveComplexityRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *CognitiveComplexityRule) Apply(file *lint.File, arguments lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
const expectedArgumentsCount = 1 |
||||
if len(arguments) < expectedArgumentsCount { |
||||
panic(fmt.Sprintf("not enough arguments for cognitive-complexity, expected %d, got %d", expectedArgumentsCount, len(arguments))) |
||||
} |
||||
complexity, ok := arguments[0].(int64) |
||||
if !ok { |
||||
panic(fmt.Sprintf("invalid argument type for cognitive-complexity, expected int64, got %T", arguments[0])) |
||||
} |
||||
|
||||
linter := cognitiveComplexityLinter{ |
||||
file: file, |
||||
maxComplexity: int(complexity), |
||||
onFailure: func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
}, |
||||
} |
||||
|
||||
linter.lint() |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *CognitiveComplexityRule) Name() string { |
||||
return "cognitive-complexity" |
||||
} |
||||
|
||||
type cognitiveComplexityLinter struct { |
||||
file *lint.File |
||||
maxComplexity int |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w cognitiveComplexityLinter) lint() { |
||||
f := w.file |
||||
for _, decl := range f.AST.Decls { |
||||
if fn, ok := decl.(*ast.FuncDecl); ok { |
||||
v := cognitiveComplexityVisitor{} |
||||
c := v.subTreeComplexity(fn.Body) |
||||
if c > w.maxComplexity { |
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Category: "maintenance", |
||||
Failure: fmt.Sprintf("function %s has cognitive complexity %d (> max enabled %d)", funcName(fn), c, w.maxComplexity), |
||||
Node: fn, |
||||
}) |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
type cognitiveComplexityVisitor struct { |
||||
complexity int |
||||
nestingLevel int |
||||
} |
||||
|
||||
// subTreeComplexity calculates the cognitive complexity of an AST-subtree.
|
||||
func (v cognitiveComplexityVisitor) subTreeComplexity(n ast.Node) int { |
||||
ast.Walk(&v, n) |
||||
return v.complexity |
||||
} |
||||
|
||||
// Visit implements the ast.Visitor interface.
|
||||
func (v *cognitiveComplexityVisitor) Visit(n ast.Node) ast.Visitor { |
||||
switch n := n.(type) { |
||||
case *ast.IfStmt: |
||||
targets := []ast.Node{n.Cond, n.Body, n.Else} |
||||
v.walk(1, targets...) |
||||
return nil |
||||
case *ast.ForStmt: |
||||
targets := []ast.Node{n.Cond, n.Body} |
||||
v.walk(1, targets...) |
||||
return nil |
||||
case *ast.RangeStmt: |
||||
v.walk(1, n.Body) |
||||
return nil |
||||
case *ast.SelectStmt: |
||||
v.walk(1, n.Body) |
||||
return nil |
||||
case *ast.SwitchStmt: |
||||
v.walk(1, n.Body) |
||||
return nil |
||||
case *ast.TypeSwitchStmt: |
||||
v.walk(1, n.Body) |
||||
return nil |
||||
case *ast.FuncLit: |
||||
v.walk(0, n.Body) // do not increment the complexity, just do the nesting
|
||||
return nil |
||||
case *ast.BinaryExpr: |
||||
v.complexity += v.binExpComplexity(n) |
||||
return nil // skip visiting binexp sub-tree (already visited by binExpComplexity)
|
||||
case *ast.BranchStmt: |
||||
if n.Label != nil { |
||||
v.complexity += 1 |
||||
} |
||||
} |
||||
// TODO handle (at least) direct recursion
|
||||
|
||||
return v |
||||
} |
||||
|
||||
func (v *cognitiveComplexityVisitor) walk(complexityIncrement int, targets ...ast.Node) { |
||||
v.complexity += complexityIncrement + v.nestingLevel |
||||
nesting := v.nestingLevel |
||||
v.nestingLevel++ |
||||
|
||||
for _, t := range targets { |
||||
if t == nil { |
||||
continue |
||||
} |
||||
|
||||
ast.Walk(v, t) |
||||
} |
||||
|
||||
v.nestingLevel = nesting |
||||
} |
||||
|
||||
func (cognitiveComplexityVisitor) binExpComplexity(n *ast.BinaryExpr) int { |
||||
calculator := binExprComplexityCalculator{opsStack: []token.Token{}} |
||||
|
||||
astutil.Apply(n, calculator.pre, calculator.post) |
||||
|
||||
return calculator.complexity |
||||
} |
||||
|
||||
type binExprComplexityCalculator struct { |
||||
complexity int |
||||
opsStack []token.Token // stack of bool operators
|
||||
subexpStarted bool |
||||
} |
||||
|
||||
func (becc *binExprComplexityCalculator) pre(c *astutil.Cursor) bool { |
||||
switch n := c.Node().(type) { |
||||
case *ast.BinaryExpr: |
||||
isBoolOp := n.Op == token.LAND || n.Op == token.LOR |
||||
if !isBoolOp { |
||||
break |
||||
} |
||||
|
||||
ops := len(becc.opsStack) |
||||
// if
|
||||
// is the first boolop in the expression OR
|
||||
// is the first boolop inside a subexpression (...) OR
|
||||
// is not the same to the previous one
|
||||
// then
|
||||
// increment complexity
|
||||
if ops == 0 || becc.subexpStarted || n.Op != becc.opsStack[ops-1] { |
||||
becc.complexity++ |
||||
becc.subexpStarted = false |
||||
} |
||||
|
||||
becc.opsStack = append(becc.opsStack, n.Op) |
||||
case *ast.ParenExpr: |
||||
becc.subexpStarted = true |
||||
} |
||||
|
||||
return true |
||||
} |
||||
|
||||
func (becc *binExprComplexityCalculator) post(c *astutil.Cursor) bool { |
||||
switch n := c.Node().(type) { |
||||
case *ast.BinaryExpr: |
||||
isBoolOp := n.Op == token.LAND || n.Op == token.LOR |
||||
if !isBoolOp { |
||||
break |
||||
} |
||||
|
||||
ops := len(becc.opsStack) |
||||
if ops > 0 { |
||||
becc.opsStack = becc.opsStack[:ops-1] |
||||
} |
||||
case *ast.ParenExpr: |
||||
becc.subexpStarted = false |
||||
} |
||||
|
||||
return true |
||||
} |
||||
@ -0,0 +1,190 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"fmt" |
||||
"go/ast" |
||||
|
||||
"strings" |
||||
"sync" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
type referenceMethod struct { |
||||
fileName string |
||||
id *ast.Ident |
||||
} |
||||
|
||||
type pkgMethods struct { |
||||
pkg *lint.Package |
||||
methods map[string]map[string]*referenceMethod |
||||
mu *sync.Mutex |
||||
} |
||||
|
||||
type packages struct { |
||||
pkgs []pkgMethods |
||||
mu sync.Mutex |
||||
} |
||||
|
||||
func (ps *packages) methodNames(lp *lint.Package) pkgMethods { |
||||
ps.mu.Lock() |
||||
|
||||
for _, pkg := range ps.pkgs { |
||||
if pkg.pkg == lp { |
||||
ps.mu.Unlock() |
||||
return pkg |
||||
} |
||||
} |
||||
|
||||
pkgm := pkgMethods{pkg: lp, methods: make(map[string]map[string]*referenceMethod), mu: &sync.Mutex{}} |
||||
ps.pkgs = append(ps.pkgs, pkgm) |
||||
|
||||
ps.mu.Unlock() |
||||
return pkgm |
||||
} |
||||
|
||||
var allPkgs = packages{pkgs: make([]pkgMethods, 1)} |
||||
|
||||
// ConfusingNamingRule lints method names that differ only by capitalization
|
||||
type ConfusingNamingRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *ConfusingNamingRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
fileAst := file.AST |
||||
pkgm := allPkgs.methodNames(file.Pkg) |
||||
walker := lintConfusingNames{ |
||||
fileName: file.Name, |
||||
pkgm: pkgm, |
||||
onFailure: func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
}, |
||||
} |
||||
|
||||
ast.Walk(&walker, fileAst) |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *ConfusingNamingRule) Name() string { |
||||
return "confusing-naming" |
||||
} |
||||
|
||||
//checkMethodName checks if a given method/function name is similar (just case differences) to other method/function of the same struct/file.
|
||||
func checkMethodName(holder string, id *ast.Ident, w *lintConfusingNames) { |
||||
if id.Name == "init" && holder == defaultStructName { |
||||
// ignore init functions
|
||||
return |
||||
} |
||||
|
||||
pkgm := w.pkgm |
||||
name := strings.ToUpper(id.Name) |
||||
|
||||
pkgm.mu.Lock() |
||||
defer pkgm.mu.Unlock() |
||||
|
||||
if pkgm.methods[holder] != nil { |
||||
if pkgm.methods[holder][name] != nil { |
||||
refMethod := pkgm.methods[holder][name] |
||||
// confusing names
|
||||
var kind string |
||||
if holder == defaultStructName { |
||||
kind = "function" |
||||
} else { |
||||
kind = "method" |
||||
} |
||||
var fileName string |
||||
if w.fileName == refMethod.fileName { |
||||
fileName = "the same source file" |
||||
} else { |
||||
fileName = refMethod.fileName |
||||
} |
||||
w.onFailure(lint.Failure{ |
||||
Failure: fmt.Sprintf("Method '%s' differs only by capitalization to %s '%s' in %s", id.Name, kind, refMethod.id.Name, fileName), |
||||
Confidence: 1, |
||||
Node: id, |
||||
Category: "naming", |
||||
}) |
||||
|
||||
return |
||||
} |
||||
} else { |
||||
pkgm.methods[holder] = make(map[string]*referenceMethod, 1) |
||||
} |
||||
|
||||
// update the black list
|
||||
if pkgm.methods[holder] == nil { |
||||
println("no entry for '", holder, "'") |
||||
} |
||||
pkgm.methods[holder][name] = &referenceMethod{fileName: w.fileName, id: id} |
||||
} |
||||
|
||||
type lintConfusingNames struct { |
||||
fileName string |
||||
pkgm pkgMethods |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
const defaultStructName = "_" // used to map functions
|
||||
|
||||
//getStructName of a function receiver. Defaults to defaultStructName
|
||||
func getStructName(r *ast.FieldList) string { |
||||
result := defaultStructName |
||||
|
||||
if r == nil || len(r.List) < 1 { |
||||
return result |
||||
} |
||||
|
||||
t := r.List[0].Type |
||||
|
||||
if p, _ := t.(*ast.StarExpr); p != nil { // if a pointer receiver => dereference pointer receiver types
|
||||
t = p.X |
||||
} |
||||
|
||||
if p, _ := t.(*ast.Ident); p != nil { |
||||
result = p.Name |
||||
} |
||||
|
||||
return result |
||||
} |
||||
|
||||
func checkStructFields(fields *ast.FieldList, structName string, w *lintConfusingNames) { |
||||
bl := make(map[string]bool, len(fields.List)) |
||||
for _, f := range fields.List { |
||||
for _, id := range f.Names { |
||||
normName := strings.ToUpper(id.Name) |
||||
if bl[normName] { |
||||
w.onFailure(lint.Failure{ |
||||
Failure: fmt.Sprintf("Field '%s' differs only by capitalization to other field in the struct type %s", id.Name, structName), |
||||
Confidence: 1, |
||||
Node: id, |
||||
Category: "naming", |
||||
}) |
||||
} else { |
||||
bl[normName] = true |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
func (w *lintConfusingNames) Visit(n ast.Node) ast.Visitor { |
||||
switch v := n.(type) { |
||||
case *ast.FuncDecl: |
||||
// Exclude naming warnings for functions that are exported to C but
|
||||
// not exported in the Go API.
|
||||
// See https://github.com/golang/lint/issues/144.
|
||||
if ast.IsExported(v.Name.Name) || !isCgoExported(v) { |
||||
checkMethodName(getStructName(v.Recv), v.Name, w) |
||||
} |
||||
case *ast.TypeSpec: |
||||
if s, ok := v.Type.(*ast.StructType); ok { |
||||
checkStructFields(s.Fields, v.Name.Name, w) |
||||
} |
||||
|
||||
default: |
||||
// will add other checks like field names, struct names, etc.
|
||||
} |
||||
|
||||
return w |
||||
} |
||||
@ -0,0 +1,67 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"go/ast" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// ConfusingResultsRule lints given function declarations
|
||||
type ConfusingResultsRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *ConfusingResultsRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
fileAst := file.AST |
||||
walker := lintConfusingResults{ |
||||
onFailure: func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
}, |
||||
} |
||||
|
||||
ast.Walk(walker, fileAst) |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *ConfusingResultsRule) Name() string { |
||||
return "confusing-results" |
||||
} |
||||
|
||||
type lintConfusingResults struct { |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w lintConfusingResults) Visit(n ast.Node) ast.Visitor { |
||||
fn, ok := n.(*ast.FuncDecl) |
||||
if !ok || fn.Type.Results == nil || len(fn.Type.Results.List) < 2 { |
||||
return w |
||||
} |
||||
lastType := "" |
||||
for _, result := range fn.Type.Results.List { |
||||
if len(result.Names) > 0 { |
||||
return w |
||||
} |
||||
|
||||
t, ok := result.Type.(*ast.Ident) |
||||
if !ok { |
||||
return w |
||||
} |
||||
|
||||
if t.Name == lastType { |
||||
w.onFailure(lint.Failure{ |
||||
Node: n, |
||||
Confidence: 1, |
||||
Category: "naming", |
||||
Failure: "unnamed results of the same type may be confusing, consider using named results", |
||||
}) |
||||
break |
||||
} |
||||
lastType = t.Name |
||||
|
||||
} |
||||
|
||||
return w |
||||
} |
||||
@ -0,0 +1,88 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"github.com/mgechev/revive/lint" |
||||
"go/ast" |
||||
"go/token" |
||||
) |
||||
|
||||
// ConstantLogicalExprRule warns on constant logical expressions.
|
||||
type ConstantLogicalExprRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *ConstantLogicalExprRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
onFailure := func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
} |
||||
|
||||
astFile := file.AST |
||||
w := &lintConstantLogicalExpr{astFile, onFailure} |
||||
ast.Walk(w, astFile) |
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *ConstantLogicalExprRule) Name() string { |
||||
return "constant-logical-expr" |
||||
} |
||||
|
||||
type lintConstantLogicalExpr struct { |
||||
file *ast.File |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w *lintConstantLogicalExpr) Visit(node ast.Node) ast.Visitor { |
||||
switch n := node.(type) { |
||||
case *ast.BinaryExpr: |
||||
if !w.isOperatorWithLogicalResult(n.Op) { |
||||
return w |
||||
} |
||||
|
||||
if gofmt(n.X) != gofmt(n.Y) { // check if subexpressions are the same
|
||||
return w |
||||
} |
||||
|
||||
if n.Op == token.EQL { |
||||
w.newFailure(n, "expression always evaluates to true") |
||||
return w |
||||
} |
||||
|
||||
if w.isInequalityOperator(n.Op) { |
||||
w.newFailure(n, "expression always evaluates to false") |
||||
return w |
||||
} |
||||
|
||||
w.newFailure(n, "left and right hand-side sub-expressions are the same") |
||||
} |
||||
|
||||
return w |
||||
} |
||||
|
||||
func (w *lintConstantLogicalExpr) isOperatorWithLogicalResult(t token.Token) bool { |
||||
switch t { |
||||
case token.LAND, token.LOR, token.EQL, token.LSS, token.GTR, token.NEQ, token.LEQ, token.GEQ: |
||||
return true |
||||
} |
||||
|
||||
return false |
||||
} |
||||
|
||||
func (w *lintConstantLogicalExpr) isInequalityOperator(t token.Token) bool { |
||||
switch t { |
||||
case token.LSS, token.GTR, token.NEQ, token.LEQ, token.GEQ: |
||||
return true |
||||
} |
||||
|
||||
return false |
||||
} |
||||
|
||||
func (w lintConstantLogicalExpr) newFailure(node ast.Node, msg string) { |
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Node: node, |
||||
Category: "logic", |
||||
Failure: msg, |
||||
}) |
||||
} |
||||
@ -0,0 +1,60 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"go/ast" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// ContextAsArgumentRule lints given else constructs.
|
||||
type ContextAsArgumentRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *ContextAsArgumentRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
fileAst := file.AST |
||||
walker := lintContextArguments{ |
||||
file: file, |
||||
fileAst: fileAst, |
||||
onFailure: func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
}, |
||||
} |
||||
|
||||
ast.Walk(walker, fileAst) |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *ContextAsArgumentRule) Name() string { |
||||
return "context-as-argument" |
||||
} |
||||
|
||||
type lintContextArguments struct { |
||||
file *lint.File |
||||
fileAst *ast.File |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w lintContextArguments) Visit(n ast.Node) ast.Visitor { |
||||
fn, ok := n.(*ast.FuncDecl) |
||||
if !ok || len(fn.Type.Params.List) <= 1 { |
||||
return w |
||||
} |
||||
// A context.Context should be the first parameter of a function.
|
||||
// Flag any that show up after the first.
|
||||
for _, arg := range fn.Type.Params.List[1:] { |
||||
if isPkgDot(arg.Type, "context", "Context") { |
||||
w.onFailure(lint.Failure{ |
||||
Node: fn, |
||||
Category: "arg-order", |
||||
Failure: "context.Context should be the first parameter of a function", |
||||
Confidence: 0.9, |
||||
}) |
||||
break // only flag one
|
||||
} |
||||
} |
||||
return w |
||||
} |
||||
@ -0,0 +1,81 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"fmt" |
||||
"go/ast" |
||||
"go/types" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// ContextKeysType lints given else constructs.
|
||||
type ContextKeysType struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *ContextKeysType) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
fileAst := file.AST |
||||
walker := lintContextKeyTypes{ |
||||
file: file, |
||||
fileAst: fileAst, |
||||
onFailure: func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
}, |
||||
} |
||||
|
||||
file.Pkg.TypeCheck() |
||||
ast.Walk(walker, fileAst) |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *ContextKeysType) Name() string { |
||||
return "context-keys-type" |
||||
} |
||||
|
||||
type lintContextKeyTypes struct { |
||||
file *lint.File |
||||
fileAst *ast.File |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w lintContextKeyTypes) Visit(n ast.Node) ast.Visitor { |
||||
switch n := n.(type) { |
||||
case *ast.CallExpr: |
||||
checkContextKeyType(w, n) |
||||
} |
||||
|
||||
return w |
||||
} |
||||
|
||||
func checkContextKeyType(w lintContextKeyTypes, x *ast.CallExpr) { |
||||
f := w.file |
||||
sel, ok := x.Fun.(*ast.SelectorExpr) |
||||
if !ok { |
||||
return |
||||
} |
||||
pkg, ok := sel.X.(*ast.Ident) |
||||
if !ok || pkg.Name != "context" { |
||||
return |
||||
} |
||||
if sel.Sel.Name != "WithValue" { |
||||
return |
||||
} |
||||
|
||||
// key is second argument to context.WithValue
|
||||
if len(x.Args) != 3 { |
||||
return |
||||
} |
||||
key := f.Pkg.TypesInfo.Types[x.Args[1]] |
||||
|
||||
if ktyp, ok := key.Type.(*types.Basic); ok && ktyp.Kind() != types.Invalid { |
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Node: x, |
||||
Category: "content", |
||||
Failure: fmt.Sprintf("should not use basic type %s as key in context.WithValue", key.Type), |
||||
}) |
||||
} |
||||
} |
||||
@ -0,0 +1,115 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"fmt" |
||||
"go/ast" |
||||
"go/token" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// Based on https://github.com/fzipp/gocyclo
|
||||
|
||||
// CyclomaticRule lints given else constructs.
|
||||
type CyclomaticRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *CyclomaticRule) Apply(file *lint.File, arguments lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
complexity, ok := arguments[0].(int64) // Alt. non panicking version
|
||||
if !ok { |
||||
panic("invalid argument for cyclomatic complexity") |
||||
} |
||||
|
||||
fileAst := file.AST |
||||
walker := lintCyclomatic{ |
||||
file: file, |
||||
complexity: int(complexity), |
||||
onFailure: func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
}, |
||||
} |
||||
|
||||
ast.Walk(walker, fileAst) |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *CyclomaticRule) Name() string { |
||||
return "cyclomatic" |
||||
} |
||||
|
||||
type lintCyclomatic struct { |
||||
file *lint.File |
||||
complexity int |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w lintCyclomatic) Visit(_ ast.Node) ast.Visitor { |
||||
f := w.file |
||||
for _, decl := range f.AST.Decls { |
||||
if fn, ok := decl.(*ast.FuncDecl); ok { |
||||
c := complexity(fn) |
||||
if c > w.complexity { |
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Category: "maintenance", |
||||
Failure: fmt.Sprintf("function %s has cyclomatic complexity %d", funcName(fn), c), |
||||
Node: fn, |
||||
}) |
||||
} |
||||
} |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
// funcName returns the name representation of a function or method:
|
||||
// "(Type).Name" for methods or simply "Name" for functions.
|
||||
func funcName(fn *ast.FuncDecl) string { |
||||
if fn.Recv != nil { |
||||
if fn.Recv.NumFields() > 0 { |
||||
typ := fn.Recv.List[0].Type |
||||
return fmt.Sprintf("(%s).%s", recvString(typ), fn.Name) |
||||
} |
||||
} |
||||
return fn.Name.Name |
||||
} |
||||
|
||||
// recvString returns a string representation of recv of the
|
||||
// form "T", "*T", or "BADRECV" (if not a proper receiver type).
|
||||
func recvString(recv ast.Expr) string { |
||||
switch t := recv.(type) { |
||||
case *ast.Ident: |
||||
return t.Name |
||||
case *ast.StarExpr: |
||||
return "*" + recvString(t.X) |
||||
} |
||||
return "BADRECV" |
||||
} |
||||
|
||||
// complexity calculates the cyclomatic complexity of a function.
|
||||
func complexity(fn *ast.FuncDecl) int { |
||||
v := complexityVisitor{} |
||||
ast.Walk(&v, fn) |
||||
return v.Complexity |
||||
} |
||||
|
||||
type complexityVisitor struct { |
||||
// Complexity is the cyclomatic complexity
|
||||
Complexity int |
||||
} |
||||
|
||||
// Visit implements the ast.Visitor interface.
|
||||
func (v *complexityVisitor) Visit(n ast.Node) ast.Visitor { |
||||
switch n := n.(type) { |
||||
case *ast.FuncDecl, *ast.IfStmt, *ast.ForStmt, *ast.RangeStmt, *ast.CaseClause, *ast.CommClause: |
||||
v.Complexity++ |
||||
case *ast.BinaryExpr: |
||||
if n.Op == token.LAND || n.Op == token.LOR { |
||||
v.Complexity++ |
||||
} |
||||
} |
||||
return v |
||||
} |
||||
@ -0,0 +1,94 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"fmt" |
||||
"go/ast" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// DeepExitRule lints program exit at functions other than main or init.
|
||||
type DeepExitRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *DeepExitRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
onFailure := func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
} |
||||
|
||||
var exitFunctions = map[string]map[string]bool{ |
||||
"os": map[string]bool{"Exit": true}, |
||||
"syscall": map[string]bool{"Exit": true}, |
||||
"log": map[string]bool{ |
||||
"Fatal": true, |
||||
"Fatalf": true, |
||||
"Fatalln": true, |
||||
"Panic": true, |
||||
"Panicf": true, |
||||
"Panicln": true, |
||||
}, |
||||
} |
||||
|
||||
w := lintDeepExit{onFailure, exitFunctions, file.IsTest()} |
||||
ast.Walk(w, file.AST) |
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *DeepExitRule) Name() string { |
||||
return "deep-exit" |
||||
} |
||||
|
||||
type lintDeepExit struct { |
||||
onFailure func(lint.Failure) |
||||
exitFunctions map[string]map[string]bool |
||||
isTestFile bool |
||||
} |
||||
|
||||
func (w lintDeepExit) Visit(node ast.Node) ast.Visitor { |
||||
if fd, ok := node.(*ast.FuncDecl); ok { |
||||
if w.mustIgnore(fd) { |
||||
return nil // skip analysis of this function
|
||||
} |
||||
|
||||
return w |
||||
} |
||||
|
||||
se, ok := node.(*ast.ExprStmt) |
||||
if !ok { |
||||
return w |
||||
} |
||||
ce, ok := se.X.(*ast.CallExpr) |
||||
if !ok { |
||||
return w |
||||
} |
||||
|
||||
fc, ok := ce.Fun.(*ast.SelectorExpr) |
||||
if !ok { |
||||
return w |
||||
} |
||||
id, ok := fc.X.(*ast.Ident) |
||||
if !ok { |
||||
return w |
||||
} |
||||
|
||||
fn := fc.Sel.Name |
||||
pkg := id.Name |
||||
if w.exitFunctions[pkg] != nil && w.exitFunctions[pkg][fn] { // it's a call to an exit function
|
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Node: ce, |
||||
Category: "bad practice", |
||||
Failure: fmt.Sprintf("calls to %s.%s only in main() or init() functions", pkg, fn), |
||||
}) |
||||
} |
||||
|
||||
return w |
||||
} |
||||
|
||||
func (w *lintDeepExit) mustIgnore(fd *ast.FuncDecl) bool { |
||||
fn := fd.Name.Name |
||||
|
||||
return fn == "init" || fn == "main" || (w.isTestFile && fn == "TestMain") |
||||
} |
||||
@ -0,0 +1,54 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"go/ast" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// DotImportsRule lints given else constructs.
|
||||
type DotImportsRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *DotImportsRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
fileAst := file.AST |
||||
walker := lintImports{ |
||||
file: file, |
||||
fileAst: fileAst, |
||||
onFailure: func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
}, |
||||
} |
||||
|
||||
ast.Walk(walker, fileAst) |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *DotImportsRule) Name() string { |
||||
return "dot-imports" |
||||
} |
||||
|
||||
type lintImports struct { |
||||
file *lint.File |
||||
fileAst *ast.File |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w lintImports) Visit(_ ast.Node) ast.Visitor { |
||||
for i, is := range w.fileAst.Imports { |
||||
_ = i |
||||
if is.Name != nil && is.Name.Name == "." && !w.file.IsTest() { |
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Failure: "should not use dot imports", |
||||
Node: is, |
||||
Category: "imports", |
||||
}) |
||||
} |
||||
} |
||||
return nil |
||||
} |
||||
@ -0,0 +1,39 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"fmt" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// DuplicatedImportsRule lints given else constructs.
|
||||
type DuplicatedImportsRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *DuplicatedImportsRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
impPaths := map[string]struct{}{} |
||||
for _, imp := range file.AST.Imports { |
||||
path := imp.Path.Value |
||||
_, ok := impPaths[path] |
||||
if ok { |
||||
failures = append(failures, lint.Failure{ |
||||
Confidence: 1, |
||||
Failure: fmt.Sprintf("Package %s already imported", path), |
||||
Node: imp, |
||||
Category: "imports", |
||||
}) |
||||
continue |
||||
} |
||||
|
||||
impPaths[path] = struct{}{} |
||||
} |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *DuplicatedImportsRule) Name() string { |
||||
return "duplicated-imports" |
||||
} |
||||
@ -0,0 +1,76 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"go/ast" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// EmptyBlockRule lints given else constructs.
|
||||
type EmptyBlockRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *EmptyBlockRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
onFailure := func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
} |
||||
|
||||
w := lintEmptyBlock{make([]*ast.BlockStmt, 0), onFailure} |
||||
ast.Walk(w, file.AST) |
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *EmptyBlockRule) Name() string { |
||||
return "empty-block" |
||||
} |
||||
|
||||
type lintEmptyBlock struct { |
||||
ignore []*ast.BlockStmt |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w lintEmptyBlock) Visit(node ast.Node) ast.Visitor { |
||||
fd, ok := node.(*ast.FuncDecl) |
||||
if ok { |
||||
w.ignore = append(w.ignore, fd.Body) |
||||
return w |
||||
} |
||||
|
||||
fl, ok := node.(*ast.FuncLit) |
||||
if ok { |
||||
w.ignore = append(w.ignore, fl.Body) |
||||
return w |
||||
} |
||||
|
||||
block, ok := node.(*ast.BlockStmt) |
||||
if !ok { |
||||
return w |
||||
} |
||||
|
||||
if mustIgnore(block, w.ignore) { |
||||
return w |
||||
} |
||||
|
||||
if len(block.List) == 0 { |
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Node: block, |
||||
Category: "logic", |
||||
Failure: "this block is empty, you can remove it", |
||||
}) |
||||
} |
||||
|
||||
return w |
||||
} |
||||
|
||||
func mustIgnore(block *ast.BlockStmt, blackList []*ast.BlockStmt) bool { |
||||
for _, b := range blackList { |
||||
if b == block { |
||||
return true |
||||
} |
||||
} |
||||
return false |
||||
} |
||||
@ -0,0 +1,113 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"go/ast" |
||||
"go/token" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// EmptyLinesRule lints empty lines in blocks.
|
||||
type EmptyLinesRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *EmptyLinesRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
onFailure := func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
} |
||||
|
||||
w := lintEmptyLines{file, file.CommentMap(), onFailure} |
||||
ast.Walk(w, file.AST) |
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *EmptyLinesRule) Name() string { |
||||
return "empty-lines" |
||||
} |
||||
|
||||
type lintEmptyLines struct { |
||||
file *lint.File |
||||
cmap ast.CommentMap |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w lintEmptyLines) Visit(node ast.Node) ast.Visitor { |
||||
block, ok := node.(*ast.BlockStmt) |
||||
if !ok { |
||||
return w |
||||
} |
||||
|
||||
w.checkStart(block) |
||||
w.checkEnd(block) |
||||
|
||||
return w |
||||
} |
||||
|
||||
func (w lintEmptyLines) checkStart(block *ast.BlockStmt) { |
||||
if len(block.List) == 0 { |
||||
return |
||||
} |
||||
|
||||
start := w.position(block.Lbrace) |
||||
firstNode := block.List[0] |
||||
|
||||
if w.commentBetween(start, firstNode) { |
||||
return |
||||
} |
||||
|
||||
first := w.position(firstNode.Pos()) |
||||
if first.Line-start.Line > 1 { |
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Node: block, |
||||
Category: "style", |
||||
Failure: "extra empty line at the start of a block", |
||||
}) |
||||
} |
||||
} |
||||
|
||||
func (w lintEmptyLines) checkEnd(block *ast.BlockStmt) { |
||||
if len(block.List) < 1 { |
||||
return |
||||
} |
||||
|
||||
end := w.position(block.Rbrace) |
||||
lastNode := block.List[len(block.List)-1] |
||||
|
||||
if w.commentBetween(end, lastNode) { |
||||
return |
||||
} |
||||
|
||||
last := w.position(lastNode.End()) |
||||
if end.Line-last.Line > 1 { |
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Node: lastNode, |
||||
Category: "style", |
||||
Failure: "extra empty line at the end of a block", |
||||
}) |
||||
} |
||||
} |
||||
|
||||
func (w lintEmptyLines) commentBetween(position token.Position, node ast.Node) bool { |
||||
comments := w.cmap.Filter(node).Comments() |
||||
if len(comments) == 0 { |
||||
return false |
||||
} |
||||
|
||||
for _, comment := range comments { |
||||
start, end := w.position(comment.Pos()), w.position(comment.End()) |
||||
if start.Line-position.Line == 1 || position.Line-end.Line == 1 { |
||||
return true |
||||
} |
||||
} |
||||
|
||||
return false |
||||
} |
||||
|
||||
func (w lintEmptyLines) position(pos token.Pos) token.Position { |
||||
return w.file.ToPosition(pos) |
||||
} |
||||
@ -0,0 +1,79 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"fmt" |
||||
"go/ast" |
||||
"go/token" |
||||
"strings" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// ErrorNamingRule lints given else constructs.
|
||||
type ErrorNamingRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *ErrorNamingRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
fileAst := file.AST |
||||
walker := lintErrors{ |
||||
file: file, |
||||
fileAst: fileAst, |
||||
onFailure: func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
}, |
||||
} |
||||
|
||||
ast.Walk(walker, fileAst) |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *ErrorNamingRule) Name() string { |
||||
return "error-naming" |
||||
} |
||||
|
||||
type lintErrors struct { |
||||
file *lint.File |
||||
fileAst *ast.File |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w lintErrors) Visit(_ ast.Node) ast.Visitor { |
||||
for _, decl := range w.fileAst.Decls { |
||||
gd, ok := decl.(*ast.GenDecl) |
||||
if !ok || gd.Tok != token.VAR { |
||||
continue |
||||
} |
||||
for _, spec := range gd.Specs { |
||||
spec := spec.(*ast.ValueSpec) |
||||
if len(spec.Names) != 1 || len(spec.Values) != 1 { |
||||
continue |
||||
} |
||||
ce, ok := spec.Values[0].(*ast.CallExpr) |
||||
if !ok { |
||||
continue |
||||
} |
||||
if !isPkgDot(ce.Fun, "errors", "New") && !isPkgDot(ce.Fun, "fmt", "Errorf") { |
||||
continue |
||||
} |
||||
|
||||
id := spec.Names[0] |
||||
prefix := "err" |
||||
if id.IsExported() { |
||||
prefix = "Err" |
||||
} |
||||
if !strings.HasPrefix(id.Name, prefix) { |
||||
w.onFailure(lint.Failure{ |
||||
Node: id, |
||||
Confidence: 0.9, |
||||
Category: "naming", |
||||
Failure: fmt.Sprintf("error var %s should have name of the form %sFoo", id.Name, prefix), |
||||
}) |
||||
} |
||||
} |
||||
} |
||||
return nil |
||||
} |
||||
@ -0,0 +1,67 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"go/ast" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// ErrorReturnRule lints given else constructs.
|
||||
type ErrorReturnRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *ErrorReturnRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
fileAst := file.AST |
||||
walker := lintErrorReturn{ |
||||
file: file, |
||||
fileAst: fileAst, |
||||
onFailure: func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
}, |
||||
} |
||||
|
||||
ast.Walk(walker, fileAst) |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *ErrorReturnRule) Name() string { |
||||
return "error-return" |
||||
} |
||||
|
||||
type lintErrorReturn struct { |
||||
file *lint.File |
||||
fileAst *ast.File |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w lintErrorReturn) Visit(n ast.Node) ast.Visitor { |
||||
fn, ok := n.(*ast.FuncDecl) |
||||
if !ok || fn.Type.Results == nil { |
||||
return w |
||||
} |
||||
ret := fn.Type.Results.List |
||||
if len(ret) <= 1 { |
||||
return w |
||||
} |
||||
if isIdent(ret[len(ret)-1].Type, "error") { |
||||
return nil |
||||
} |
||||
// An error return parameter should be the last parameter.
|
||||
// Flag any error parameters found before the last.
|
||||
for _, r := range ret[:len(ret)-1] { |
||||
if isIdent(r.Type, "error") { |
||||
w.onFailure(lint.Failure{ |
||||
Category: "arg-order", |
||||
Confidence: 0.9, |
||||
Node: fn, |
||||
Failure: "error should be the last type when returning multiple items", |
||||
}) |
||||
break // only flag one
|
||||
} |
||||
} |
||||
return w |
||||
} |
||||
@ -0,0 +1,98 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"go/ast" |
||||
"go/token" |
||||
"strconv" |
||||
"unicode" |
||||
"unicode/utf8" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// ErrorStringsRule lints given else constructs.
|
||||
type ErrorStringsRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *ErrorStringsRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
fileAst := file.AST |
||||
walker := lintErrorStrings{ |
||||
file: file, |
||||
fileAst: fileAst, |
||||
onFailure: func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
}, |
||||
} |
||||
|
||||
ast.Walk(walker, fileAst) |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *ErrorStringsRule) Name() string { |
||||
return "error-strings" |
||||
} |
||||
|
||||
type lintErrorStrings struct { |
||||
file *lint.File |
||||
fileAst *ast.File |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w lintErrorStrings) Visit(n ast.Node) ast.Visitor { |
||||
ce, ok := n.(*ast.CallExpr) |
||||
if !ok { |
||||
return w |
||||
} |
||||
if !isPkgDot(ce.Fun, "errors", "New") && !isPkgDot(ce.Fun, "fmt", "Errorf") { |
||||
return w |
||||
} |
||||
if len(ce.Args) < 1 { |
||||
return w |
||||
} |
||||
str, ok := ce.Args[0].(*ast.BasicLit) |
||||
if !ok || str.Kind != token.STRING { |
||||
return w |
||||
} |
||||
s, _ := strconv.Unquote(str.Value) // can assume well-formed Go
|
||||
if s == "" { |
||||
return w |
||||
} |
||||
clean, conf := lintErrorString(s) |
||||
if clean { |
||||
return w |
||||
} |
||||
|
||||
w.onFailure(lint.Failure{ |
||||
Node: str, |
||||
Confidence: conf, |
||||
Category: "errors", |
||||
Failure: "error strings should not be capitalized or end with punctuation or a newline", |
||||
}) |
||||
return w |
||||
} |
||||
|
||||
func lintErrorString(s string) (isClean bool, conf float64) { |
||||
const basicConfidence = 0.8 |
||||
const capConfidence = basicConfidence - 0.2 |
||||
first, firstN := utf8.DecodeRuneInString(s) |
||||
last, _ := utf8.DecodeLastRuneInString(s) |
||||
if last == '.' || last == ':' || last == '!' || last == '\n' { |
||||
return false, basicConfidence |
||||
} |
||||
if unicode.IsUpper(first) { |
||||
// People use proper nouns and exported Go identifiers in error strings,
|
||||
// so decrease the confidence of warnings for capitalization.
|
||||
if len(s) <= firstN { |
||||
return false, capConfidence |
||||
} |
||||
// Flag strings starting with something that doesn't look like an initialism.
|
||||
if second, _ := utf8.DecodeRuneInString(s[firstN:]); !unicode.IsUpper(second) { |
||||
return false, capConfidence |
||||
} |
||||
} |
||||
return true, 0 |
||||
} |
||||
@ -0,0 +1,93 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"fmt" |
||||
"go/ast" |
||||
"regexp" |
||||
"strings" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// ErrorfRule lints given else constructs.
|
||||
type ErrorfRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *ErrorfRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
fileAst := file.AST |
||||
walker := lintErrorf{ |
||||
file: file, |
||||
fileAst: fileAst, |
||||
onFailure: func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
}, |
||||
} |
||||
|
||||
file.Pkg.TypeCheck() |
||||
ast.Walk(walker, fileAst) |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *ErrorfRule) Name() string { |
||||
return "errorf" |
||||
} |
||||
|
||||
type lintErrorf struct { |
||||
file *lint.File |
||||
fileAst *ast.File |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w lintErrorf) Visit(n ast.Node) ast.Visitor { |
||||
ce, ok := n.(*ast.CallExpr) |
||||
if !ok || len(ce.Args) != 1 { |
||||
return w |
||||
} |
||||
isErrorsNew := isPkgDot(ce.Fun, "errors", "New") |
||||
var isTestingError bool |
||||
se, ok := ce.Fun.(*ast.SelectorExpr) |
||||
if ok && se.Sel.Name == "Error" { |
||||
if typ := w.file.Pkg.TypeOf(se.X); typ != nil { |
||||
isTestingError = typ.String() == "*testing.T" |
||||
} |
||||
} |
||||
if !isErrorsNew && !isTestingError { |
||||
return w |
||||
} |
||||
arg := ce.Args[0] |
||||
ce, ok = arg.(*ast.CallExpr) |
||||
if !ok || !isPkgDot(ce.Fun, "fmt", "Sprintf") { |
||||
return w |
||||
} |
||||
errorfPrefix := "fmt" |
||||
if isTestingError { |
||||
errorfPrefix = w.file.Render(se.X) |
||||
} |
||||
|
||||
failure := lint.Failure{ |
||||
Category: "errors", |
||||
Node: n, |
||||
Confidence: 1, |
||||
Failure: fmt.Sprintf("should replace %s(fmt.Sprintf(...)) with %s.Errorf(...)", w.file.Render(se), errorfPrefix), |
||||
} |
||||
|
||||
m := srcLineWithMatch(w.file, ce, `^(.*)`+w.file.Render(se)+`\(fmt\.Sprintf\((.*)\)\)(.*)$`) |
||||
if m != nil { |
||||
failure.ReplacementLine = m[1] + errorfPrefix + ".Errorf(" + m[2] + ")" + m[3] |
||||
} |
||||
|
||||
w.onFailure(failure) |
||||
|
||||
return w |
||||
} |
||||
|
||||
func srcLineWithMatch(file *lint.File, node ast.Node, pattern string) (m []string) { |
||||
line := srcLine(file.Content(), file.ToPosition(node.Pos())) |
||||
line = strings.TrimSuffix(line, "\n") |
||||
rx := regexp.MustCompile(pattern) |
||||
return rx.FindStringSubmatch(line) |
||||
} |
||||
@ -0,0 +1,272 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"fmt" |
||||
"go/ast" |
||||
"go/token" |
||||
"strings" |
||||
"unicode" |
||||
"unicode/utf8" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// ExportedRule lints given else constructs.
|
||||
type ExportedRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *ExportedRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
if isTest(file) { |
||||
return failures |
||||
} |
||||
|
||||
fileAst := file.AST |
||||
walker := lintExported{ |
||||
file: file, |
||||
fileAst: fileAst, |
||||
onFailure: func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
}, |
||||
genDeclMissingComments: make(map[*ast.GenDecl]bool), |
||||
} |
||||
|
||||
ast.Walk(&walker, fileAst) |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *ExportedRule) Name() string { |
||||
return "exported" |
||||
} |
||||
|
||||
type lintExported struct { |
||||
file *lint.File |
||||
fileAst *ast.File |
||||
lastGen *ast.GenDecl |
||||
genDeclMissingComments map[*ast.GenDecl]bool |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w *lintExported) lintFuncDoc(fn *ast.FuncDecl) { |
||||
if !ast.IsExported(fn.Name.Name) { |
||||
// func is unexported
|
||||
return |
||||
} |
||||
kind := "function" |
||||
name := fn.Name.Name |
||||
if fn.Recv != nil && len(fn.Recv.List) > 0 { |
||||
// method
|
||||
kind = "method" |
||||
recv := receiverType(fn) |
||||
if !ast.IsExported(recv) { |
||||
// receiver is unexported
|
||||
return |
||||
} |
||||
if commonMethods[name] { |
||||
return |
||||
} |
||||
switch name { |
||||
case "Len", "Less", "Swap": |
||||
if w.file.Pkg.Sortable[recv] { |
||||
return |
||||
} |
||||
} |
||||
name = recv + "." + name |
||||
} |
||||
if fn.Doc == nil { |
||||
w.onFailure(lint.Failure{ |
||||
Node: fn, |
||||
Confidence: 1, |
||||
Category: "comments", |
||||
Failure: fmt.Sprintf("exported %s %s should have comment or be unexported", kind, name), |
||||
}) |
||||
return |
||||
} |
||||
s := normalizeText(fn.Doc.Text()) |
||||
prefix := fn.Name.Name + " " |
||||
if !strings.HasPrefix(s, prefix) { |
||||
w.onFailure(lint.Failure{ |
||||
Node: fn.Doc, |
||||
Confidence: 0.8, |
||||
Category: "comments", |
||||
Failure: fmt.Sprintf(`comment on exported %s %s should be of the form "%s..."`, kind, name, prefix), |
||||
}) |
||||
} |
||||
} |
||||
|
||||
func (w *lintExported) checkStutter(id *ast.Ident, thing string) { |
||||
pkg, name := w.fileAst.Name.Name, id.Name |
||||
if !ast.IsExported(name) { |
||||
// unexported name
|
||||
return |
||||
} |
||||
// A name stutters if the package name is a strict prefix
|
||||
// and the next character of the name starts a new word.
|
||||
if len(name) <= len(pkg) { |
||||
// name is too short to stutter.
|
||||
// This permits the name to be the same as the package name.
|
||||
return |
||||
} |
||||
if !strings.EqualFold(pkg, name[:len(pkg)]) { |
||||
return |
||||
} |
||||
// We can assume the name is well-formed UTF-8.
|
||||
// If the next rune after the package name is uppercase or an underscore
|
||||
// the it's starting a new word and thus this name stutters.
|
||||
rem := name[len(pkg):] |
||||
if next, _ := utf8.DecodeRuneInString(rem); next == '_' || unicode.IsUpper(next) { |
||||
w.onFailure(lint.Failure{ |
||||
Node: id, |
||||
Confidence: 0.8, |
||||
Category: "naming", |
||||
Failure: fmt.Sprintf("%s name will be used as %s.%s by other packages, and that stutters; consider calling this %s", thing, pkg, name, rem), |
||||
}) |
||||
} |
||||
} |
||||
|
||||
func (w *lintExported) lintTypeDoc(t *ast.TypeSpec, doc *ast.CommentGroup) { |
||||
if !ast.IsExported(t.Name.Name) { |
||||
return |
||||
} |
||||
if doc == nil { |
||||
w.onFailure(lint.Failure{ |
||||
Node: t, |
||||
Confidence: 1, |
||||
Category: "comments", |
||||
Failure: fmt.Sprintf("exported type %v should have comment or be unexported", t.Name), |
||||
}) |
||||
return |
||||
} |
||||
|
||||
s := normalizeText(doc.Text()) |
||||
articles := [...]string{"A", "An", "The", "This"} |
||||
for _, a := range articles { |
||||
if t.Name.Name == a { |
||||
continue |
||||
} |
||||
if strings.HasPrefix(s, a+" ") { |
||||
s = s[len(a)+1:] |
||||
break |
||||
} |
||||
} |
||||
if !strings.HasPrefix(s, t.Name.Name+" ") { |
||||
w.onFailure(lint.Failure{ |
||||
Node: doc, |
||||
Confidence: 1, |
||||
Category: "comments", |
||||
Failure: fmt.Sprintf(`comment on exported type %v should be of the form "%v ..." (with optional leading article)`, t.Name, t.Name), |
||||
}) |
||||
} |
||||
} |
||||
|
||||
func (w *lintExported) lintValueSpecDoc(vs *ast.ValueSpec, gd *ast.GenDecl, genDeclMissingComments map[*ast.GenDecl]bool) { |
||||
kind := "var" |
||||
if gd.Tok == token.CONST { |
||||
kind = "const" |
||||
} |
||||
|
||||
if len(vs.Names) > 1 { |
||||
// Check that none are exported except for the first.
|
||||
for _, n := range vs.Names[1:] { |
||||
if ast.IsExported(n.Name) { |
||||
w.onFailure(lint.Failure{ |
||||
Category: "comments", |
||||
Confidence: 1, |
||||
Failure: fmt.Sprintf("exported %s %s should have its own declaration", kind, n.Name), |
||||
Node: vs, |
||||
}) |
||||
return |
||||
} |
||||
} |
||||
} |
||||
|
||||
// Only one name.
|
||||
name := vs.Names[0].Name |
||||
if !ast.IsExported(name) { |
||||
return |
||||
} |
||||
|
||||
if vs.Doc == nil && gd.Doc == nil { |
||||
if genDeclMissingComments[gd] { |
||||
return |
||||
} |
||||
block := "" |
||||
if kind == "const" && gd.Lparen.IsValid() { |
||||
block = " (or a comment on this block)" |
||||
} |
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Node: vs, |
||||
Category: "comments", |
||||
Failure: fmt.Sprintf("exported %s %s should have comment%s or be unexported", kind, name, block), |
||||
}) |
||||
genDeclMissingComments[gd] = true |
||||
return |
||||
} |
||||
// If this GenDecl has parens and a comment, we don't check its comment form.
|
||||
if gd.Lparen.IsValid() && gd.Doc != nil { |
||||
return |
||||
} |
||||
// The relevant text to check will be on either vs.Doc or gd.Doc.
|
||||
// Use vs.Doc preferentially.
|
||||
doc := vs.Doc |
||||
if doc == nil { |
||||
doc = gd.Doc |
||||
} |
||||
prefix := name + " " |
||||
s := normalizeText(doc.Text()) |
||||
if !strings.HasPrefix(s, prefix) { |
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Node: doc, |
||||
Category: "comments", |
||||
Failure: fmt.Sprintf(`comment on exported %s %s should be of the form "%s..."`, kind, name, prefix), |
||||
}) |
||||
} |
||||
} |
||||
|
||||
// normalizeText is a helper function that normalizes comment strings by:
|
||||
// * removing one leading space
|
||||
//
|
||||
// This function is needed because ast.CommentGroup.Text() does not handle //-style and /*-style comments uniformly
|
||||
func normalizeText(t string) string { |
||||
return strings.TrimPrefix(t, " ") |
||||
} |
||||
|
||||
func (w *lintExported) Visit(n ast.Node) ast.Visitor { |
||||
switch v := n.(type) { |
||||
case *ast.GenDecl: |
||||
if v.Tok == token.IMPORT { |
||||
return nil |
||||
} |
||||
// token.CONST, token.TYPE or token.VAR
|
||||
w.lastGen = v |
||||
return w |
||||
case *ast.FuncDecl: |
||||
w.lintFuncDoc(v) |
||||
if v.Recv == nil { |
||||
// Only check for stutter on functions, not methods.
|
||||
// Method names are not used package-qualified.
|
||||
w.checkStutter(v.Name, "func") |
||||
} |
||||
// Don't proceed inside funcs.
|
||||
return nil |
||||
case *ast.TypeSpec: |
||||
// inside a GenDecl, which usually has the doc
|
||||
doc := v.Doc |
||||
if doc == nil { |
||||
doc = w.lastGen.Doc |
||||
} |
||||
w.lintTypeDoc(v, doc) |
||||
w.checkStutter(v.Name, "type") |
||||
// Don't proceed inside types.
|
||||
return nil |
||||
case *ast.ValueSpec: |
||||
w.lintValueSpecDoc(v, w.lastGen, w.genDeclMissingComments) |
||||
return nil |
||||
} |
||||
return w |
||||
} |
||||
@ -0,0 +1,69 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"regexp" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// FileHeaderRule lints given else constructs.
|
||||
type FileHeaderRule struct{} |
||||
|
||||
var ( |
||||
multiRegexp = regexp.MustCompile("^/\\*") |
||||
singleRegexp = regexp.MustCompile("^//") |
||||
) |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *FileHeaderRule) Apply(file *lint.File, arguments lint.Arguments) []lint.Failure { |
||||
if len(arguments) != 1 { |
||||
panic(`invalid configuration for "file-header" rule`) |
||||
} |
||||
|
||||
header, ok := arguments[0].(string) |
||||
if !ok { |
||||
panic(`invalid argument for "file-header" rule: first argument should be a string`) |
||||
} |
||||
|
||||
failure := []lint.Failure{ |
||||
{ |
||||
Node: file.AST, |
||||
Confidence: 1, |
||||
Failure: "the file doesn't have an appropriate header", |
||||
}, |
||||
} |
||||
|
||||
if len(file.AST.Comments) == 0 { |
||||
return failure |
||||
} |
||||
|
||||
g := file.AST.Comments[0] |
||||
if g == nil { |
||||
return failure |
||||
} |
||||
comment := "" |
||||
for _, c := range g.List { |
||||
text := c.Text |
||||
if multiRegexp.Match([]byte(text)) { |
||||
text = text[2 : len(text)-2] |
||||
} else if singleRegexp.Match([]byte(text)) { |
||||
text = text[2:] |
||||
} |
||||
comment += text |
||||
} |
||||
|
||||
regex, err := regexp.Compile(header) |
||||
if err != nil { |
||||
panic(err.Error()) |
||||
} |
||||
|
||||
if !regex.Match([]byte(comment)) { |
||||
return failure |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *FileHeaderRule) Name() string { |
||||
return "file-header" |
||||
} |
||||
@ -0,0 +1,104 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"fmt" |
||||
"github.com/mgechev/revive/lint" |
||||
"go/ast" |
||||
) |
||||
|
||||
// FlagParamRule lints given else constructs.
|
||||
type FlagParamRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *FlagParamRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
onFailure := func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
} |
||||
|
||||
w := lintFlagParamRule{onFailure: onFailure} |
||||
ast.Walk(w, file.AST) |
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *FlagParamRule) Name() string { |
||||
return "flag-parameter" |
||||
} |
||||
|
||||
type lintFlagParamRule struct { |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w lintFlagParamRule) Visit(node ast.Node) ast.Visitor { |
||||
fd, ok := node.(*ast.FuncDecl) |
||||
if !ok { |
||||
return w |
||||
} |
||||
|
||||
if fd.Body == nil { |
||||
return nil // skip whole function declaration
|
||||
} |
||||
|
||||
for _, p := range fd.Type.Params.List { |
||||
t := p.Type |
||||
|
||||
id, ok := t.(*ast.Ident) |
||||
if !ok { |
||||
continue |
||||
} |
||||
|
||||
if id.Name != "bool" { |
||||
continue |
||||
} |
||||
|
||||
cv := conditionVisitor{p.Names, fd, w} |
||||
ast.Walk(cv, fd.Body) |
||||
} |
||||
|
||||
return w |
||||
} |
||||
|
||||
type conditionVisitor struct { |
||||
ids []*ast.Ident |
||||
fd *ast.FuncDecl |
||||
linter lintFlagParamRule |
||||
} |
||||
|
||||
func (w conditionVisitor) Visit(node ast.Node) ast.Visitor { |
||||
ifStmt, ok := node.(*ast.IfStmt) |
||||
if !ok { |
||||
return w |
||||
} |
||||
|
||||
fselect := func(n ast.Node) bool { |
||||
ident, ok := n.(*ast.Ident) |
||||
if !ok { |
||||
return false |
||||
} |
||||
|
||||
for _, id := range w.ids { |
||||
if ident.Name == id.Name { |
||||
return true |
||||
} |
||||
} |
||||
|
||||
return false |
||||
} |
||||
|
||||
uses := pick(ifStmt.Cond, fselect, nil) |
||||
|
||||
if len(uses) < 1 { |
||||
return w |
||||
} |
||||
|
||||
w.linter.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Node: w.fd.Type.Params, |
||||
Category: "bad practice", |
||||
Failure: fmt.Sprintf("parameter '%s' seems to be a control flag, avoid control coupling", uses[0]), |
||||
}) |
||||
|
||||
return nil |
||||
} |
||||
@ -0,0 +1,68 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"fmt" |
||||
"go/ast" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// FunctionResultsLimitRule lints given else constructs.
|
||||
type FunctionResultsLimitRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *FunctionResultsLimitRule) Apply(file *lint.File, arguments lint.Arguments) []lint.Failure { |
||||
if len(arguments) != 1 { |
||||
panic(`invalid configuration for "function-result-limit"`) |
||||
} |
||||
|
||||
max, ok := arguments[0].(int64) // Alt. non panicking version
|
||||
if !ok { |
||||
panic(fmt.Sprintf(`invalid value passed as return results number to the "function-result-limit" rule; need int64 but got %T`, arguments[0])) |
||||
} |
||||
if max < 0 { |
||||
panic(`the value passed as return results number to the "function-result-limit" rule cannot be negative`) |
||||
} |
||||
|
||||
var failures []lint.Failure |
||||
|
||||
walker := lintFunctionResultsNum{ |
||||
max: int(max), |
||||
onFailure: func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
}, |
||||
} |
||||
|
||||
ast.Walk(walker, file.AST) |
||||
|
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *FunctionResultsLimitRule) Name() string { |
||||
return "function-result-limit" |
||||
} |
||||
|
||||
type lintFunctionResultsNum struct { |
||||
max int |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func (w lintFunctionResultsNum) Visit(n ast.Node) ast.Visitor { |
||||
node, ok := n.(*ast.FuncDecl) |
||||
if ok { |
||||
num := 0 |
||||
if node.Type.Results != nil { |
||||
num = node.Type.Results.NumFields() |
||||
} |
||||
if num > w.max { |
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 1, |
||||
Failure: fmt.Sprintf("maximum number of return results per function exceeded; max %d but got %d", w.max, num), |
||||
Node: node.Type, |
||||
}) |
||||
return w |
||||
} |
||||
} |
||||
return w |
||||
} |
||||
@ -0,0 +1,70 @@ |
||||
package rule |
||||
|
||||
import ( |
||||
"fmt" |
||||
"go/ast" |
||||
"strings" |
||||
|
||||
"github.com/mgechev/revive/lint" |
||||
) |
||||
|
||||
// GetReturnRule lints given else constructs.
|
||||
type GetReturnRule struct{} |
||||
|
||||
// Apply applies the rule to given file.
|
||||
func (r *GetReturnRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure { |
||||
var failures []lint.Failure |
||||
|
||||
onFailure := func(failure lint.Failure) { |
||||
failures = append(failures, failure) |
||||
} |
||||
|
||||
w := lintReturnRule{onFailure} |
||||
ast.Walk(w, file.AST) |
||||
return failures |
||||
} |
||||
|
||||
// Name returns the rule name.
|
||||
func (r *GetReturnRule) Name() string { |
||||
return "get-return" |
||||
} |
||||
|
||||
type lintReturnRule struct { |
||||
onFailure func(lint.Failure) |
||||
} |
||||
|
||||
func isGetter(name string) bool { |
||||
if strings.HasPrefix(strings.ToUpper(name), "GET") { |
||||
if len(name) > 3 { |
||||
c := name[3] |
||||
return !(c >= 'a' && c <= 'z') |
||||
} |
||||
} |
||||
|
||||
return false |
||||
} |
||||
|
||||
func hasResults(rs *ast.FieldList) bool { |
||||
return rs != nil && len(rs.List) > 0 |
||||
} |
||||
|
||||
func (w lintReturnRule) Visit(node ast.Node) ast.Visitor { |
||||
fd, ok := node.(*ast.FuncDecl) |
||||
if !ok { |
||||
return w |
||||
} |
||||
|
||||
if !isGetter(fd.Name.Name) { |
||||
return w |
||||
} |
||||
if !hasResults(fd.Type.Results) { |
||||
w.onFailure(lint.Failure{ |
||||
Confidence: 0.8, |
||||
Node: fd, |
||||
Category: "logic", |
||||
Failure: fmt.Sprintf("function '%s' seems to be a getter but it does not return any result", fd.Name.Name), |
||||
}) |
||||
} |
||||
|
||||
return w |
||||
} |
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue