Skip to content

Commit

Permalink
reset client
Browse files Browse the repository at this point in the history
  • Loading branch information
@LY1806620741
LY1806620741 committed Jan 2, 2024
1 parent d5d907e commit e6f628e
Show file tree
Hide file tree
Showing 8 changed files with 99 additions and 865 deletions.
138 changes: 40 additions & 98 deletions client/client.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,13 +16,13 @@ import (
const rootName = "rootCA.pem"

type mkcert struct {
CAROOT string
caCert *x509.Certificate
CAROOT string
caCert *x509.Certificate

// The system cert pool is only loaded once. After installing the root, checks
// will keep failing until the next execution. TODO: maybe execve?
// https://github.com/golang/go/issues/24540 (thanks, myself)
ignoreCheckFailure bool
ignoreCheckFailure bool
}

func (m *mkcert) checkPlatform() bool {
Expand All @@ -37,127 +37,75 @@ func (m *mkcert) checkPlatform() bool {
func (m *mkcert) install() {
if storeEnabled("system") {
if m.checkPlatform() {
log.Print(i18nText.
scan95,
)
log.Print("The local CA is already installed in the system trust store! 👍")
} else {
if m.installPlatform() {
log.Print(i18nText.
scan95,
)
log.Print("The local CA is now installed in the system trust store! ⚡️")
}
m.ignoreCheckFailure = true // TODO: replace with a check for a successful install
m.ignoreCheckFailure = true // TODO: replace with a check for a successful install
}
}
if storeEnabled("nss") && hasNSS {
if m.checkNSS() {
log.Printf(i18nText.
scan95,

NSSBrowsers)
log.Printf("The local CA is already installed in the %s trust store! 👍", NSSBrowsers)
} else {
if hasCertutil && m.installNSS() {
log.Printf(i18nText.
scan95,

NSSBrowsers)
log.Printf("The local CA is now installed in the %s trust store (requires browser restart)! 🦊", NSSBrowsers)
} else if CertutilInstallHelp == "" {
log.Printf(i18nText.
scan65,

NSSBrowsers)
log.Printf(`Note: %s support is not available on your platform. ℹ️`, NSSBrowsers)
} else if !hasCertutil {
log.Printf(i18nText.
scan66,

NSSBrowsers)
log.Printf(i18nText.
scan67,

CertutilInstallHelp)
log.Printf(`Warning: "certutil" is not available, so the CA can't be automatically installed in %s! ⚠️`, NSSBrowsers)
log.Printf(`Install "certutil" with "%s" and re-run "mkcert -install" 👈`, CertutilInstallHelp)
}
}
}
if storeEnabled("java") && hasJava {
if m.checkJava() {
log.Println(i18nText.
scan95,
)
log.Println("The local CA is already installed in Java's trust store! 👍")
} else {
if hasKeytool {
m.installJava()
log.Println(i18nText.
scan95,
)
log.Println("The local CA is now installed in Java's trust store! ☕️")
} else {
log.Println(i18nText.
scan70,
)
log.Println(`Warning: "keytool" is not available, so the CA can't be automatically installed in Java's trust store! ⚠️`)
}
}
}
log.Print(i18nText.
scan95,
)
log.Print("")
}

func (m *mkcert) uninstall() {
if storeEnabled("nss") && hasNSS {
if hasCertutil {
m.uninstallNSS()
} else if CertutilInstallHelp != "" {
log.Print(i18nText.
scan95,
)
log.Printf(i18nText.
scan72,

NSSBrowsers)
log.Printf(i18nText.
scan73,

CertutilInstallHelp)
log.Print(i18nText.
scan95,
)
log.Print("")
log.Printf(`Warning: "certutil" is not available, so the CA can't be automatically uninstalled from %s (if it was ever installed)! ⚠️`, NSSBrowsers)
log.Printf(`You can install "certutil" with "%s" and re-run "mkcert -uninstall" 👈`, CertutilInstallHelp)
log.Print("")
}
}
if storeEnabled("java") && hasJava {
if hasKeytool {
m.uninstallJava()
} else {
log.Print(i18nText.
scan95,
)
log.Println(i18nText.
scan74,
)
log.Print(i18nText.
scan95,
)
log.Print("")
log.Println(`Warning: "keytool" is not available, so the CA can't be automatically uninstalled from Java's trust store (if it was ever installed)! ⚠️`)
log.Print("")
}
}
if storeEnabled("system") && m.uninstallPlatform() {
log.Print(i18nText.
scan95,
)
log.Print(i18nText.
scan95,
)
log.Print("The local CA is now uninstalled from the system trust store(s)! 👋")
log.Print("")
} else if storeEnabled("nss") && hasCertutil {
log.Printf(i18nText.
scan95,

NSSBrowsers)
log.Print(i18nText.
scan95,
)
log.Printf("The local CA is now uninstalled from the %s trust store(s)! 👋", NSSBrowsers)
log.Print("")
}
}

type item struct {
Name string
Description string
Name string
Description string
}

func (m *mkcert) caUniqueName() string {
Expand All @@ -179,10 +127,10 @@ func main() {
}

templates := &promptui.SelectTemplates{
Label: "{{ . }}?",
Active: "\U0001F336 {{ .Name | cyan }}",
Inactive: " {{ .Name | cyan }}",
Selected: "\U0001F336 {{ .Name | red | cyan }}",
Label: "{{ . }}?",
Active: "\U0001F336 {{ .Name | cyan }}",
Inactive: " {{ .Name | cyan }}",
Selected: "\U0001F336 {{ .Name | red | cyan }}",
Details: `
--------- 详情 ----------
{{ "名字:" | faint }} {{ .Name }}
Expand All @@ -198,33 +146,27 @@ func main() {
}

prompt := promptui.Select{
Label: "当前是授信客户端,你要做什么",
Items: items,
Templates: templates,
Size: 4,
Searcher: searcher,
Label: "当前是授信客户端,你要做什么",
Items: items,
Templates: templates,
Size: 4,
Searcher: searcher,
}

i, _, err := prompt.Run()

if err != nil {
fmt.Printf("Prompt failed %v\n", err)
log.Fatalln(i18nText.
scan95,
)
log.Fatalln("选择错误")
} else {
m := &mkcert{}
m.CAROOT = "./"
certDERBlock, _ := pem.Decode(cert)
if certDERBlock == nil || certDERBlock.Type != "CERTIFICATE" {
log.Fatalln(i18nText.
scan95,
)
log.Fatalln("ERROR: failed to read the CA certificate: unexpected content")
}
m.caCert, err = x509.ParseCertificate(certDERBlock.Bytes)
fatalIfErr(err, i18nText.
scan95,
)
fatalIfErr(err, "failed to parse the CA certificate")
if i == 0 {
m.install()
} else if i == 1 {
Expand Down
14 changes: 3 additions & 11 deletions client/func.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -17,9 +17,7 @@ func commandWithSudo(cmd ...string) *exec.Cmd {
}
if !binaryExists("sudo") {
sudoWarningOnce.Do(func() {
log.Println(i18nText.
scan79,
)
log.Println(`Warning: "sudo" is not available, and mkcert is not running as root. The (un)install operation might fail. ⚠️`)
})
return exec.Command(cmd[0], cmd[1:]...)
}
Expand All @@ -28,10 +26,7 @@ func commandWithSudo(cmd ...string) *exec.Cmd {

func fatalIfCmdErr(err error, cmd string, out []byte) {
if err != nil {
log.Fatalf(i18nText.
scan95,

cmd, err, out)
log.Fatalf("ERROR: failed to execute \"%s\": %s\n\n%s\n", cmd, err, out)
}
}

Expand Down Expand Up @@ -60,9 +55,6 @@ func storeEnabled(name string) bool {

func fatalIfErr(err error, msg string) {
if err != nil {
log.Fatalf(i18nText.
scan95,

msg, err)
log.Fatalf("ERROR: %s: %s", msg, err)
}
}
Loading

0 comments on commit e6f628e

Please sign in to comment.