Refactor code by using switch instead of if-else (#318)

This commit is contained in:
Oleksandr Redko 2023-07-18 09:44:48 +03:00 committed by GitHub
parent f53600aa9f
commit 873d96d0a0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 85 additions and 76 deletions

View File

@ -60,13 +60,14 @@ func main() {
// Figure out which thing to do and then do that
func start() error {
if *flagSign != "" {
switch {
case *flagSign != "":
return signToken()
} else if *flagVerify != "" {
case *flagVerify != "":
return verifyToken()
} else if *flagShow != "" {
case *flagShow != "":
return showToken()
} else {
default:
flag.Usage()
return fmt.Errorf("none of the required flags are present. What do you want me to do?")
}
@ -79,17 +80,18 @@ func loadData(p string) ([]byte, error) {
}
var rdr io.Reader
if p == "-" {
switch p {
case "-":
rdr = os.Stdin
} else if p == "+" {
case "+":
return []byte("{}"), nil
} else {
if f, err := os.Open(p); err == nil {
rdr = f
defer f.Close()
} else {
default:
f, err := os.Open(p)
if err != nil {
return nil, err
}
rdr = f
defer f.Close()
}
return io.ReadAll(rdr)
}
@ -136,14 +138,16 @@ func verifyToken() error {
if err != nil {
return nil, err
}
if isEs() {
switch {
case isEs():
return jwt.ParseECPublicKeyFromPEM(data)
} else if isRs() {
case isRs():
return jwt.ParseRSAPublicKeyFromPEM(data)
} else if isEd() {
case isEd():
return jwt.ParseEdPublicKeyFromPEM(data)
}
default:
return data, nil
}
})
// Print some debug data
@ -221,40 +225,41 @@ func signToken() error {
}
}
if isEs() {
if k, ok := key.([]byte); !ok {
switch {
case isEs():
k, ok := key.([]byte)
if !ok {
return fmt.Errorf("couldn't convert key data to key")
} else {
}
key, err = jwt.ParseECPrivateKeyFromPEM(k)
if err != nil {
return err
}
}
} else if isRs() {
if k, ok := key.([]byte); !ok {
case isRs():
k, ok := key.([]byte)
if !ok {
return fmt.Errorf("couldn't convert key data to key")
} else {
}
key, err = jwt.ParseRSAPrivateKeyFromPEM(k)
if err != nil {
return err
}
}
} else if isEd() {
if k, ok := key.([]byte); !ok {
case isEd():
k, ok := key.([]byte)
if !ok {
return fmt.Errorf("couldn't convert key data to key")
} else {
}
key, err = jwt.ParseEdPrivateKeyFromPEM(k)
if err != nil {
return err
}
}
}
if out, err := token.SignedString(key); err == nil {
fmt.Println(out)
} else {
out, err := token.SignedString(key)
if err != nil {
return fmt.Errorf("error signing token: %w", err)
}
fmt.Println(out)
return nil
}

View File

@ -163,17 +163,18 @@ func ExampleParse_errorChecking() {
return []byte("AllYourBase"), nil
})
if token.Valid {
switch {
case token.Valid:
fmt.Println("You look nice today")
} else if errors.Is(err, jwt.ErrTokenMalformed) {
case errors.Is(err, jwt.ErrTokenMalformed):
fmt.Println("That's not even a token")
} else if errors.Is(err, jwt.ErrTokenSignatureInvalid) {
case errors.Is(err, jwt.ErrTokenSignatureInvalid):
// Invalid signature
fmt.Println("Invalid signature")
} else if errors.Is(err, jwt.ErrTokenExpired) || errors.Is(err, jwt.ErrTokenNotValidYet) {
case errors.Is(err, jwt.ErrTokenExpired) || errors.Is(err, jwt.ErrTokenNotValidYet):
// Token is either expired or not active yet
fmt.Println("Timing is everything")
} else {
default:
fmt.Println("Couldn't handle this token:", err)
}

View File

@ -66,7 +66,9 @@ func TestHMACVerify(t *testing.T) {
func TestHMACSign(t *testing.T) {
for _, data := range hmacTestData {
if data.valid {
if !data.valid {
continue
}
parts := strings.Split(data.tokenString, ".")
method := jwt.GetSigningMethod(data.alg)
sig, err := method.Sign(strings.Join(parts[0:2], "."), hmacTestKey)
@ -78,7 +80,6 @@ func TestHMACSign(t *testing.T) {
}
}
}
}
func BenchmarkHS256Signing(b *testing.B) {
benchmarkSigning(b, jwt.SigningMethodHS256, hmacTestKey)

View File

@ -59,7 +59,9 @@ func TestNoneVerify(t *testing.T) {
func TestNoneSign(t *testing.T) {
for _, data := range noneTestData {
if data.valid {
if !data.valid {
continue
}
parts := strings.Split(data.tokenString, ".")
method := jwt.GetSigningMethod(data.alg)
sig, err := method.Sign(strings.Join(parts[0:2], "."), data.key)
@ -71,4 +73,3 @@ func TestNoneSign(t *testing.T) {
}
}
}
}

View File

@ -84,7 +84,9 @@ func TestRSAPSSSign(t *testing.T) {
}
for _, data := range rsaPSSTestData {
if data.valid {
if !data.valid {
continue
}
parts := strings.Split(data.tokenString, ".")
method := jwt.GetSigningMethod(data.alg)
sig, err := method.Sign(strings.Join(parts[0:2], "."), rsaPSSKey)
@ -98,7 +100,6 @@ func TestRSAPSSSign(t *testing.T) {
}
}
}
}
func TestRSAPSSSaltLengthCompatibility(t *testing.T) {
// Fails token verify, if salt length is auto.