package analyzer

import (
	"flag"
	"fmt"
	"go/ast"
	"go/token"
	"strings"

	"golang.org/x/tools/go/analysis"
	"golang.org/x/tools/go/analysis/passes/inspect"
	"golang.org/x/tools/go/ast/inspector"

	"github.com/sashamelentyev/usestdlibvars/pkg/analyzer/internal/mapping"
)

const (
	TimeWeekdayFlag        = "time-weekday"
	TimeMonthFlag          = "time-month"
	TimeLayoutFlag         = "time-layout"
	CryptoHashFlag         = "crypto-hash"
	HTTPMethodFlag         = "http-method"
	HTTPStatusCodeFlag     = "http-status-code"
	RPCDefaultPathFlag     = "rpc-default-path"
	OSDevNullFlag          = "os-dev-null"
	SQLIsolationLevelFlag  = "sql-isolation-level"
	TLSSignatureSchemeFlag = "tls-signature-scheme"
	ConstantKindFlag       = "constant-kind"
	SyslogPriorityFlag     = "syslog-priority"
)

// New returns new usestdlibvars analyzer.
func New() *analysis.Analyzer {
	return &analysis.Analyzer{
		Name:     "usestdlibvars",
		Doc:      "A linter that detect the possibility to use variables/constants from the Go standard library.",
		Run:      run,
		Flags:    flags(),
		Requires: []*analysis.Analyzer{inspect.Analyzer},
	}
}

func flags() flag.FlagSet {
	flags := flag.NewFlagSet("", flag.ExitOnError)
	flags.Bool(HTTPMethodFlag, true, "suggest the use of http.MethodXX")
	flags.Bool(HTTPStatusCodeFlag, true, "suggest the use of http.StatusXX")
	flags.Bool(TimeWeekdayFlag, false, "suggest the use of time.Weekday.String()")
	flags.Bool(TimeMonthFlag, false, "suggest the use of time.Month.String()")
	flags.Bool(TimeLayoutFlag, false, "suggest the use of time.Layout")
	flags.Bool(CryptoHashFlag, false, "suggest the use of crypto.Hash.String()")
	flags.Bool(RPCDefaultPathFlag, false, "suggest the use of rpc.DefaultXXPath")
	flags.Bool(OSDevNullFlag, false, "[DEPRECATED] suggest the use of os.DevNull")
	flags.Bool(SQLIsolationLevelFlag, false, "suggest the use of sql.LevelXX.String()")
	flags.Bool(TLSSignatureSchemeFlag, false, "suggest the use of tls.SignatureScheme.String()")
	flags.Bool(ConstantKindFlag, false, "suggest the use of constant.Kind.String()")
	flags.Bool(SyslogPriorityFlag, false, "[DEPRECATED] suggest the use of syslog.Priority")
	return *flags
}

func run(pass *analysis.Pass) (interface{}, error) {
	insp := pass.ResultOf[inspect.Analyzer].(*inspector.Inspector)

	types := []ast.Node{
		(*ast.CallExpr)(nil),
		(*ast.BasicLit)(nil),
		(*ast.CompositeLit)(nil),
		(*ast.BinaryExpr)(nil),
		(*ast.SwitchStmt)(nil),
	}

	insp.Preorder(types, func(node ast.Node) {
		switch n := node.(type) {
		case *ast.CallExpr:
			fun, ok := n.Fun.(*ast.SelectorExpr)
			if !ok {
				return
			}

			x, ok := fun.X.(*ast.Ident)
			if !ok {
				return
			}

			funArgs(pass, x, fun, n.Args)

		case *ast.BasicLit:
			for _, c := range []struct {
				flag      string
				checkFunc func(pass *analysis.Pass, basicLit *ast.BasicLit)
			}{
				{flag: TimeWeekdayFlag, checkFunc: checkTimeWeekday},
				{flag: TimeMonthFlag, checkFunc: checkTimeMonth},
				{flag: TimeLayoutFlag, checkFunc: checkTimeLayout},
				{flag: CryptoHashFlag, checkFunc: checkCryptoHash},
				{flag: RPCDefaultPathFlag, checkFunc: checkRPCDefaultPath},
				{flag: OSDevNullFlag, checkFunc: checkOSDevNull},
				{flag: SQLIsolationLevelFlag, checkFunc: checkSQLIsolationLevel},
				{flag: TLSSignatureSchemeFlag, checkFunc: checkTLSSignatureScheme},
				{flag: ConstantKindFlag, checkFunc: checkConstantKind},
			} {
				if lookupFlag(pass, c.flag) {
					c.checkFunc(pass, n)
				}
			}

		case *ast.CompositeLit:
			typ, ok := n.Type.(*ast.SelectorExpr)
			if !ok {
				return
			}

			x, ok := typ.X.(*ast.Ident)
			if !ok {
				return
			}

			typeElts(pass, x, typ, n.Elts)

		case *ast.BinaryExpr:
			switch n.Op {
			case token.LSS, token.GTR, token.LEQ, token.GEQ, token.QUO, token.ADD, token.SUB, token.MUL:
				return
			default:
			}

			x, ok := n.X.(*ast.SelectorExpr)
			if !ok {
				return
			}

			y, ok := n.Y.(*ast.BasicLit)
			if !ok {
				return
			}

			binaryExpr(pass, x, y)

		case *ast.SwitchStmt:
			x, ok := n.Tag.(*ast.SelectorExpr)
			if ok {
				switchStmt(pass, x, n.Body.List)
			}
		}
	})

	return nil, nil
}

// funArgs checks arguments of function or method.
func funArgs(pass *analysis.Pass, x *ast.Ident, fun *ast.SelectorExpr, args []ast.Expr) {
	switch x.Name {
	case "http":
		switch fun.Sel.Name {
		// http.NewRequest(http.MethodGet, "localhost", http.NoBody)
		case "NewRequest":
			if !lookupFlag(pass, HTTPMethodFlag) {
				return
			}

			if basicLit := getBasicLitFromArgs(args, 3, 0, token.STRING); basicLit != nil {
				checkHTTPMethod(pass, basicLit)
			}

		// http.NewRequestWithContext(context.Background(), http.MethodGet, "localhost", http.NoBody)
		case "NewRequestWithContext":
			if !lookupFlag(pass, HTTPMethodFlag) {
				return
			}

			if basicLit := getBasicLitFromArgs(args, 4, 1, token.STRING); basicLit != nil {
				checkHTTPMethod(pass, basicLit)
			}

		// http.Error(w, err, http.StatusInternalServerError)
		case "Error":
			if !lookupFlag(pass, HTTPStatusCodeFlag) {
				return
			}

			if basicLit := getBasicLitFromArgs(args, 3, 2, token.INT); basicLit != nil {
				checkHTTPStatusCode(pass, basicLit)
			}

		// http.StatusText(http.StatusOK)
		case "StatusText":
			if !lookupFlag(pass, HTTPStatusCodeFlag) {
				return
			}

			if basicLit := getBasicLitFromArgs(args, 1, 0, token.INT); basicLit != nil {
				checkHTTPStatusCode(pass, basicLit)
			}

		// http.Redirect(w, r, "localhost", http.StatusMovedPermanently)
		case "Redirect":
			if !lookupFlag(pass, HTTPStatusCodeFlag) {
				return
			}

			if basicLit := getBasicLitFromArgs(args, 4, 3, token.INT); basicLit != nil {
				checkHTTPStatusCode(pass, basicLit)
			}

		// http.RedirectHandler("localhost", http.StatusMovedPermanently)
		case "RedirectHandler":
			if !lookupFlag(pass, HTTPStatusCodeFlag) {
				return
			}

			if basicLit := getBasicLitFromArgs(args, 2, 1, token.INT); basicLit != nil {
				checkHTTPStatusCode(pass, basicLit)
			}
		}
	case "httptest":
		if fun.Sel.Name == "NewRequest" {
			if !lookupFlag(pass, HTTPMethodFlag) {
				return
			}

			if basicLit := getBasicLitFromArgs(args, 3, 0, token.STRING); basicLit != nil {
				checkHTTPMethod(pass, basicLit)
			}
		}
	case "syslog":
		if !lookupFlag(pass, SyslogPriorityFlag) {
			return
		}

		switch fun.Sel.Name {
		case "New":
			if basicLit := getBasicLitFromArgs(args, 2, 0, token.INT); basicLit != nil {
				checkSyslogPriority(pass, basicLit)
			}

		case "Dial":
			if basicLit := getBasicLitFromArgs(args, 4, 2, token.INT); basicLit != nil {
				checkSyslogPriority(pass, basicLit)
			}

		case "NewLogger":
			if basicLit := getBasicLitFromArgs(args, 2, 0, token.INT); basicLit != nil {
				checkSyslogPriority(pass, basicLit)
			}
		}
	default:
		// w.WriteHeader(http.StatusOk)
		if fun.Sel.Name == "WriteHeader" {
			if !lookupFlag(pass, HTTPStatusCodeFlag) {
				return
			}

			if basicLit := getBasicLitFromArgs(args, 1, 0, token.INT); basicLit != nil {
				checkHTTPStatusCode(pass, basicLit)
			}
		}
	}
}

// typeElts checks elements of type.
func typeElts(pass *analysis.Pass, x *ast.Ident, typ *ast.SelectorExpr, elts []ast.Expr) {
	switch x.Name {
	case "http":
		switch typ.Sel.Name {
		// http.Request{Method: http.MethodGet}
		case "Request":
			if !lookupFlag(pass, HTTPMethodFlag) {
				return
			}

			if basicLit := getBasicLitFromElts(elts, "Method"); basicLit != nil {
				checkHTTPMethod(pass, basicLit)
			}

		// http.Response{StatusCode: http.StatusOK}
		case "Response":
			if !lookupFlag(pass, HTTPStatusCodeFlag) {
				return
			}

			if basicLit := getBasicLitFromElts(elts, "StatusCode"); basicLit != nil {
				checkHTTPStatusCode(pass, basicLit)
			}
		}
	case "httptest":
		if typ.Sel.Name == "ResponseRecorder" {
			if !lookupFlag(pass, HTTPStatusCodeFlag) {
				return
			}

			if basicLit := getBasicLitFromElts(elts, "Code"); basicLit != nil {
				checkHTTPStatusCode(pass, basicLit)
			}
		}
	}
}

// binaryExpr checks X and Y in binary expressions, including:
//   - if-else-statement
//   - for loops conditions
//   - switch cases without a tag expression
func binaryExpr(pass *analysis.Pass, x *ast.SelectorExpr, y *ast.BasicLit) {
	switch x.Sel.Name {
	case "StatusCode":
		if !lookupFlag(pass, HTTPStatusCodeFlag) {
			return
		}

		checkHTTPStatusCode(pass, y)

	case "Method":
		if !lookupFlag(pass, HTTPMethodFlag) {
			return
		}

		checkHTTPMethod(pass, y)
	}
}

func switchStmt(pass *analysis.Pass, x *ast.SelectorExpr, cases []ast.Stmt) {
	var checkFunc func(pass *analysis.Pass, basicLit *ast.BasicLit)

	switch x.Sel.Name {
	case "StatusCode":
		if !lookupFlag(pass, HTTPStatusCodeFlag) {
			return
		}

		checkFunc = checkHTTPStatusCode

	case "Method":
		if !lookupFlag(pass, HTTPMethodFlag) {
			return
		}

		checkFunc = checkHTTPMethod

	default:
		return
	}

	for _, c := range cases {
		caseClause, ok := c.(*ast.CaseClause)
		if !ok {
			continue
		}

		for _, expr := range caseClause.List {
			basicLit, ok := expr.(*ast.BasicLit)
			if !ok {
				continue
			}

			checkFunc(pass, basicLit)
		}
	}
}

func lookupFlag(pass *analysis.Pass, name string) bool {
	return pass.Analyzer.Flags.Lookup(name).Value.(flag.Getter).Get().(bool)
}

func checkHTTPMethod(pass *analysis.Pass, basicLit *ast.BasicLit) {
	currentVal := getBasicLitValue(basicLit)

	key := strings.ToUpper(currentVal)

	if newVal, ok := mapping.HTTPMethod[key]; ok {
		report(pass, basicLit, currentVal, newVal)
	}
}

func checkHTTPStatusCode(pass *analysis.Pass, basicLit *ast.BasicLit) {
	currentVal := getBasicLitValue(basicLit)

	if newVal, ok := mapping.HTTPStatusCode[currentVal]; ok {
		report(pass, basicLit, currentVal, newVal)
	}
}

func checkTimeWeekday(pass *analysis.Pass, basicLit *ast.BasicLit) {
	currentVal := getBasicLitValue(basicLit)

	if newVal, ok := mapping.TimeWeekday[currentVal]; ok {
		report(pass, basicLit, currentVal, newVal)
	}
}

func checkTimeMonth(pass *analysis.Pass, basicLit *ast.BasicLit) {
	currentVal := getBasicLitValue(basicLit)

	if newVal, ok := mapping.TimeMonth[currentVal]; ok {
		report(pass, basicLit, currentVal, newVal)
	}
}

func checkTimeLayout(pass *analysis.Pass, basicLit *ast.BasicLit) {
	currentVal := getBasicLitValue(basicLit)

	if newVal, ok := mapping.TimeLayout[currentVal]; ok {
		report(pass, basicLit, currentVal, newVal)
	}
}

func checkCryptoHash(pass *analysis.Pass, basicLit *ast.BasicLit) {
	currentVal := getBasicLitValue(basicLit)

	if newVal, ok := mapping.CryptoHash[currentVal]; ok {
		report(pass, basicLit, currentVal, newVal)
	}
}

func checkRPCDefaultPath(pass *analysis.Pass, basicLit *ast.BasicLit) {
	currentVal := getBasicLitValue(basicLit)

	if newVal, ok := mapping.RPCDefaultPath[currentVal]; ok {
		report(pass, basicLit, currentVal, newVal)
	}
}

func checkOSDevNull(pass *analysis.Pass, basicLit *ast.BasicLit) {}

func checkSQLIsolationLevel(pass *analysis.Pass, basicLit *ast.BasicLit) {
	currentVal := getBasicLitValue(basicLit)

	if newVal, ok := mapping.SQLIsolationLevel[currentVal]; ok {
		report(pass, basicLit, currentVal, newVal)
	}
}

func checkTLSSignatureScheme(pass *analysis.Pass, basicLit *ast.BasicLit) {
	currentVal := getBasicLitValue(basicLit)

	if newVal, ok := mapping.TLSSignatureScheme[currentVal]; ok {
		report(pass, basicLit, currentVal, newVal)
	}
}

func checkConstantKind(pass *analysis.Pass, basicLit *ast.BasicLit) {
	currentVal := getBasicLitValue(basicLit)

	if newVal, ok := mapping.ConstantKind[currentVal]; ok {
		report(pass, basicLit, currentVal, newVal)
	}
}

func checkSyslogPriority(pass *analysis.Pass, basicLit *ast.BasicLit) {}

// getBasicLitFromArgs gets the *ast.BasicLit of a function argument.
//
// Arguments:
//   - args - slice of function arguments
//   - count - expected number of argument in function
//   - idx - index of the argument to get the *ast.BasicLit
//   - typ - argument type
func getBasicLitFromArgs(args []ast.Expr, count, idx int, typ token.Token) *ast.BasicLit {
	if len(args) != count {
		return nil
	}

	if idx > count-1 {
		return nil
	}

	basicLit, ok := args[idx].(*ast.BasicLit)
	if !ok {
		return nil
	}

	if basicLit.Kind != typ {
		return nil
	}

	return basicLit
}

// getBasicLitFromElts gets the *ast.BasicLit of a struct elements.
//
// Arguments:
//   - elts - slice of a struct elements
//   - key - name of key in struct
func getBasicLitFromElts(elts []ast.Expr, key string) *ast.BasicLit {
	for _, e := range elts {
		keyValueExpr, ok := e.(*ast.KeyValueExpr)
		if !ok {
			continue
		}

		ident, ok := keyValueExpr.Key.(*ast.Ident)
		if !ok {
			continue
		}

		if ident.Name != key {
			continue
		}

		if basicLit, ok := keyValueExpr.Value.(*ast.BasicLit); ok {
			return basicLit
		}
	}

	return nil
}

// getBasicLitValue returns BasicLit value as string without quotes.
func getBasicLitValue(basicLit *ast.BasicLit) string {
	var val strings.Builder
	for _, r := range basicLit.Value {
		if r == '"' {
			continue
		} else {
			val.WriteRune(r)
		}
	}
	return val.String()
}

func report(pass *analysis.Pass, rg analysis.Range, currentVal, newVal string) {
	pass.Report(analysis.Diagnostic{
		Pos:     rg.Pos(),
		Message: fmt.Sprintf("%q can be replaced by %s", currentVal, newVal),
		SuggestedFixes: []analysis.SuggestedFix{{
			TextEdits: []analysis.TextEdit{{
				Pos:     rg.Pos(),
				End:     rg.End(),
				NewText: []byte(newVal),
			}},
		}},
	})
}
