refactor: add interface

cloudingcity · Nov 10, 2020 · 1e2009b · 1e2009b
1 parent a207fec
commit 1e2009b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/generator.go b/generator.go
@@ -12,7 +12,7 @@ type Generator struct {
 	chars bytes.Buffer
 }
 
-func New(opts ...Option) *Generator {
+func New(opts ...Option) Gopass {
 	g := &Generator{}
 	g.With(opts...)
 	return g

diff --git a/gopass.go b/gopass.go
@@ -1,11 +1,17 @@
 package gopass
 
-var g *Generator
-
 func init() {
 	g = New()
 }
 
+type Gopass interface {
+	Generate(length int) []byte
+	GenerateString(length int) string
+	With(opts ...Option)
+}
+
+var g Gopass
+
 func Generate(length int) []byte {
 	return g.Generate(length)
 }