converted to FilterGroups with filters per URL

This commit is contained in:
BroodjeAap 2022-08-01 19:40:11 +00:00
parent 71afae411c
commit 242b10e21b
7 changed files with 183 additions and 123 deletions

View file

@ -1,8 +1,7 @@
package main
type QueryUpdate struct {
ID uint `form:"query_id" binding:"required"`
Name string `form:"query_name" binding:"required" validate:"min=1"`
Type string `form:"query_type" binding:"required" validate:"oneof=css xpath regex json"`
Query string `form:"query" binding:"required"`
type FilterGroupUpdate struct {
ID uint `form:"group_id" binding:"required"`
Name string `form:"group_name" binding:"required" validate:"min=1"`
Type string `form:"group_type" binding:"required" validate:"oneof=diff enum number"`
}

55
main.go
View file

@ -63,7 +63,7 @@ func (web Web) viewWatch(c *gin.Context) {
id := c.Param("id")
var watch Watch
web.db.Model(&Watch{}).Preload("URLs.Queries.Filters").First(&watch, id)
web.db.Model(&Watch{}).Preload("URLs.GroupFilters.Filters").First(&watch, id)
c.HTML(http.StatusOK, "viewWatch", watch)
}
@ -79,20 +79,20 @@ func (web Web) createURL(c *gin.Context) {
c.Redirect(http.StatusSeeOther, fmt.Sprintf("/watch/view/%d", url.WatchID))
}
func (web Web) createQuery(c *gin.Context) {
func (web Web) createFilterGroup(c *gin.Context) {
watch_id, err := strconv.ParseUint(c.PostForm("w_id"), 10, 64)
if err != nil {
log.Print(err)
c.HTML(http.StatusInternalServerError, "500", gin.H{})
return
}
var query Query
errMap, err := bindAndValidateQuery(&query, c)
var group FilterGroup
errMap, err := bindAndValidateGroup(&group, c)
if err != nil {
c.HTML(http.StatusBadRequest, "500", errMap)
return
}
web.db.Create(&query)
web.db.Create(&group)
c.Redirect(http.StatusSeeOther, fmt.Sprintf("/watch/view/%d", watch_id))
}
@ -105,39 +105,38 @@ func (web Web) createFilter(c *gin.Context) {
return
}
web.db.Create(&filter)
c.Redirect(http.StatusSeeOther, fmt.Sprintf("/query/edit/%d", filter.QueryID))
c.Redirect(http.StatusSeeOther, fmt.Sprintf("/group/edit/%d", filter.FilterGroupID))
}
func (web Web) editQuery(c *gin.Context) {
query_id, err := strconv.ParseUint(c.Param("id"), 10, 64)
func (web Web) editGroup(c *gin.Context) {
group_id, err := strconv.ParseUint(c.Param("id"), 10, 64)
if err != nil {
c.Redirect(http.StatusSeeOther, "/watch/new")
return // TODO response
}
var query Query
web.db.Preload("URL.Watch").Preload("Filters").Preload("URL").First(&query, query_id)
var group FilterGroup
web.db.Preload("URL.Watch").Preload("Filters").Preload("URL").First(&group, group_id)
c.HTML(http.StatusOK, "editQuery", gin.H{
"Query": query,
"currentResult": getQueryResult(&query),
c.HTML(http.StatusOK, "editGroup", gin.H{
"Group": group,
"currentResult": getGroupResult(&group),
})
}
func (web Web) updateQuery(c *gin.Context) {
var queryUpdate QueryUpdate
errMap, err := bindAndValidateQueryUpdate(&queryUpdate, c)
func (web Web) updateGroup(c *gin.Context) {
var groupUpdate FilterGroupUpdate
errMap, err := bindAndValidateGroupUpdate(&groupUpdate, c)
if err != nil {
log.Print(err)
c.HTML(http.StatusBadRequest, "500", errMap)
return
}
var query Query
web.db.First(&query, queryUpdate.ID)
query.Name = queryUpdate.Name
query.Type = queryUpdate.Type
query.Query = queryUpdate.Query
web.db.Save(&query)
c.Redirect(http.StatusSeeOther, fmt.Sprintf("/query/edit/%d", +query.ID))
var group FilterGroup
web.db.First(&group, groupUpdate.ID)
group.Name = groupUpdate.Name
group.Type = groupUpdate.Type
web.db.Save(&group)
c.Redirect(http.StatusSeeOther, fmt.Sprintf("/group/edit/%d", +group.ID))
}
func passiveBot(bot *tgbotapi.BotAPI) {
@ -183,7 +182,7 @@ func main() {
}
db, _ := gorm.Open(sqlite.Open(viper.GetString("database.dsn")))
db.AutoMigrate(&Watch{}, &URL{}, &Query{}, &Filter{})
db.AutoMigrate(&Watch{}, &URL{}, &FilterGroup{}, &Filter{})
//bot, _ := tgbotapi.NewBotAPI(viper.GetString("telegram.token"))
@ -205,7 +204,7 @@ func main() {
templates.AddFromFiles("index", "templates/base.html", "templates/index.html")
templates.AddFromFiles("newWatch", "templates/base.html", "templates/newWatch.html")
templates.AddFromFiles("viewWatch", "templates/base.html", "templates/viewWatch.html")
templates.AddFromFiles("editQuery", "templates/base.html", "templates/editQuery.html")
templates.AddFromFiles("editGroup", "templates/base.html", "templates/editGroup.html")
templates.AddFromFiles("500", "templates/base.html", "templates/500.html")
router.HTMLRender = templates
@ -216,9 +215,9 @@ func main() {
router.POST("/watch/delete", web.deleteWatch)
router.GET("/watch/view/:id/", web.viewWatch)
router.POST("/url/create/", web.createURL)
router.POST("/query/create/", web.createQuery)
router.GET("/query/edit/:id", web.editQuery)
router.POST("/query/update", web.updateQuery)
router.POST("/group/create/", web.createFilterGroup)
router.GET("/group/edit/:id", web.editGroup)
router.POST("/group/update", web.updateGroup)
router.POST("/filter/create/", web.createFilter)
router.Run("0.0.0.0:8080")

View file

@ -13,29 +13,28 @@ type Watch struct {
type URL struct {
gorm.Model
WatchID uint `form:"url_watch_id" yaml:"url_watch_id" binding:"required"`
Watch *Watch `form:"watch" yaml:"watch" validate:"omitempty"`
Name string `form:"url_name" yaml:"url_name" binding:"required" validate:"min=1"`
URL string `form:"url" yaml:"url" binding:"required,url" validate:"min=1"`
Queries []Query
WatchID uint `form:"url_watch_id" yaml:"url_watch_id" binding:"required"`
Watch *Watch `form:"watch" yaml:"watch" validate:"omitempty"`
Name string `form:"url_name" yaml:"url_name" binding:"required" validate:"min=1"`
URL string `form:"url" yaml:"url" binding:"required,url" validate:"min=1"`
GroupFilters []FilterGroup
}
type Query struct {
type FilterGroup struct {
gorm.Model
URLID uint `form:"query_url_id" yaml:"query_url_id" binding:"required"`
URLID uint `form:"group_url_id" yaml:"group_url_id" binding:"required"`
URL *URL
Name string `form:"query_name" yaml:"query_name" binding:"required" validate:"min=1"`
Type string `form:"query_type" yaml:"query_type" binding:"required" validate:"oneof=css xpath regex json"`
Query string `form:"query" yaml:"query" binding:"required"`
Name string `form:"group_name" yaml:"group_name" binding:"required" validate:"min=1"`
Type string `form:"group_type" yaml:"group_type" binding:"required" validate:"oneof=diff enum number bool"`
Filters []Filter
}
type Filter struct {
gorm.Model
QueryID uint `form:"filter_query_id" yaml:"filter_query_id" binding:"required"`
Query *Query
Name string `form:"filter_name" yaml:"filter_name" binding:"required" validate:"min=1"`
Type string `form:"filter_type" yaml:"filter_type" binding:"required" validate:"oneof=replace regex substring"`
From string `form:"from" yaml:"from" binding:"required"`
To string `form:"to" yaml:"to" binding:"required"`
FilterGroupID uint `form:"filter_group_id" yaml:"filter_group_id" binding:"required"`
FilterGroup *FilterGroup
Name string `form:"filter_name" yaml:"filter_name" binding:"required" validate:"min=1"`
Type string `form:"filter_type" yaml:"filter_type" binding:"required" validate:"oneof=replace regex substring"`
From string `form:"from" yaml:"from" binding:"required"`
To string `form:"to" yaml:"to" binding:"required"`
}

View file

@ -2,7 +2,9 @@ package main
import (
"bytes"
"io/ioutil"
"log"
"net/http"
"regexp"
"strconv"
"strings"
@ -11,43 +13,68 @@ import (
"golang.org/x/net/html"
)
func getQueryResult(query *Query) []string {
doc, err := htmlquery.LoadURL(query.URL.URL)
func getGroupResult(group *FilterGroup) []string {
resp, err := http.Get(group.URL.URL)
if err != nil {
log.Print("Something went wrong loading loading", query.URL.URL)
log.Print("Something went wrong loading", group.URL.URL)
return []string{}
}
nodes, _ := htmlquery.QueryAll(doc, query.Query)
nodeStrings := make([]string, len(nodes))
for i, node := range nodes {
var b bytes.Buffer
html.Render(&b, node)
nodeStrings[i] = html.UnescapeString(b.String())
defer resp.Body.Close()
html, err := ioutil.ReadAll(resp.Body)
if err != nil {
log.Print("Something went wrong loading ", group.URL.URL)
return []string{}
}
for _, filter := range query.Filters {
for i, nodeString := range nodeStrings {
nodeStrings[i] = getFilterResult(nodeString, &filter)
resultStrings := []string{string(html)}
newStrings := []string{}
for _, filter := range group.Filters {
for _, resultString := range resultStrings {
getFilterResult(resultString, &filter, &newStrings)
}
resultStrings = newStrings
log.Println(resultStrings)
}
return nodeStrings
return resultStrings
}
func getFilterResult(s string, filter *Filter) string {
func getFilterResult(s string, filter *Filter, newStrings *[]string) {
switch {
case filter.Type == "css":
{
//getFilterResultReplace(s, filter, newStrings)
}
case filter.Type == "xpath":
{
getFilterResultXPath(s, filter, newStrings)
}
case filter.Type == "replace":
{
return getFilterResultReplace(s, filter)
//getFilterResultReplace(s, filter, newStrings)
}
case filter.Type == "regex":
{
return getFilterResultRegex(s, filter)
//getFilterResultRegex(s, filter, newStrings)
}
case filter.Type == "substring":
{
return getFilterResultSubstring(s, filter)
//getFilterResultSubstring(s, filter, newStrings)
}
default:
return s
}
}
func getFilterResultXPath(s string, filter *Filter, newStrings *[]string) {
doc, err := htmlquery.Parse(strings.NewReader(s))
if err != nil {
log.Print(err)
return
}
nodes, _ := htmlquery.QueryAll(doc, filter.From)
for _, node := range nodes {
var b bytes.Buffer
html.Render(&b, node)
*newStrings = append(*newStrings, html.UnescapeString(b.String()))
}
}

View file

@ -1,19 +1,16 @@
{{define "content"}}
<div class="card mb-5">
<div class="card-header text-center text-white bg-dark ">
<div class="h2">{{ .Query.URL.Watch.Name }} - {{ .Query.Name }}</div>
<div class="h6">{{ .Query.URL.URL }}</div>
<div class="h2">{{ .Group.URL.Watch.Name }} - {{ .Group.Name }}</div>
<div class="h6">{{ .Group.URL.URL }}</div>
</div>
<div class="card-header text-white bg-secondary">
<div class="row">
<div class="col-2">
{{ .Query.Type }}
</div>
<div class="col-8">
{{ .Query.Query }}
{{ .Group.Type }}
</div>
<div class="col-1">
<button type="button" class="btn btn-primary btn-sm" data-bs-toggle="modal" data-bs-target="#updateQueryModal">
<button type="button" class="btn btn-primary btn-sm" data-bs-toggle="modal" data-bs-target="#updateGroupModal">
Update
</button>
</div>
@ -22,17 +19,22 @@
<div class="card-body">
<div class="card mb-2">
<form action="/filter/create" method="post">
<input type="hidden" name="query_id" value="{{ .ID }}" >
<input type="hidden" name="group_id" value="{{ .ID }}" >
<input type="hidden" name="watch_id" value="{{ $.ID }}" >
<table class="table table-hover caption-top">
<tbody>
{{ range .Query.Filters }}
{{ range .Group.Filters }}
<tr>
<td>{{ .Name }}</td>
<td>{{ .Type }}</td>
<td>{{ .From }}</td>
<td>{{ .To }}</td>
<td></td>
<td>
<form action="/filter/delete" method="post">
<input type="hidden" name="filter_id" value="{{ .ID }}">
<input type="submit" class="btn btn-sm btn-danger" value="X">
</form>
</td>
</tr>
{{ end }}
<tr>
@ -68,31 +70,28 @@
{{ end }}
<div class="modal fade" id="updateQueryModal" tabindex="-1" aria-labelledby="updateQueryModal" aria-hidden="true">
<div class="modal fade" id="updateGroupModal" tabindex="-1" aria-labelledby="updateGroupModal" aria-hidden="true">
<div class="modal-dialog">
<div class="modal-content">
<div class="modal-header">
<h5 class="modal-title" id="updateQueryModal">Update Query</h5>
<h5 class="modal-title" id="updateGroupModal">Update Group</h5>
<button type="button" class="btn-close" data-bs-dismiss="modal" aria-label="Close"></button>
</div>
<div class="modal-body">
<form action="/query/update" method="post">
<input type="hidden" name="query_id" value="{{ .Query.ID }}" >
<form action="/group/update" method="post">
<input type="hidden" name="group_id" value="{{ .Group.ID }}" >
<div class="col">
<div class="mb-3 m-3 row">
<input type="text" name="query_name" value="{{ .Query.Name }}" class="form-control">
<input type="text" name="group_name" value="{{ .Group.Name }}" class="form-control">
</div>
<div class="mb-3 m-3 row">
<select name="query_type" class="form-control" >
<option value="css" {{ if eq .Query.Type "css" }} selected="true" {{ end }}>CSS</option>
<option value="xpath" {{ if eq .Query.Type "xpath" }} selected="true" {{ end }}>XPath</option>
<option value="regex" {{ if eq .Query.Type "regex" }} selected="true" {{ end }}>Regex</option>
<option value="json" {{ if eq .Query.Type "json" }} selected="true" {{ end }}>JSON</option>
<select name="group_type" class="form-control" >
<option value="diff" {{ if eq .Group.Type "diff" }} selected="true" {{ end }}>Diff</option>
<option value="enum" {{ if eq .Group.Type "enum" }} selected="true" {{ end }}>Enum</option>
<option value="number" {{ if eq .Group.Type "number" }} selected="true" {{ end }}>Number</option>
<option value="bool" {{ if eq .Group.Type "bool" }} selected="true" {{ end }}>Boolean</option>
</select>
</div>
<div class="mb-3 m-3 row">
<input type="text" name="query" class="form-control" value="{{ .Query.Query }}">
</div>
<button class="btn btn-primary">Update</button>
</div>
</form>
@ -113,21 +112,65 @@
<ul class="nav nav-tabs" id="newFilterTabs" role="tablist">
<li class="nav-item" role="presentation">
<button class="nav-link active" id="replace-tab" data-bs-toggle="tab" data-bs-target="#replace-tab-pane" type="button" role="tab" aria-controls="home-tab-pane" aria-selected="true">Replace</button>
<button class="nav-link active" id="xpath-tab" data-bs-toggle="tab" data-bs-target="#xpath-tab-pane" aria-controls="xpath-tab-pane" type="button" role="tab" aria-selected="true">XPath</button>
</li>
<li class="nav-item" role="presentation">
<button class="nav-link" id="regex-tab" data-bs-toggle="tab" data-bs-target="#regex-tab-pane" type="button" role="tab" aria-controls="profile-tab-pane" aria-selected="false">Regex</button>
<button class="nav-link" id="css-tab" data-bs-toggle="tab" data-bs-target="#css-tab-pane" aria-controls="css-tab-pane" type="button" role="tab" aria-selected="false">CSS</button>
</li>
<li class="nav-item" role="presentation">
<button class="nav-link" id="substring-tab" data-bs-toggle="tab" data-bs-target="#substring-tab-pane" type="button" role="tab" aria-controls="contact-tab-pane" aria-selected="false">Substring</button>
<button class="nav-link" id="replace-tab" data-bs-toggle="tab" data-bs-target="#replace-tab-pane" aria-controls="replace-tab-pane" type="button" role="tab" aria-selected="false">Replace</button>
</li>
<li class="nav-item" role="presentation">
<button class="nav-link" id="regex-tab" data-bs-toggle="tab" data-bs-target="#regex-tab-pane" aria-controls="regex-tab-pane" type="button" role="tab" aria-selected="false">Regex</button>
</li>
<li class="nav-item" role="presentation">
<button class="nav-link" id="substring-tab" data-bs-toggle="tab" data-bs-target="#substring-tab-pane" aria-controls="substring-tab-pane" type="button" role="tab" aria-selected="false">Substring</button>
</li>
</ul>
<div class="tab-content" id="newFilterTabsContent">
<div class="tab-pane fade show active" id="replace-tab-pane" role="tabpanel" aria-labelledby="replace-tab" tabindex="0">
<div class="tab-pane fade show active" id="xpath-tab-pane" role="tabpanel" aria-labelledby="xpath-tab" tabindex="0">
<form action="/filter/create" method="post">
<div class="mb-3 m-3 row">
<input type="hidden" name="filter_group_id" value="{{ .Group.ID }}" >
<input type="hidden" name="filter_type" value="xpath" >
<input type="hidden" name="to" value="-" >
<label for="nameInput" class="col-sm-2 col-form-label">Name:</label>
<div class="col-sm-10 p-2">
<input type="text" class="form-control" name="filter_name" id="nameInput" placeholder="Name">
</div>
<label for="fromInput" class="col-sm-2 col-form-label">XPath:</label>
<div class="col-sm-10 p-2">
<input type="text" class="form-control" name="from" id="fromInput" placeholder="//a[@class='price']">
</div>
<button class="btn btn-primary mt-4">Add Filter</button>
</div>
</form>
</div>
<div class="tab-pane fade" id="css-tab-pane" role="tabpanel" aria-labelledby="css-tab" tabindex="0">
<form action="/filter/create" method="post">
<div class="mb-3 m-3 row">
<input type="hidden" name="filter_group_id" value="{{ .Group.ID }}" >
<input type="hidden" name="filter_type" value="css" >
<input type="hidden" name="to" value="-" >
<label for="nameInput" class="col-sm-2 col-form-label">Name:</label>
<div class="col-sm-10 p-2">
<input type="text" class="form-control" name="filter_name" id="nameInput" placeholder="Name">
</div>
<label for="fromInput" class="col-sm-2 col-form-label">Selector:</label>
<div class="col-sm-10 p-2">
<input type="text" class="form-control" name="from" id="fromInput" placeholder=".price">
</div>
<button class="btn btn-primary mt-4">Add Filter</button>
</div>
</form>
</div>
<div class="tab-pane fade" id="replace-tab-pane" role="tabpanel" aria-labelledby="replace-tab" tabindex="0">
<form action="/filter/create" method="post">
<div class="mb-3 m-3 row">
<input type="hidden" name="filter_query_id" value="{{ .Query.ID }}" >
<input type="hidden" name="filter_group_id" value="{{ .Group.ID }}" >
<input type="hidden" name="filter_type" value="replace" >
<label for="nameInput" class="col-sm-2 col-form-label">Name:</label>
@ -149,7 +192,7 @@
<div class="tab-pane fade" id="regex-tab-pane" role="tabpanel" aria-labelledby="regex-tab" tabindex="0">
<form action="/filter/create" method="post">
<div class="mb-3 m-3 row">
<input type="hidden" name="filter_query_id" value="{{ .Query.ID }}" >
<input type="hidden" name="filter_group_id" value="{{ .Group.ID }}" >
<input type="hidden" name="filter_type" value="regex" >
<label for="nameInput" class="col-sm-2 col-form-label">Name:</label>
@ -171,7 +214,7 @@
<div class="tab-pane fade" id="substring-tab-pane" role="tabpanel" aria-labelledby="substring-tab" tabindex="0">
<form action="/filter/create" method="post">
<div class="mb-3 m-3 row">
<input type="hidden" name="filter_query_id" value="{{ .Query.ID }}" >
<input type="hidden" name="filter_group_id" value="{{ .Group.ID }}" >
<input type="hidden" name="filter_type" value="substring" >
<input type="hidden" name="to" value="-" >

View file

@ -18,20 +18,19 @@
<div>{{ .URL }}</div>
</div>
<div class="card-body">
{{ range .Queries }}
{{ range .GroupFilters }}
<div class="card mb-2">
<div class="card-header">
<div class="row">
<div class="col h5 text-start">{{ .Type }}</div>
<div class="col h4 text-center" >{{ .Name }}</div>
<div class="col h5 text-end"><a class="btn btn-success btn-sm" href="/query/edit/{{ .ID }}">Edit</a></div>
<div class="col h4 text-start" >{{ .Name }}</div>
<div class="col h5 text-end"><a class="btn btn-success btn-sm" href="/group/edit/{{ .ID }}">Edit</a></div>
</div>
<div class="text-center text-muted">
{{ .Query }}
{{ .Type }}
</div>
</div>
<div class="card-body">
<input type="hidden" name="query_id" value="{{ .ID }}" >
<input type="hidden" name="group_id" value="{{ .ID }}" >
<input type="hidden" name="w_id" value="{{ $.ID }}" >
<table class="table table-hover caption-top">
<tbody>
@ -53,34 +52,28 @@
</tbody>
</table>
</div>
<div class="card-footer">
</div>
</div>
{{ end }}
<form action="/query/create" method="post">
<input type="hidden" name="query_url_id" value="{{ .ID }}" >
<form action="/group/create" method="post">
<input type="hidden" name="group_url_id" value="{{ .ID }}" >
<input type="hidden" name="w_id" value="{{ $.ID }}" >
<table class="table table-hover caption-top">
<tbody>
<tr>
<td>
<input type="text" class="form-control" name="query_name" placeholder="Query Name">
<input type="text" class="form-control" name="group_name" placeholder="Group Name">
</td>
<td>
<select class="form-control" id="query_type" name="query_type">
<option value="css">CSS Selector</option>
<option value="xpath">XPath</option>
<option value="regex">Regex</option>
<option value="json">JSON</option>
<!-- additions/changes should also be added to Query.Type oneof validator -->
<select class="form-control" id="group_type" name="group_type">
<option value="diff">Difference</option>
<option value="enum">Enum</option>
<option value="number">Number</option>
<option value="bool">Boolean</option>
<!-- additions/changes should also be added to GroupFilter.Type oneof validator -->
</select>
</td>
<td>
<input type="text" class="form-control" name="query" placeholder="Query">
</td>
<td>
<button class="btn btn-primary">Add Query</button>
<button class="btn btn-primary">Add Group</button>
</td>
</tr>
</tbody>

View file

@ -17,8 +17,8 @@ func bindAndValidateURL(url *URL, c *gin.Context) (map[string]string, error) {
return validate(err), err
}
func bindAndValidateQuery(query *Query, c *gin.Context) (map[string]string, error) {
err := c.ShouldBind(query)
func bindAndValidateGroup(group *FilterGroup, c *gin.Context) (map[string]string, error) {
err := c.ShouldBind(group)
return validate(err), err
}
@ -27,8 +27,8 @@ func bindAndValidateFilter(filter *Filter, c *gin.Context) (map[string]string, e
return validate(err), err
}
func bindAndValidateQueryUpdate(query *QueryUpdate, c *gin.Context) (map[string]string, error) {
err := c.ShouldBind(query)
func bindAndValidateGroupUpdate(group *FilterGroupUpdate, c *gin.Context) (map[string]string, error) {
err := c.ShouldBind(group)
return validate(err), err
}