2023-05-10 22:47:27 +09:00
|
|
|
|
package main
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"fmt"
|
|
|
|
|
"os"
|
|
|
|
|
"strings"
|
|
|
|
|
"strconv"
|
|
|
|
|
)
|
|
|
|
|
|
2023-07-17 01:02:37 +09:00
|
|
|
|
var version = "1.1.0"
|
2023-05-10 22:47:27 +09:00
|
|
|
|
|
|
|
|
|
func help () {
|
|
|
|
|
fmt.Println("使い方:");
|
|
|
|
|
fmt.Println("hozonsite -v :バージョンを表示");
|
|
|
|
|
fmt.Println("hozonsite -s [ポート番号] :ポート番号でウェブサーバーを実行(デフォルト=9920)");
|
|
|
|
|
fmt.Println("hozonsite -h :ヘルプを表示");
|
|
|
|
|
fmt.Println("hozonsite <URL> :コマンドラインでウェブサイトを保存");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func main () {
|
2023-07-17 02:00:40 +09:00
|
|
|
|
cnf, err := getconf() // コンフィグファイル
|
|
|
|
|
if err != nil {
|
|
|
|
|
fmt.Println(err)
|
|
|
|
|
return
|
|
|
|
|
}
|
2023-06-27 13:48:35 +09:00
|
|
|
|
args := os.Args // コマンドラインのパラメートル
|
2023-05-10 22:47:27 +09:00
|
|
|
|
if len(args) == 2 {
|
2023-06-27 13:48:35 +09:00
|
|
|
|
if args[1] == "-v" { // バージョンを表示
|
2023-05-10 22:47:27 +09:00
|
|
|
|
fmt.Println("hozonsite-" + version)
|
|
|
|
|
return
|
2023-06-27 13:48:35 +09:00
|
|
|
|
} else if args[1] == "-s" { // :9920でウェブサーバーを実行
|
2023-05-10 22:47:27 +09:00
|
|
|
|
serv(cnf, 9920)
|
2023-06-27 13:48:35 +09:00
|
|
|
|
} else if args[1] == "-h" { // ヘルプを表示
|
2023-05-10 22:47:27 +09:00
|
|
|
|
help()
|
|
|
|
|
return
|
2023-06-27 13:48:35 +09:00
|
|
|
|
} else { // コマンドラインでウェブサイトを保存
|
|
|
|
|
if checkprefix(args[1]) { // プロトコールはあってるかどうか
|
|
|
|
|
eurl := stripurl(args[1]) // パラメートルの文字(?、=等)を削除
|
|
|
|
|
exist := checkexist(eurl, cnf.datapath) // 既に/usr/share/hozonsite/archiveに存在するかどうか
|
|
|
|
|
var confirm string // 既に存在したら、使う
|
|
|
|
|
if len(exist) > 0 { // あ、既に存在する
|
2023-05-11 07:17:45 +09:00
|
|
|
|
fmt.Println("このページが既に保存されているみたいです。")
|
|
|
|
|
fmt.Println("本当に手続きましょうか? [y/N]")
|
2023-06-27 13:48:35 +09:00
|
|
|
|
for _, ex := range exist { // 既に存在するページのURLを表示
|
2023-05-11 07:17:45 +09:00
|
|
|
|
fmt.Println(strings.Replace(ex, cnf.datapath, cnf.domain, 1))
|
|
|
|
|
}
|
|
|
|
|
fmt.Scanf("%s", &confirm)
|
|
|
|
|
}
|
2023-06-27 13:48:35 +09:00
|
|
|
|
if len(exist) == 0 || confirm == "y" || confirm == "Y" { // 存在しない OR 「本当に手続きましょうか?」でYを入力した場合
|
2023-06-23 12:48:13 +09:00
|
|
|
|
path := mkdirs(eurl, cnf.datapath)
|
2023-06-27 13:48:35 +09:00
|
|
|
|
getpage(args[1], path) // ページをダウンロード
|
|
|
|
|
scanpage(path, eurl, cnf.datapath) // 色々の必須な編集
|
|
|
|
|
fmt.Println(cnf.domain + strings.Replace(path, cnf.datapath, "", 1)) // 新しいURLを表示
|
2023-05-11 07:17:45 +09:00
|
|
|
|
}
|
2023-05-10 22:47:27 +09:00
|
|
|
|
return
|
2023-06-27 13:48:35 +09:00
|
|
|
|
} else { // 結局HTTPかHTTPSじゃないわね…
|
2023-05-10 22:47:27 +09:00
|
|
|
|
fmt.Println("URLは不正です。終了…")
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
}
|
2023-06-27 13:48:35 +09:00
|
|
|
|
} else if len(args) == 3 && args[1] == "-s" { // 好みなポート番号でウェブサーバーを実行
|
|
|
|
|
if port, err := strconv.Atoi(args[2]); err != nil { // でも、数字じゃないかもしん
|
2023-05-10 22:47:27 +09:00
|
|
|
|
fmt.Printf("%qは数字ではありません。\n", args[2])
|
|
|
|
|
return
|
2023-06-27 13:48:35 +09:00
|
|
|
|
} else { // OK、実行しよ〜
|
2023-05-10 22:47:27 +09:00
|
|
|
|
serv(cnf, port)
|
|
|
|
|
}
|
2023-06-27 13:48:35 +09:00
|
|
|
|
} else { // パラメートルは不明の場合、ヘルプを表示
|
2023-05-10 22:47:27 +09:00
|
|
|
|
help()
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
}
|