2021-03-18 21:12:48 +01:00
|
|
|
package main
|
|
|
|
|
2021-03-18 22:14:56 +01:00
|
|
|
import (
|
|
|
|
"context"
|
2021-05-16 05:13:14 +02:00
|
|
|
"flag"
|
2021-04-19 21:25:34 +02:00
|
|
|
"fmt"
|
2021-03-18 22:14:56 +01:00
|
|
|
"log"
|
|
|
|
"net"
|
|
|
|
"os"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
pb "github.com/lbryio/hub/protobuf/go"
|
2021-04-19 21:25:34 +02:00
|
|
|
"github.com/lbryio/hub/server"
|
|
|
|
|
2021-03-18 22:14:56 +01:00
|
|
|
"google.golang.org/grpc"
|
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
|
|
|
port = ":50051"
|
|
|
|
)
|
2021-03-18 21:12:48 +01:00
|
|
|
|
2021-05-16 05:13:14 +02:00
|
|
|
func parseArgs(searchRequest *pb.SearchRequest) {
|
|
|
|
query:= flag.String("query", "", "query string")
|
|
|
|
claimType := flag.String("claimType", "", "claim type")
|
|
|
|
id := flag.String("id", "", "_id")
|
|
|
|
author := flag.String("author", "", "author")
|
|
|
|
title := flag.String("title", "", "title")
|
2021-05-18 12:02:55 +02:00
|
|
|
channelName := flag.String("channelName", "", "channel name")
|
|
|
|
description := flag.String("description", "", "description")
|
2021-05-16 05:13:14 +02:00
|
|
|
|
|
|
|
flag.Parse()
|
|
|
|
|
|
|
|
if *query != "" {
|
|
|
|
searchRequest.Query = *query
|
|
|
|
}
|
|
|
|
if *claimType != "" {
|
|
|
|
searchRequest.ClaimType = []string{*claimType}
|
|
|
|
}
|
|
|
|
if *id != "" {
|
|
|
|
searchRequest.XId = [][]byte{[]byte(*id)}
|
|
|
|
}
|
|
|
|
if *author != "" {
|
|
|
|
searchRequest.Author = []string{*author}
|
|
|
|
}
|
|
|
|
if *title != "" {
|
|
|
|
searchRequest.Title = []string{*title}
|
|
|
|
}
|
2021-05-18 12:02:55 +02:00
|
|
|
if *channelName != "" {
|
|
|
|
searchRequest.ChannelId = &pb.InvertibleField{Invert: false, Value: []string{*channelName}}
|
|
|
|
}
|
|
|
|
if *description != "" {
|
|
|
|
searchRequest.Description = []string{*description}
|
|
|
|
}
|
2021-05-16 05:13:14 +02:00
|
|
|
}
|
|
|
|
|
2021-03-18 21:12:48 +01:00
|
|
|
func main() {
|
2021-04-19 21:25:34 +02:00
|
|
|
if len(os.Args) == 2 && os.Args[1] == "serve" {
|
2021-03-18 22:14:56 +01:00
|
|
|
l, err := net.Listen("tcp", port)
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalf("failed to listen: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
s := grpc.NewServer()
|
2021-04-19 21:25:34 +02:00
|
|
|
pb.RegisterHubServer(s, &server.Server{})
|
2021-03-18 22:14:56 +01:00
|
|
|
|
|
|
|
log.Printf("listening on %s\n", l.Addr().String())
|
|
|
|
if err := s.Serve(l); err != nil {
|
|
|
|
log.Fatalf("failed to serve: %v", err)
|
|
|
|
}
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
conn, err := grpc.Dial("localhost"+port, grpc.WithInsecure(), grpc.WithBlock())
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalf("did not connect: %v", err)
|
|
|
|
}
|
|
|
|
defer conn.Close()
|
|
|
|
|
|
|
|
c := pb.NewHubClient(conn)
|
|
|
|
|
2021-05-16 05:13:14 +02:00
|
|
|
/*
|
2021-04-19 21:25:34 +02:00
|
|
|
var query string
|
2021-03-18 22:14:56 +01:00
|
|
|
if len(os.Args) > 1 {
|
2021-04-19 21:25:34 +02:00
|
|
|
query = strings.Join(os.Args[1:], " ")
|
|
|
|
} else {
|
|
|
|
log.Printf("error: no search query provided\n")
|
|
|
|
os.Exit(1)
|
2021-03-18 22:14:56 +01:00
|
|
|
}
|
2021-05-16 05:13:14 +02:00
|
|
|
*/
|
2021-03-18 22:14:56 +01:00
|
|
|
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second)
|
|
|
|
defer cancel()
|
|
|
|
|
2021-05-16 05:13:14 +02:00
|
|
|
searchRequest := &pb.SearchRequest{}
|
|
|
|
|
|
|
|
parseArgs(searchRequest)
|
|
|
|
|
2021-05-13 22:06:19 +02:00
|
|
|
r, err := c.Search(ctx, searchRequest)
|
2021-03-18 22:14:56 +01:00
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
log.Printf("found %d results\n", r.GetTotal())
|
2021-04-19 21:25:34 +02:00
|
|
|
|
|
|
|
for _, t := range r.Txos {
|
|
|
|
fmt.Printf("%s:%d\n", server.FromHash(t.TxHash), t.Nout)
|
|
|
|
}
|
2021-03-18 21:12:48 +01:00
|
|
|
}
|