110 lines
2.6 KiB
Go
110 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"go-site-clone/services"
|
|
"go-site-clone/utils"
|
|
"net/url"
|
|
|
|
"github.com/go-rod/rod/lib/proto"
|
|
"github.com/wailsapp/wails/v3/pkg/application"
|
|
)
|
|
|
|
type App struct {
|
|
app *application.App
|
|
}
|
|
|
|
var siteService services.SiteService
|
|
|
|
// startup is called when the app starts. The context is saved
|
|
// so we can call the runtime methods
|
|
// func (a *App) ServiceStartup(ctx context.Context, options application.ServiceOptions) error {
|
|
|
|
// return nil
|
|
// }
|
|
|
|
// func (a *App) ServiceShutdown(ctx context.Context) error {
|
|
|
|
// return nil
|
|
// }
|
|
|
|
func (a *App) GetResources(rawURL string) *services.ResourcesList {
|
|
RequestParamsAll, RouterAll := siteService.GetAllResources(rawURL)
|
|
// 再次去重 根据url链接去除重复项
|
|
dataList := siteService.DeduplicationRequestByUrl(RequestParamsAll)
|
|
resources := &services.ResourcesList{
|
|
Dom: RouterAll,
|
|
}
|
|
// 将页面及资源一起返回
|
|
for _, v := range dataList {
|
|
if v.Type == proto.NetworkResourceTypeImage {
|
|
resources.Image = append(resources.Image, v.URL)
|
|
}
|
|
if v.Type == proto.NetworkResourceTypeStylesheet {
|
|
resources.CSS = append(resources.CSS, v.URL)
|
|
}
|
|
if v.Type == proto.NetworkResourceTypeScript {
|
|
resources.Script = append(resources.Script, v.URL)
|
|
}
|
|
if v.Type == proto.NetworkResourceTypeMedia {
|
|
resources.Video = append(resources.Video, v.URL)
|
|
}
|
|
}
|
|
return resources
|
|
}
|
|
|
|
func (a *App) DownloadSite(uri string, obj services.ResourcesList) bool {
|
|
// 将页面及资源一起返回
|
|
parsed, _ := url.Parse(uri)
|
|
var File utils.File
|
|
if len(obj.CSS) > 0 {
|
|
for k, v := range obj.CSS {
|
|
u, _ := url.Parse(v)
|
|
if parsed.Hostname() == u.Hostname() {
|
|
File.Download(v)
|
|
}
|
|
a.app.Event.Emit("download:css", k)
|
|
}
|
|
}
|
|
if len(obj.Script) > 0 {
|
|
for k, v := range obj.Script {
|
|
u, _ := url.Parse(v)
|
|
if parsed.Hostname() == u.Hostname() {
|
|
File.Download(v)
|
|
}
|
|
a.app.Event.Emit("download:script", k)
|
|
}
|
|
}
|
|
if len(obj.Image) > 0 {
|
|
for k, v := range obj.Image {
|
|
u, _ := url.Parse(v)
|
|
if parsed.Hostname() == u.Hostname() {
|
|
File.Download(v)
|
|
}
|
|
a.app.Event.Emit("download:image", k)
|
|
}
|
|
}
|
|
if len(obj.Video) > 0 {
|
|
for k, v := range obj.Video {
|
|
u, _ := url.Parse(v)
|
|
if parsed.Hostname() == u.Hostname() {
|
|
File.Download(v)
|
|
}
|
|
a.app.Event.Emit("download:video", k)
|
|
}
|
|
}
|
|
if len(obj.Dom) > 0 {
|
|
for k, v := range obj.Dom {
|
|
u, _ := url.Parse(v)
|
|
if parsed.Hostname() == u.Hostname() {
|
|
File.HTMLDownload(v)
|
|
}
|
|
a.app.Event.Emit("download:dom", k)
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func (a *App) GetDownloadList() []utils.FileDir {
|
|
return siteService.GetLocalSiteList()
|
|
}
|