mirror of
https://github.com/documize/community.git
synced 2025-07-25 08:09:43 +02:00
restructure directories
This commit is contained in:
Elliott Stoneham
parent
7e4ed6545b
commit
a2ce777762
159 changed files with 320 additions and 323 deletions
84
core/section/papertrail/model.go
Normal file
84
core/section/papertrail/model.go
Normal file
|
|
@ -0,0 +1,84 @@
|
|||
// Copyright 2016 Documize Inc. <legal@documize.com>. All rights reserved.
|
||||
//
|
||||
// This software (Documize Community Edition) is licensed under
|
||||
// GNU AGPL v3 http://www.gnu.org/licenses/agpl-3.0.en.html
|
||||
//
|
||||
// You can operate outside the AGPL restrictions by purchasing
|
||||
// Documize Enterprise Edition and obtaining a commercial license
|
||||
// by contacting <sales@documize.com>.
|
||||
//
|
||||
// https://documize.com
|
||||
|
||||
package papertrail
|
||||
|
||||
import "strings"
|
||||
|
||||
// the HTML that is rendered by this section.
|
||||
const renderTemplate = `
|
||||
{{if .HasData}}
|
||||
<p class="margin-left-20">The <a href="https://papertrailapp.com">Papertrail log</a> for query <em>{{.Config.Query}}</em> contains {{.Count}} entries.</p>
|
||||
<table class="basic-table section-papertrail-table">
|
||||
<thead>
|
||||
<tr>
|
||||
<th class="bordered no-width">Date</th>
|
||||
<th class="bordered no-width">Severity</th>
|
||||
<th class="bordered">Message</th>
|
||||
</tr>
|
||||
</thead>
|
||||
<tbody>
|
||||
{{range $item := .Events}}
|
||||
<tr>
|
||||
<td class="bordered no-width color-gray">{{ $item.Dated }}</td>
|
||||
<td class="bordered no-width">{{ $item.Severity }}</td>
|
||||
<td class="bordered width-90">{{ $item.Message }}</td>
|
||||
</tr>
|
||||
{{end}}
|
||||
</tbody>
|
||||
</table>
|
||||
{{else}}
|
||||
<p>There are no Papertrail log entries to see.</p>
|
||||
{{end}}
|
||||
`
|
||||
|
||||
// Papertrail helpers
|
||||
type papertrailRender struct {
|
||||
Config papertrailConfig
|
||||
Events []papertrailEvent
|
||||
Count int
|
||||
Authenticated bool
|
||||
HasData bool
|
||||
}
|
||||
|
||||
type papertrailSearch struct {
|
||||
Events []papertrailEvent `json:"events"`
|
||||
}
|
||||
|
||||
type papertrailEvent struct {
|
||||
ID string `json:"id"`
|
||||
Dated string `json:"display_received_at"`
|
||||
Message string `json:"message"`
|
||||
Severity string `json:"severity"`
|
||||
}
|
||||
|
||||
type papertrailConfig struct {
|
||||
APIToken string `json:"APIToken"` // only contains the correct token just after it is typed in
|
||||
Query string `json:"query"`
|
||||
Max int `json:"max"`
|
||||
Group papertrailOption `json:"group"`
|
||||
System papertrailOption `json:"system"`
|
||||
}
|
||||
|
||||
func (c *papertrailConfig) Clean() {
|
||||
c.APIToken = strings.TrimSpace(c.APIToken)
|
||||
c.Query = strings.TrimSpace(c.Query)
|
||||
}
|
||||
|
||||
type papertrailOption struct {
|
||||
ID int `json:"id"`
|
||||
Name string `json:"name"`
|
||||
}
|
||||
|
||||
type papertrailOptions struct {
|
||||
Groups []papertrailOption `json:"groups"`
|
||||
Systems []papertrailOption `json:"systems"`
|
||||
}
|
||||
299
core/section/papertrail/papertrail.go
Normal file
299
core/section/papertrail/papertrail.go
Normal file
|
|
@ -0,0 +1,299 @@
|
|||
// Copyright 2016 Documize Inc. <legal@documize.com>. All rights reserved.
|
||||
//
|
||||
// This software (Documize Community Edition) is licensed under
|
||||
// GNU AGPL v3 http://www.gnu.org/licenses/agpl-3.0.en.html
|
||||
//
|
||||
// You can operate outside the AGPL restrictions by purchasing
|
||||
// Documize Enterprise Edition and obtaining a commercial license
|
||||
// by contacting <sales@documize.com>.
|
||||
//
|
||||
// https://documize.com
|
||||
|
||||
package papertrail
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"html/template"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"net/url"
|
||||
|
||||
"github.com/documize/community/core/section/provider"
|
||||
"github.com/documize/community/core/log"
|
||||
)
|
||||
|
||||
const me = "papertrail"
|
||||
|
||||
// Provider represents Gemini
|
||||
type Provider struct {
|
||||
}
|
||||
|
||||
// Meta describes us.
|
||||
func (*Provider) Meta() provider.TypeMeta {
|
||||
section := provider.TypeMeta{}
|
||||
section.ID = "db0a3a0a-b5d4-4d00-bfac-ee28abba451d"
|
||||
section.Title = "Papertrail"
|
||||
section.Description = "Display log entries"
|
||||
section.ContentType = "papertrail"
|
||||
|
||||
return section
|
||||
}
|
||||
|
||||
// Render converts Papertrail data into HTML suitable for browser rendering.
|
||||
func (*Provider) Render(ctx *provider.Context, config, data string) string {
|
||||
var search papertrailSearch
|
||||
var events []papertrailEvent
|
||||
var payload = papertrailRender{}
|
||||
var c = papertrailConfig{}
|
||||
|
||||
json.Unmarshal([]byte(data), &search)
|
||||
json.Unmarshal([]byte(config), &c)
|
||||
|
||||
c.APIToken = ctx.GetSecrets("APIToken")
|
||||
|
||||
max := len(search.Events)
|
||||
if c.Max < max {
|
||||
max = c.Max
|
||||
}
|
||||
|
||||
events = search.Events[:max]
|
||||
payload.Count = len(events)
|
||||
payload.HasData = payload.Count > 0
|
||||
|
||||
payload.Events = events
|
||||
payload.Config = c
|
||||
payload.Authenticated = c.APIToken != ""
|
||||
|
||||
t := template.New("items")
|
||||
t, _ = t.Parse(renderTemplate)
|
||||
|
||||
buffer := new(bytes.Buffer)
|
||||
t.Execute(buffer, payload)
|
||||
|
||||
return buffer.String()
|
||||
}
|
||||
|
||||
// Command handles authentication, workspace listing and items retrieval.
|
||||
func (p *Provider) Command(ctx *provider.Context, w http.ResponseWriter, r *http.Request) {
|
||||
query := r.URL.Query()
|
||||
method := query.Get("method")
|
||||
|
||||
if len(method) == 0 {
|
||||
provider.WriteMessage(w, me, "missing method name")
|
||||
return
|
||||
}
|
||||
|
||||
defer r.Body.Close()
|
||||
body, err := ioutil.ReadAll(r.Body)
|
||||
|
||||
if err != nil {
|
||||
provider.WriteMessage(w, me, "Bad payload")
|
||||
return
|
||||
}
|
||||
|
||||
var config = papertrailConfig{}
|
||||
err = json.Unmarshal(body, &config)
|
||||
|
||||
if err != nil {
|
||||
provider.WriteMessage(w, me, "Bad config")
|
||||
return
|
||||
}
|
||||
|
||||
config.Clean()
|
||||
|
||||
if config.APIToken == provider.SecretReplacement || config.APIToken == "" {
|
||||
config.APIToken = ctx.GetSecrets("APIToken")
|
||||
}
|
||||
|
||||
if len(config.APIToken) == 0 {
|
||||
provider.WriteMessage(w, me, "Missing API token")
|
||||
return
|
||||
}
|
||||
|
||||
switch method {
|
||||
case "auth":
|
||||
auth(ctx, config, w, r)
|
||||
case "options":
|
||||
options(config, w, r)
|
||||
}
|
||||
}
|
||||
|
||||
// Refresh just sends back data as-is.
|
||||
func (*Provider) Refresh(ctx *provider.Context, config, data string) (newData string) {
|
||||
var c = papertrailConfig{}
|
||||
err := json.Unmarshal([]byte(config), &c)
|
||||
|
||||
if err != nil {
|
||||
log.Error("unable to read Papertrail config", err)
|
||||
return
|
||||
}
|
||||
|
||||
c.Clean()
|
||||
|
||||
c.APIToken = ctx.GetSecrets("APIToken")
|
||||
|
||||
if len(c.APIToken) == 0 {
|
||||
log.Error("missing API token", err)
|
||||
return
|
||||
}
|
||||
|
||||
result, err := fetchEvents(c)
|
||||
|
||||
if err != nil {
|
||||
log.Error("Papertrail fetchEvents failed", err)
|
||||
return
|
||||
}
|
||||
|
||||
j, err := json.Marshal(result)
|
||||
|
||||
if err != nil {
|
||||
log.Error("unable to marshal Papaertrail events", err)
|
||||
return
|
||||
}
|
||||
|
||||
newData = string(j)
|
||||
return
|
||||
}
|
||||
|
||||
func auth(ctx *provider.Context, config papertrailConfig, w http.ResponseWriter, r *http.Request) {
|
||||
result, err := fetchEvents(config)
|
||||
|
||||
if result == nil {
|
||||
err = errors.New("nil result of papertrail query")
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
|
||||
log.IfErr(ctx.SaveSecrets(`{"APIToken":""}`)) // invalid token, so reset it
|
||||
|
||||
if err.Error() == "forbidden" {
|
||||
provider.WriteForbidden(w)
|
||||
} else {
|
||||
provider.WriteError(w, me, err)
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
log.IfErr(ctx.SaveSecrets(`{"APIToken":"` + config.APIToken + `"}`))
|
||||
|
||||
provider.WriteJSON(w, result)
|
||||
}
|
||||
|
||||
func options(config papertrailConfig, w http.ResponseWriter, r *http.Request) {
|
||||
// get systems
|
||||
req, err := http.NewRequest("GET", "https://papertrailapp.com/api/v1/systems.json", nil)
|
||||
req.Header.Set("X-Papertrail-Token", config.APIToken)
|
||||
|
||||
client := &http.Client{}
|
||||
res, err := client.Do(req)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
provider.WriteError(w, me, err)
|
||||
return
|
||||
}
|
||||
|
||||
if res.StatusCode != http.StatusOK {
|
||||
provider.WriteForbidden(w)
|
||||
return
|
||||
}
|
||||
|
||||
defer res.Body.Close()
|
||||
var systems []papertrailOption
|
||||
|
||||
dec := json.NewDecoder(res.Body)
|
||||
err = dec.Decode(&systems)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
provider.WriteError(w, me, err)
|
||||
return
|
||||
}
|
||||
|
||||
// get groups
|
||||
req, err = http.NewRequest("GET", "https://papertrailapp.com/api/v1/groups.json", nil)
|
||||
req.Header.Set("X-Papertrail-Token", config.APIToken)
|
||||
|
||||
client = &http.Client{}
|
||||
res, err = client.Do(req)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
provider.WriteError(w, me, err)
|
||||
return
|
||||
}
|
||||
|
||||
if res.StatusCode != http.StatusOK {
|
||||
provider.WriteForbidden(w)
|
||||
return
|
||||
}
|
||||
|
||||
defer res.Body.Close()
|
||||
var groups []papertrailOption
|
||||
|
||||
dec = json.NewDecoder(res.Body)
|
||||
err = dec.Decode(&groups)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
provider.WriteError(w, me, err)
|
||||
return
|
||||
}
|
||||
|
||||
var options = papertrailOptions{}
|
||||
options.Groups = groups
|
||||
options.Systems = systems
|
||||
|
||||
provider.WriteJSON(w, options)
|
||||
}
|
||||
|
||||
func fetchEvents(config papertrailConfig) (result interface{}, err error) {
|
||||
var filter string
|
||||
if len(config.Query) > 0 {
|
||||
filter = fmt.Sprintf("q=%s", url.QueryEscape(config.Query))
|
||||
}
|
||||
if config.Group.ID > 0 {
|
||||
prefix := ""
|
||||
if len(filter) > 0 {
|
||||
prefix = "&"
|
||||
}
|
||||
filter = fmt.Sprintf("%s%sgroup_id=%d", filter, prefix, config.Group.ID)
|
||||
}
|
||||
|
||||
var req *http.Request
|
||||
req, err = http.NewRequest("GET", "https://papertrailapp.com/api/v1/events/search.json?"+filter, nil)
|
||||
if err != nil {
|
||||
log.Error("new request", err)
|
||||
return
|
||||
}
|
||||
req.Header.Set("X-Papertrail-Token", config.APIToken)
|
||||
|
||||
client := &http.Client{}
|
||||
var res *http.Response
|
||||
res, err = client.Do(req)
|
||||
|
||||
if err != nil {
|
||||
log.Error("message", err)
|
||||
return
|
||||
}
|
||||
|
||||
if res.StatusCode != http.StatusOK {
|
||||
log.Error("forbidden", err)
|
||||
return
|
||||
}
|
||||
|
||||
defer res.Body.Close()
|
||||
|
||||
dec := json.NewDecoder(res.Body)
|
||||
err = dec.Decode(&result)
|
||||
|
||||
if err != nil {
|
||||
log.Error("unable to read result", err)
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue