1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
/*
* This program is free software: you can redistribute it and/or
* modify it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public
* License along with this program. If not, see
* <http://www.gnu.org/licenses/>.
*
* (c) Vincenzo "KatolaZ" Nicosia 2017 -- <katolaz@freaknet.org>
*
*
* This file is part of "binnit", a minimal no-fuss pastebin-like
* server written in golang
*
*/
/*
*
* Store/Retrieve functions for FS-based paste storage
*
*/
package paste
import(
"crypto/sha256"
"fmt"
"log"
"os"
"io/ioutil"
"errors"
"bufio"
"strings"
)
func Store(title, date, content, dest_dir string) (string, error) {
h := sha256.New()
h.Write([]byte(title))
h.Write([]byte(date))
h.Write([]byte(content))
paste := fmt.Sprintf("# Title: %s\n# Date: %s\n%s", title, date, content)
paste_hash := fmt.Sprintf("%x", h.Sum(nil))
log.Printf(" `-- hash: %s\n", paste_hash)
paste_dir := dest_dir + "/"
// Now we save the file
for i := 0; i < len(paste_hash)-16; i++ {
paste_name := paste_hash[i:i+16]
if _, err := os.Stat(paste_dir + paste_name); os.IsNotExist(err) {
// The file does not exist, so we can create it
if err := ioutil.WriteFile(paste_dir + paste_name, []byte(paste), 0644); err == nil {
// and then we return the URL:
log.Printf(" `-- saving new paste to : %s", paste_dir + paste_name)
return paste_name, nil
} else {
log.Printf("Cannot create the paste: %s!\n", paste_dir + paste_name)
}
}
}
return "", errors.New("Cannot store the paste...Sorry!")
}
func Retrieve(URI string) (title, date, content string, err error) {
f_cont, err := os.Open(URI)
defer f_cont.Close()
if err == nil {
stuff := bufio.NewScanner(f_cont)
// The first line contains the title
stuff.Scan()
title = strings.Trim(strings.Split(stuff.Text(), ":")[1], " ")
stuff.Scan()
date = strings.Trim(strings.Join(strings.Split(stuff.Text(), ":")[1:], ":"), " ")
for stuff.Scan() {
content += stuff.Text() + "\n"
}
} else {
return "", "", "", errors.New("Cannot retrieve paste!!!")
}
return title, date, content, nil
}
|
