summaryrefslogtreecommitdiffstats
path: root/vendor/gitea.com/jolheiser/gitea-vet/checks/imports.go
blob: 6ef5975e2e033e2d14de2f0f995981e2b5a628ed (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
// 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 checks

import (
	"strings"

	"golang.org/x/tools/go/analysis"
)

var Imports = &analysis.Analyzer{
	Name: "imports",
	Doc:  "check for import order.",
	Run:  runImports,
}

func runImports(pass *analysis.Pass) (interface{}, error) {
	for _, file := range pass.Files {
		level := 0
		for _, im := range file.Imports {
			var lvl int
			val := im.Path.Value
			if importHasPrefix(val, "code.gitea.io") {
				lvl = 2
			} else if strings.Contains(val, ".") {
				lvl = 3
			} else {
				lvl = 1
			}

			if lvl < level {
				pass.Reportf(file.Pos(), "Imports are sorted wrong")
				break
			}
			level = lvl
		}
	}
	return nil, nil
}

func importHasPrefix(s, p string) bool {
	return strings.HasPrefix(s, "\""+p)
}

func sliceHasPrefix(s string, prefixes ...string) bool {
	for _, p := range prefixes {
		if importHasPrefix(s, p) {
			return true
		}
	}
	return false
}