summaryrefslogtreecommitdiff
path: root/paste/paste.go
blob: a55a3b3b7c578efb1405ee2c42a8f57eead8a376 (plain)
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
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()
		}
	} else {

		return "", "", "", errors.New("Cannot retrieve paste!!!")
	}
	
	return title, date, content, nil
}