56 lines
1.4 KiB
Go
56 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"go-site-clone/services"
|
|
|
|
"github.com/go-rod/rod/lib/proto"
|
|
"github.com/wailsapp/wails/v3/pkg/application"
|
|
)
|
|
|
|
type App struct {
|
|
app *application.App
|
|
}
|
|
|
|
var siteService services.SiteService
|
|
|
|
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() {
|
|
// 将页面及资源一起返回
|
|
// for _, v := range dataList {
|
|
// u, _ := url.Parse(v.URL)
|
|
// if v.Type == proto.NetworkResourceTypeImage && u.Hostname() == parsed.Hostname() {
|
|
// var File utils.File
|
|
// File.Download(v.URL)
|
|
// }
|
|
// }
|
|
// for _, v := range RouterAll {
|
|
// var File utils.File
|
|
// File.HTMLDownload(v)
|
|
// }
|
|
}
|