Franz Kafka
e5295fa69d
A search engine named after a man who proved answers don't exist. Renamed everywhere user-facing: - Brand name, UI titles, OpenSearch description, CSS filename - Docker service name, NixOS module (services.kafka) - Cache key prefix (kafka:), User-Agent strings (kafka/0.1) - README, config.example.toml, flake.nix descriptions Kept unchanged (internal): - Go module path: github.com/ashie/gosearch - Git repository URL: git.ashisgreat.xyz/penal-colony/gosearch - Binary entrypoint: cmd/searxng-go
120 lines
3.2 KiB
Go
120 lines
3.2 KiB
Go
package engines
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/ashie/gosearch/internal/contracts"
|
|
)
|
|
|
|
// RedditEngine searches Reddit posts via the public JSON API.
|
|
type RedditEngine struct {
|
|
client *http.Client
|
|
}
|
|
|
|
func (e *RedditEngine) Name() string { return "reddit" }
|
|
|
|
func (e *RedditEngine) Search(ctx context.Context, req contracts.SearchRequest) (contracts.SearchResponse, error) {
|
|
if strings.TrimSpace(req.Query) == "" {
|
|
return contracts.SearchResponse{Query: req.Query}, nil
|
|
}
|
|
if e == nil || e.client == nil {
|
|
return contracts.SearchResponse{}, errors.New("reddit engine not initialized")
|
|
}
|
|
|
|
endpoint := fmt.Sprintf(
|
|
"https://www.reddit.com/search.json?q=%s&limit=25&sort=relevance&t=all",
|
|
url.QueryEscape(req.Query),
|
|
)
|
|
|
|
httpReq, err := http.NewRequestWithContext(ctx, http.MethodGet, endpoint, nil)
|
|
if err != nil {
|
|
return contracts.SearchResponse{}, err
|
|
}
|
|
httpReq.Header.Set("User-Agent", "kafka/0.1 (compatible; +https://git.ashisgreat.xyz/penal-colony/gosearch)")
|
|
|
|
resp, err := e.client.Do(httpReq)
|
|
if err != nil {
|
|
return contracts.SearchResponse{}, err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
body, _ := io.ReadAll(io.LimitReader(resp.Body, 4096))
|
|
return contracts.SearchResponse{}, fmt.Errorf("reddit api error: status=%d body=%q", resp.StatusCode, string(body))
|
|
}
|
|
|
|
var data struct {
|
|
Data struct {
|
|
Children []struct {
|
|
Data struct {
|
|
Title string `json:"title"`
|
|
URL string `json:"url"`
|
|
Permalink string `json:"permalink"`
|
|
Score int `json:"score"`
|
|
NumComments int `json:"num_comments"`
|
|
Subreddit string `json:"subreddit"`
|
|
CreatedUTC float64 `json:"created_utc"`
|
|
IsSelf bool `json:"is_self"`
|
|
Over18 bool `json:"over_18"`
|
|
} `json:"data"`
|
|
} `json:"children"`
|
|
} `json:"data"`
|
|
}
|
|
|
|
if err := json.NewDecoder(resp.Body).Decode(&data); err != nil {
|
|
return contracts.SearchResponse{}, err
|
|
}
|
|
|
|
results := make([]contracts.MainResult, 0, len(data.Data.Children))
|
|
for _, child := range data.Data.Children {
|
|
post := child.Data
|
|
|
|
// Skip NSFW results unless explicitly allowed.
|
|
if post.Over18 && req.Safesearch > 0 {
|
|
continue
|
|
}
|
|
|
|
// For self-posts, link to the Reddit thread.
|
|
linkURL := post.URL
|
|
if post.IsSelf || strings.HasPrefix(linkURL, "/r/") {
|
|
linkURL = "https://www.reddit.com" + post.Permalink
|
|
}
|
|
|
|
content := fmt.Sprintf("r/%s · ⬆ %d · 💬 %d", post.Subreddit, post.Score, post.NumComments)
|
|
if req.Safesearch == 0 {
|
|
// No additional content for safe mode
|
|
}
|
|
|
|
title := post.Title
|
|
urlPtr := linkURL
|
|
|
|
results = append(results, contracts.MainResult{
|
|
Template: "default.html",
|
|
Title: title,
|
|
Content: content,
|
|
URL: &urlPtr,
|
|
Engine: "reddit",
|
|
Score: float64(post.Score),
|
|
Category: "general",
|
|
Engines: []string{"reddit"},
|
|
})
|
|
}
|
|
|
|
return contracts.SearchResponse{
|
|
Query: req.Query,
|
|
NumberOfResults: len(results),
|
|
Results: results,
|
|
Answers: []map[string]any{},
|
|
Corrections: []string{},
|
|
Infoboxes: []map[string]any{},
|
|
Suggestions: []string{},
|
|
UnresponsiveEngines: [][2]string{},
|
|
}, nil
|
|
}
|