package go_sensitive

import (
	"net"

	pb "go_sensitive/pb"

	log "github.com/sirupsen/logrus"
	"golang.org/x/net/context"
	"google.golang.org/grpc"
)

const VERSION = "v1.0.0"

type App struct {
	pb.SensitiveServer
}

func New() *App {
	return &App{}
}

func (app *App) Run(addr string) {
	//加载词典
	Setup()

	listen, err := net.Listen("tcp", addr)
	if err != nil {
		log.WithError(err).WithField("addr", addr).Fatal("faild listen")
	}

	// 实例化grpc Server
	s := grpc.NewServer()

	// 注册Service
	pb.RegisterSensitiveServer(s, app)

	log.WithField("addr", addr).Info("listen on")
	if err := s.Serve(listen); err != nil {
		log.Fatalf("failed to serve: %v", err)
	}

}
func (app *App) Contains(ctx context.Context, t *pb.ContainsRequest) (*pb.ContainsResponse, error) {
	response := new(pb.ContainsResponse)
	if Util.Contains(t.GetText()) {
		response.Text = "true"
	} else {
		response.Text = "false"
	}
	return response, nil
}