Mercurial > mercurial > hgweb_rsearcher.cgi

comparison go/client.go @ 0:d3b8cd5aeb70

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
make repo.
author pyon@macmini
date Sun, 30 Sep 2018 17:27:04 +0900
parents
children eaa27e4ed5be
comparison
equal deleted inserted replaced
-1:000000000000 0:d3b8cd5aeb70
1 /*
2 client.go : client-program.
3 Version : 0.0
4 Last Change: 2018-09-30 Sun 17:09:04.
5
6 install to: rsearcher_root/
7
8 rsearcher_root/client.exe
9 + grsearcher.exe
10 + app.conf
11 + verup.bat
12 + auth.db ( temporary )
13 + hhs.db ( temporary )
14 + index.db ( 10 days )
15 + image/*.png
16 + doc/
17 + .cache/*.jpg ( temporary )
18
19 $ client.exe -r 192.168.0.1:3910 # get new release
20 $ client.exe -u 192.168.0.1:3910 # get new dbs
21 $ client.exe 192.168.0.1:3910 0800012345 20171231 # get images
22 */
23 package main
24
25 import (
26 "archive/tar"
27 "bytes"
28 "compress/gzip"
29 "flag"
30 "fmt"
31 "io"
32 "io/ioutil"
33 "log"
34 "net/http"
35 "os"
36 "path/filepath"
37 "regexp"
38 "time"
39 )
40
41 func main() {
42 var upgrade = flag.Bool( "r", false, "" ) // get grsearcher.exe, client.exe
43 var update = flag.Bool( "u", false, "" ) // get auth.db, hhs.db, index.db(5day)
44 flag.Parse()
45
46 if *upgrade {
47 if flag.NArg() != 1 {
48 fmt.Fprintf( os.Stderr, "no remote host.\n" )
49 os.Exit( 1 )
50 }
51 fmt.Println( "Getting grsearcher.exe..." )
52 host := flag.Args()[0]
53 if _, err := get_file( host, "/release/grsearcher.exe", "grsearcher.exe" ); err != nil {
54 log.Fatal( err )
55 }
56 fmt.Println( "Getting client.exe..." )
57 if _, err := get_file( host, "/release/client.exe", "client1.exe" ); err != nil {
58 log.Fatal( err )
59 }
60 os.Exit( 0 )
61 }
62
63 if *update {
64 if flag.NArg() != 1 {
65 fmt.Fprintf( os.Stderr, "no remote host.\n" )
66 os.Exit( 1 )
67 }
68 fmt.Println( "Getting auth.db..." )
69 host := flag.Args()[0]
70 if _, err := get_file( host, "/db/auth.db", "auth.db" ); err != nil {
71 log.Fatal( err )
72 }
73 fmt.Println( "Getting hhs.db..." )
74 if _, err := get_file( host, "/db/hhs.db", "hhs.db" ); err != nil {
75 log.Fatal( err )
76 }
77
78 if time.Now().Day() % 10 == 0 {
79 fmt.Println( "Getting index.db..." )
80 if _, err := get_file( host, "/db/index.db", "index.db" ); err != nil {
81 log.Fatal( err )
82 }
83 }
84 os.Exit( 0 )
85 }
86
87 // main
88 if flag.NArg() != 3 {
89 fmt.Fprintf( os.Stderr, "bad argument.\n" )
90 os.Exit( 1 )
91 }
92
93 host := flag.Args()[0]
94 h := flag.Args()[1]
95 d := flag.Args()[2]
96
97 r, err := regexp.Compile( "^0[1238][0-9]{8}$" )
98 if err != nil {
99 log.Fatal( err )
100 }
101 m := r.FindString( h )
102 if len( m ) == 0 {
103 fmt.Fprintf( os.Stderr, "bad hno.\n" )
104 os.Exit( 1 )
105 }
106
107 rfile := "/images/" + d + "/" + h + ".tgz"
108 b, err := get_file( host, rfile, "" )
109 if err != nil {
110 log.Fatal( err )
111 }
112 if err := untgz_images( d, b ); err != nil {
113 log.Fatal( err )
114 }
115 }
116
117 /* Subroutines */
118 // HTTP GET
119 func get_file( host, rfile, lfile string ) ( []byte, error ) {
120 res, err := http.Get( "http://" + host + "/" + rfile )
121 if err != nil {
122 return nil, err
123 }
124 b, err := ioutil.ReadAll( res.Body )
125 res.Body.Close()
126 if err != nil {
127 return nil, err
128 }
129 if lfile != "" {
130 if err := ioutil.WriteFile( lfile, b, 0644 ); err != nil {
131 return nil, err
132 }
133 return nil, nil
134 }
135 return b, nil
136 }
137
138 // Un-Tar/Gzip
139 func untgz_images( d string, b []byte ) error {
140 br := bytes.NewReader( b )
141 zr, err := gzip.NewReader( br )
142 if err != nil {
143 return err
144 }
145
146 tr := tar.NewReader( zr )
147 for i := 1; ; i++ {
148 //hdr, err := tr.Next()
149 _, err := tr.Next()
150 if err == io.EOF {
151 break // End of archive
152 }
153 if err != nil {
154 return err
155 }
156 //f, _ := os.Create( hdr.Name )
157 fn := fmt.Sprintf( "%s_%d.jpg", d, i )
158 fn = fmt.Sprintf( "00000000_%d.jpg", i ) // for test
159
160 fn = filepath.Join( ".cache", fn )
161 f, _ := os.Create( fn )
162 if _, err := io.Copy( f, tr ); err != nil {
163 return err
164 }
165 f.Close()
166 }
167
168 if err := zr.Close(); err != nil {
169 return err
170 }
171 return nil
172 }
173