parent
286ba2b26e
commit
ba9651a369
@ -1,98 +0,0 @@ |
||||
package main |
||||
|
||||
import ( |
||||
"bufio" |
||||
"errors" |
||||
"fmt" |
||||
"os" |
||||
"strconv" |
||||
"strings" |
||||
) |
||||
|
||||
type StringPrompt struct { |
||||
question string |
||||
defaultValue string |
||||
required bool |
||||
confirm bool |
||||
validators []func(string) error |
||||
} |
||||
|
||||
func String(question string) *StringPrompt { |
||||
return &StringPrompt{question: question} |
||||
} |
||||
|
||||
func Confirm(question string) *StringPrompt { |
||||
return &StringPrompt{ |
||||
question: question, |
||||
confirm: true, |
||||
} |
||||
} |
||||
|
||||
func (s *StringPrompt) Required(required bool) *StringPrompt { |
||||
s.required = required |
||||
return s |
||||
} |
||||
|
||||
func (s *StringPrompt) SetValidator(validators ...func(string) error) *StringPrompt { |
||||
s.validators = validators |
||||
return s |
||||
} |
||||
|
||||
func (s *StringPrompt) Run() string { |
||||
putsln := func(msg string) { |
||||
lines := strings.Count(s.question+"\n"+msg, "\n") + 1 |
||||
// https://blog.csdn.net/weixin_43988842/article/details/106169040
|
||||
fmt.Fprintf(os.Stderr, "\x1b[%dA", lines) // 上移 lines 行
|
||||
fmt.Fprint(os.Stderr, "\x1b[J") // 从光标位置到屏幕末尾擦除
|
||||
fmt.Fprintln(os.Stderr, msg) |
||||
} |
||||
|
||||
var input string |
||||
r := bufio.NewReader(os.Stdin) |
||||
|
||||
WHILE: |
||||
fmt.Fprint(os.Stderr, s.question+" ") |
||||
input, _ = r.ReadString('\n') |
||||
input = strings.TrimSpace(input) |
||||
if input == "" { |
||||
if !s.required { |
||||
putsln(s.question) |
||||
return "" |
||||
} |
||||
putsln("请输入内容！") |
||||
goto WHILE |
||||
} |
||||
for _, v := range s.validators { |
||||
err := v(input) |
||||
if err != nil { |
||||
putsln(err.Error()) |
||||
goto WHILE |
||||
} |
||||
} |
||||
if s.confirm && len(s.validators) == 0 { |
||||
switch strings.ToLower(input) { |
||||
case "y", "yes": |
||||
input = "y" |
||||
case "n", "no": |
||||
input = "n" |
||||
default: |
||||
input = "" |
||||
} |
||||
} |
||||
putsln(s.question + " " + input) |
||||
return input |
||||
} |
||||
|
||||
func main() { |
||||
name := String("What is your name?"). |
||||
Required(true). |
||||
SetValidator(func(s string) error { |
||||
_, err := strconv.ParseInt(s, 10, 64) |
||||
if err != nil { |
||||
return errors.New("请输入一个数字") |
||||
} |
||||
return nil |
||||
}). |
||||
Run() |
||||
fmt.Printf("Hello, %s!\n", name) |
||||
} |
||||
@ -0,0 +1,85 @@ |
||||
package main |
||||
|
||||
import ( |
||||
"crypto/rand" |
||||
"crypto/rsa" |
||||
"crypto/x509" |
||||
"encoding/asn1" |
||||
"encoding/gob" |
||||
"encoding/pem" |
||||
"os" |
||||
) |
||||
|
||||
// TicketKeysGenerator 创建 ticket 密钥
|
||||
type TicketKeysGenerator struct{} |
||||
|
||||
func (g *TicketKeysGenerator) Generate() (err error) { |
||||
defer func() { |
||||
if recovered := recover(); recovered != nil { |
||||
err = recovered.(error) |
||||
} |
||||
}() |
||||
|
||||
reader := rand.Reader |
||||
bitSize := 2048 |
||||
|
||||
key, err := rsa.GenerateKey(reader, bitSize) |
||||
g.checkError(err) |
||||
|
||||
publicKey := key.PublicKey |
||||
|
||||
g.saveGobKey("private.key", key) |
||||
g.savePEMKey("private.pem", key) |
||||
|
||||
g.saveGobKey("public.key", publicKey) |
||||
g.savePublicPEMKey("public.pem", publicKey) |
||||
|
||||
return |
||||
} |
||||
|
||||
func (g *TicketKeysGenerator) saveGobKey(fileName string, key interface{}) { |
||||
outFile, err := os.Create(fileName) |
||||
g.checkError(err) |
||||
defer outFile.Close() |
||||
|
||||
encoder := gob.NewEncoder(outFile) |
||||
err = encoder.Encode(key) |
||||
g.checkError(err) |
||||
} |
||||
|
||||
func (g *TicketKeysGenerator) savePEMKey(fileName string, key *rsa.PrivateKey) { |
||||
outFile, err := os.Create(fileName) |
||||
g.checkError(err) |
||||
defer outFile.Close() |
||||
|
||||
var privateKey = &pem.Block{ |
||||
Type: "PRIVATE KEY", |
||||
Bytes: x509.MarshalPKCS1PrivateKey(key), |
||||
} |
||||
|
||||
err = pem.Encode(outFile, privateKey) |
||||
g.checkError(err) |
||||
} |
||||
|
||||
func (g *TicketKeysGenerator) savePublicPEMKey(fileName string, pubkey rsa.PublicKey) { |
||||
asn1Bytes, err := asn1.Marshal(pubkey) |
||||
g.checkError(err) |
||||
|
||||
var pemkey = &pem.Block{ |
||||
Type: "PUBLIC KEY", |
||||
Bytes: asn1Bytes, |
||||
} |
||||
|
||||
pemfile, err := os.Create(fileName) |
||||
g.checkError(err) |
||||
defer pemfile.Close() |
||||
|
||||
err = pem.Encode(pemfile, pemkey) |
||||
g.checkError(err) |
||||
} |
||||
|
||||
func (g *TicketKeysGenerator) checkError(err error) { |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
} |
||||
@ -0,0 +1,14 @@ |
||||
package util |
||||
|
||||
import "os" |
||||
|
||||
func PathExists(path string) bool { |
||||
_, err := os.Stat(path) |
||||
if err == nil { |
||||
return true |
||||
} |
||||
if os.IsNotExist(err) { |
||||
return false |
||||
} |
||||
panic(err) |
||||
} |
||||
@ -1,76 +0,0 @@ |
||||
package main |
||||
|
||||
import ( |
||||
"crypto/rand" |
||||
"crypto/rsa" |
||||
"crypto/x509" |
||||
"encoding/asn1" |
||||
"encoding/gob" |
||||
"encoding/pem" |
||||
"fmt" |
||||
"os" |
||||
) |
||||
|
||||
func main() { |
||||
reader := rand.Reader |
||||
bitSize := 2048 |
||||
|
||||
key, err := rsa.GenerateKey(reader, bitSize) |
||||
checkError(err) |
||||
|
||||
publicKey := key.PublicKey |
||||
|
||||
saveGobKey("private.key", key) |
||||
savePEMKey("private.pem", key) |
||||
|
||||
saveGobKey("public.key", publicKey) |
||||
savePublicPEMKey("public.pem", publicKey) |
||||
} |
||||
|
||||
func saveGobKey(fileName string, key interface{}) { |
||||
outFile, err := os.Create(fileName) |
||||
checkError(err) |
||||
defer outFile.Close() |
||||
|
||||
encoder := gob.NewEncoder(outFile) |
||||
err = encoder.Encode(key) |
||||
checkError(err) |
||||
} |
||||
|
||||
func savePEMKey(fileName string, key *rsa.PrivateKey) { |
||||
outFile, err := os.Create(fileName) |
||||
checkError(err) |
||||
defer outFile.Close() |
||||
|
||||
var privateKey = &pem.Block{ |
||||
Type: "PRIVATE KEY", |
||||
Bytes: x509.MarshalPKCS1PrivateKey(key), |
||||
} |
||||
|
||||
err = pem.Encode(outFile, privateKey) |
||||
checkError(err) |
||||
} |
||||
|
||||
func savePublicPEMKey(fileName string, pubkey rsa.PublicKey) { |
||||
asn1Bytes, err := asn1.Marshal(pubkey) |
||||
checkError(err) |
||||
|
||||
var pemkey = &pem.Block{ |
||||
Type: "PUBLIC KEY", |
||||
Bytes: asn1Bytes, |
||||
} |
||||
|
||||
pemfile, err := os.Create(fileName) |
||||
checkError(err) |
||||
defer pemfile.Close() |
||||
|
||||
err = pem.Encode(pemfile, pemkey) |
||||
checkError(err) |
||||
} |
||||
|
||||
func checkError(err error) { |
||||
if err != nil { |
||||
fmt.Println("Fatal error ", err.Error()) |
||||
os.Exit(1) |
||||
} |
||||
} |
||||
Loading…
Reference in new issue