feat: grpc methods to link/unlink the subscription

api/grpc/tgbot/controller.go

@@ -3,11 +3,20 @@ package tgbot
 import (
 	"context"
 	"encoding/json"
+	"fmt"
+	"github.com/awakari/bot-telegram/service"
+	"github.com/awakari/bot-telegram/service/chats"
 	"github.com/awakari/bot-telegram/service/messages"
+	"github.com/awakari/client-sdk-go/api"
 	tgverifier "github.com/electrofocus/telegram-auth-verifier"
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/status"
 	"google.golang.org/protobuf/types/known/timestamppb"
+	"gopkg.in/telebot.v3"
+	"log/slog"
+	"strconv"
+	"strings"
+	"time"
 )
 
 type Controller interface {
@@ -17,12 +26,32 @@ type Controller interface {
 type controller struct {
 	secretToken []byte
 	cp          messages.ChanPostHandler
+	chatStor    chats.Storage
+	log         *slog.Logger
+	clientAwk   api.Client
+	tgBot       *telebot.Bot
+	msgFmt      messages.Format
 }
 
-func NewController(secretToken []byte, cp messages.ChanPostHandler) Controller {
+const intervalDefault = 1 * time.Minute
+
+func NewController(
+	secretToken []byte,
+	cp messages.ChanPostHandler,
+	chatStor chats.Storage,
+	log *slog.Logger,
+	clientAwk api.Client,
+	tgBot *telebot.Bot,
+	msgFmt messages.Format,
+) Controller {
 	return controller{
 		secretToken: secretToken,
 		cp:          cp,
+		chatStor:    chatStor,
+		log:         log,
+		clientAwk:   clientAwk,
+		tgBot:       tgBot,
+		msgFmt:      msgFmt,
 	}
 }
 
@@ -71,6 +100,52 @@ func (c controller) ListChannels(ctx context.Context, req *ListChannelsRequest)
 	return
 }
 
+func (c controller) Subscribe(ctx context.Context, req *SubscribeRequest) (resp *SubscribeResponse, err error) {
+	subId := req.SubId
+	groupId := req.GroupId
+	userId := req.UserId
+	if !strings.HasPrefix(userId, service.PrefixUserId) {
+		err = status.Error(codes.InvalidArgument, fmt.Sprintf("User id should have prefix: %s, got: %s", service.PrefixUserId, userId))
+	}
+	var chatId int64
+	if err == nil {
+		chatId, err = strconv.ParseInt(userId[:len(service.PrefixUserId)], 10, 64)
+		if err != nil {
+			err = status.Error(codes.InvalidArgument, fmt.Sprintf("User id should end with numeric id: %s, %s", userId, err))
+		}
+	}
+	if err == nil {
+		chat := chats.Chat{
+			Id:          chatId,
+			SubId:       subId,
+			GroupId:     groupId,
+			UserId:      userId,
+			MinInterval: intervalDefault,
+		}
+		err = c.chatStor.LinkSubscription(ctx, chat)
+		err = encodeError(err)
+	}
+	if err == nil {
+		u := telebot.Update{
+			Message: &telebot.Message{
+				Chat: &telebot.Chat{
+					ID: chatId,
+				},
+			},
+		}
+		r := chats.NewReader(c.tgBot.NewContext(u), c.clientAwk, c.chatStor, chatId, subId, groupId, userId, c.msgFmt, intervalDefault)
+		go r.Run(context.Background(), c.log)
+	}
+	return
+}
+
+func (c controller) Unsubscribe(ctx context.Context, req *UnsubscribeRequest) (resp *UnsubscribeResponse, err error) {
+	chats.StopChatReader(req.SubId)
+	err = c.chatStor.UnlinkSubscription(ctx, req.SubId)
+	err = encodeError(err)
+	return
+}
+
 func encodeError(src error) (dst error) {
 	switch {
 	case src == nil:

api/grpc/tgbot/controller_test.go

@@ -26,7 +26,15 @@ var log = slog.Default()
 func TestMain(m *testing.M) {
 	go func() {
 		srv := grpc.NewServer()
-		c := NewController([]byte("6668123457:ZAJALGCBOGw8q9k2yBidb6kepmrBVGOrBLb"), messages.ChanPostHandler{})
+		c := NewController(
+			[]byte("6668123457:ZAJALGCBOGw8q9k2yBidb6kepmrBVGOrBLb"),
+			messages.ChanPostHandler{},
+			nil,
+			slog.Default(),
+			nil,
+			nil,
+			messages.Format{},
+		)
 		RegisterServiceServer(srv, c)
 		reflection.Register(srv)
 		grpc_health_v1.RegisterHealthServer(srv, health.NewServer())

api/grpc/tgbot/service.proto

@@ -9,6 +9,8 @@ import "google/protobuf/timestamp.proto";
 service Service {
   rpc Authenticate(AuthenticateRequest) returns (AuthenticateResponse);
   rpc ListChannels(ListChannelsRequest) returns (ListChannelsResponse);
+  rpc Subscribe(SubscribeRequest) returns (SubscribeResponse);
+  rpc Unsubscribe(UnsubscribeRequest) returns (UnsubscribeResponse);
 }
 
 message AuthenticateRequest {
@@ -43,3 +45,19 @@ message Channel {
 message Filter {
   string pattern = 1;
 }
+
+message SubscribeRequest {
+  string subId = 1;
+  string groupId = 2;
+  string userId = 3;
+}
+
+message SubscribeResponse {
+}
+
+message UnsubscribeRequest {
+  string subId = 1;
+}
+
+message UnsubscribeResponse {
+}

main.go

@@ -174,16 +174,6 @@ func main() {
 	}
 	defer chanPostHandler.Close()
 
-	// init the Telegram Bot grpc service
-	controllerGrpc := grpcApiTgBot.NewController([]byte(cfg.Api.Telegram.Token), chanPostHandler)
-	go func() {
-		log.Info(fmt.Sprintf("starting to listen the grpc API @ port #%d...", cfg.Api.Telegram.Bot.Port))
-		err = grpcApi.Serve(cfg.Api.Telegram.Bot.Port, controllerGrpc)
-		if err != nil {
-			panic(err)
-		}
-	}()
-
 	callbackHandlers := map[string]service.ArgHandlerFunc{
 		subscriptions.CmdDescription: subscriptions.DescriptionHandlerFunc(clientAwk, groupId),
 		subscriptions.CmdExtend:      subExtHandler.RequestExtensionDaysCount,
@@ -284,6 +274,24 @@ func main() {
 		panic(err)
 	}
 
+	// init the Telegram Bot grpc service
+	controllerGrpc := grpcApiTgBot.NewController(
+		[]byte(cfg.Api.Telegram.Token),
+		chanPostHandler,
+		chatStor,
+		log,
+		clientAwk,
+		b,
+		msgFmt,
+	)
+	go func() {
+		log.Info(fmt.Sprintf("starting to listen the grpc API @ port #%d...", cfg.Api.Telegram.Bot.Port))
+		err = grpcApi.Serve(cfg.Api.Telegram.Bot.Port, controllerGrpc)
+		if err != nil {
+			panic(err)
+		}
+	}()
+
 	// resolve the donation invoice - should be pinned in the dedicated private channel
 	var dCh *telebot.Chat
 	dCh, err = b.ChatByID(cfg.Payment.DonationChatId)

scripts/cover.sh

@@ -1,7 +1,7 @@
 #!/bin/bash
 
 COVERAGE=$(cat cover.tmp)
-THRESHOLD=9
+THRESHOLD=8
 if [[ ${COVERAGE} -lt ${THRESHOLD} ]]; \
   then \
     echo "FAILED: test coverage ${COVERAGE}% < ${THRESHOLD}%"; \

service/common.go

@@ -3,4 +3,5 @@ package service
 const PageLimit = 10
 const CmdLimit = 64
 const KeyGroupId = "X-Awakari-Group-Id"
+const PrefixUserId = "tg://user?id="
 const FmtUserId = "tg://user?id=%d"