Home Explore Help
Sign In
RevolvingCow
/
enigma
Watch 1
Star 0
Fork 0
Code Issues 0 Pull Requests 0 Commits 2 Releases 0 Wiki
Browse Source

save point before baby drools on the keyboard

bmallred 10 years ago
parent
f4e1c52d3e
commit
4fe3b847c1
4 changed files with 154 additions and 1 deletions
Split View Show Diff Stats
  1. 3 1
      .gitignore
  2. 9 0
      book.html
  3. 9 0
      index.html
  4. 133 0
      main.go

+ 3 - 1
.gitignore
View File 

@ -19,4 +19,6 @@ _cgo_export.*
19 19
20 20 
_testmain.go
21 21
22 
*.exe
22 
*.exe
23 
*.safe
24 
enigma

+ 9 - 0
book.html
View File 

@ -0,0 +1,9 @@
1 
<!doctype html>
2 
<html>
3 
<head>
4 
    <link rel="stylesheet" href="https://maxcdn.bootstrapcdn.com/bootswatch/3.3.1/flatly/bootstrap.min.css" />
5 
</head>
6 
<body>
7 
    <h1>Enigma - book.html</h1>
8 
</body>
9 
</html>

+ 9 - 0
index.html
View File 

@ -0,0 +1,9 @@
1 
<!doctype html>
2 
<html>
3 
<head>
4 
    <link rel="stylesheet" href="https://maxcdn.bootstrapcdn.com/bootswatch/3.3.1/flatly/bootstrap.min.css" />
5 
</head>
6 
<body>
7 
    <h1>Enigma - index.html</h1>
8 
</body>
9 
</html>

+ 133 - 0
main.go
View File 

@ -0,0 +1,133 @@
1 
package main
2
3 
import (
4 
	"bytes"
5 
	"compress/gzip"
6 
	"encoding/json"
7 
	"fmt"
8 
	"io/ioutil"
9 
	"log"
10 
	"net/http"
11 
	"os"
12 
	"strings"
13 
)
14
15 
type Site struct {
16 
	Host                      string `json:host`
17 
	MinimumLength             int    `json:minimumLength`
18 
	MaximumLength             int    `json:maximumLength`
19 
	SpecialCharacters         string `json:specialCharacters`
20 
	NumberOfSpecialCharacters int    `json:numberOfSpecialCharacters`
21 
	NumberOfUpperCase         int    `json:numberOfUpperCase`
22 
	NumberOfDigits            int    `json:numberOfDigits`
23 
}
24
25 
func main() {
26 
	http.HandleFunc("/api", func(w http.ResponseWriter, r *http.Request) {
27 
	})
28
29 
	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
30 
		path := strings.TrimPrefix(r.URL.Path, "/")
31 
		if path == "" {
32 
			// Index
33 
			page, err := ioutil.ReadFile("index.html")
34 
			if err != nil {
35 
				http.NotFound(w, r)
36 
				return
37 
			}
38
39 
			fmt.Fprintf(w, string(page))
40 
		} else {
41 
			// A passphrase has been entered
42 
		}
43 
	})
44
45 
	log.Fatal(http.ListenAndServe("localhost:8080", nil))
46
47 
	//execDir, err := filepath.Abs(filepath.Dir(os.Args[0]))
48 
	//if err != nil {
49 
	//	log.Fatal(err)
50 
	//}
51
52 
	//file := filepath.Join(execDir, "enigma.safe")
53 
	//sites, err := Read(file)
54 
	//if err != nil {
55 
	//	log.Fatal(err)
56 
	//}
57 
	//log.Println(sites)
58
59 
	//err = Save(file, sites)
60 
	//if err != nil {
61 
	//	log.Fatal(err)
62 
	//}
63 
}
64
65 
func Save(file string, sites []Site) error {
66 
	// If the file doesn't exist then create it
67 
	if _, err := os.Stat(file); os.IsNotExist(err) {
68 
		_, err = os.Create(file)
69 
		if err != nil {
70 
			return err
71 
		}
72 
	}
73
74 
	// Marshal the JSON
75 
	b, err := json.Marshal(sites)
76 
	if err != nil {
77 
		return err
78 
	}
79
80 
	// Compress the contents
81 
	buffer := bytes.NewBuffer(b)
82 
	gzipWriter := gzip.NewWriter(buffer)
83 
	if err != nil {
84 
		return err
85 
	}
86 
	defer gzipWriter.Close()
87
88 
	// Write to the file
89 
	fi, err := os.OpenFile(file, os.O_WRONLY, 0666)
90 
	if err != nil {
91 
		return err
92 
	}
93 
	defer fi.Close()
94 
	_, err = fi.Write(buffer.Bytes())
95 
	if err != nil {
96 
		return err
97 
	}
98
99 
	return nil
100 
}
101
102 
func Read(file string) ([]Site, error) {
103 
	// If the file doesn't exist yet no worries
104 
	if _, err := os.Stat(file); os.IsNotExist(err) {
105 
		return []Site{}, nil
106 
	}
107
108 
	// Bring in the compressed data
109 
	log.Println("Reading compressed file")
110 
	compressed, err := ioutil.ReadFile(file)
111 
	if err != nil {
112 
		return nil, err
113 
	}
114
115 
	// Decompress the file contents
116 
	log.Println("Decompressing")
117 
	buffer := bytes.NewBuffer(compressed)
118 
	gzipReader, err := gzip.NewReader(buffer)
119 
	if err != nil {
120 
		return nil, err
121 
	}
122 
	defer gzipReader.Close()
123
124 
	// Unmarshal the JSON information
125 
	log.Println("Unmarshal")
126 
	var sites []Site
127 
	err = json.Unmarshal(buffer.Bytes(), &sites)
128 
	if err != nil {
129 
		return nil, err
130 
	}
131
132 
	return sites, nil
133 
}