/
megashares.go
120 lines (108 loc) · 2.46 KB
/
megashares.go
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
package megashares
import (
"fmt"
"io/ioutil"
"log"
"net/http"
"net/http/cookiejar"
"net/url"
"strings"
gq "github.com/PuerkitoBio/goquery"
)
const (
loginURL = "https://d01.megashares.com/myms_login.php"
searchURL = "http://www.megashares.com/search.php"
)
type Megashares struct {
CookieJar *cookiejar.Jar
Client *http.Client
}
func New() *Megashares {
cj, _ := cookiejar.New(nil)
return &Megashares{cj, &http.Client{Jar: cj}}
// m := new()
// m.cookieJar, _ := cookiejar.New(nil)
// m.Client := &http.Client{Jar: cookieJar}
}
type Entry struct {
Url string
Filename string
}
func (m *Entry) String() string {
return m.Filename
}
func ParseEntryFromURL(url string) (*Entry, error) {
// TODO: Double check url format.
i := strings.LastIndex(url, `fln=/`)
if i < 0 {
return nil, fmt.Errorf("Download url doesn't conform to 'fln=/'.")
}
return &Entry{url, url[i+5:]}, nil
}
func (m *Megashares) Login(username, password string) error {
values := url.Values{
"mymslogin_name": {username},
"mymspassword": {password},
"myms_login": {"Login"},
}
if r, err := m.Client.PostForm(loginURL, values); err != nil {
return err
} else {
if data, err := ioutil.ReadAll(r.Body); err != nil {
return err
} else {
if strings.Contains(string(data), "premium") {
return nil
}
}
return fmt.Errorf("Invalid login.")
}
}
func (m *Megashares) Search(query string) ([]byte, error) {
values := url.Values{
"q": {query},
"simple": {"Submit"},
}
r, err := m.Client.PostForm(searchURL, values)
if err != nil {
log.Fatal(err)
return nil, err
}
defer r.Body.Close()
return ioutil.ReadAll(r.Body)
}
func (m *Megashares) SearchResponse(query string) (*http.Response, error) {
values := url.Values{
"q": {query},
"simple": {"Submit"},
}
r, err := m.Client.PostForm(searchURL, values)
if err != nil {
log.Fatal(err)
return nil, err
}
return r, nil
}
func (m *Megashares) SearchEntries(query string) ([]*Entry, error) {
values := url.Values{
"q": {query},
"simple": {"Submit"},
}
r, err := m.Client.PostForm(searchURL, values)
if err != nil {
log.Fatal(err)
return nil, err
}
d, err := gq.NewDocumentFromResponse(r)
if err != nil {
log.Fatal(err)
}
urls := d.Find("div.float-r a img").Parent()
entries := make([]*Entry, urls.Length())
urls.Each(func(i int, s *gq.Selection) {
v, _ := s.Attr(`href`)
// TODO: Handling errors?
entries[i], _ = ParseEntryFromURL(v)
})
return entries, nil
}