pkgdash/handler/handlers.go

package handler

import (
	"context"
	"io"
	"net/http"

	cmsstorage "go.unistack.org/cms-service/storage"
	"go.unistack.org/micro/v3"
	"go.unistack.org/micro/v3/errors"
	"go.unistack.org/unistack-org/pkgdash/config"
	pb "go.unistack.org/unistack-org/pkgdash/proto/go_generate"
	cligit "go.unistack.org/unistack-org/pkgdash/service/client_git"
	"go.unistack.org/unistack-org/pkgdash/storage"
	"google.golang.org/protobuf/encoding/protojson"
)

type Handler struct {
	svc   micro.Service
	store storage.Storage

	writer writer
	protojson.MarshalOptions
	protojson.UnmarshalOptions

	git     cligit.Client
	chanUrl chan *pb.AddPackageRsp
}

func (h *Handler) ListPackage(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	logger := h.svc.Logger()
	logger.Debug(ctx, "Start getListPackage")

	dbRsp, err := h.store.ListPackage(ctx)
	if err != nil {
		logger.Errorf(ctx, "error db response: %v", err)
		h.writer.Response(ctx, w, err)
		return
	}

	rsp := new(pb.ListPackageRsp)
	rsp.Packages = dbRsp.Mapping()

	logger.Debug(ctx, "Success finish getListPackage")
	h.writer.Response(ctx, w, rsp)
}

func (h *Handler) UpdatePackage(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	logger := h.svc.Logger()
	logger.Debug(ctx, "Start UpdatePackage")

	defer r.Body.Close()
	all, err := io.ReadAll(r.Body)
	if err != nil {
		logger.Error(ctx, err)
		h.writer.Response(ctx, w, NewInternalError(err))
		return
	}

	rsp := new(pb.UpdatePackageRsp)
	if err = h.Unmarshal(all, rsp); err != nil {
		logger.Error(ctx, err)
		h.writer.Response(ctx, w, NewUnmarshalError(err))
		return
	}

	if err = rsp.Validate(); err != nil {
		logger.Error(ctx, err)
		h.writer.Response(ctx, w, NewValidationError(err))
		return
	}

	if err = h.store.UpdatePackage(ctx, rsp); err != nil {
		logger.Error(ctx, err)
		h.writer.Response(ctx, w, NewInternalError(err))
		return
	}

	logger.Debug(ctx, "Success finish UpdatePackage")
}

func (h *Handler) AddComment(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	logger := h.svc.Logger()
	logger.Debug(ctx, "Start AddComment")

	defer r.Body.Close()
	all, err := io.ReadAll(r.Body)
	if err != nil {
		logger.Error(ctx, err)
		h.writer.Response(ctx, w, NewInternalError(err))
		return
	}

	rsp := new(pb.AddCommentRsp)
	if err = h.Unmarshal(all, rsp); err != nil {
		logger.Error(ctx, err)
		h.writer.Response(ctx, w, NewUnmarshalError(err))
		return
	}

	if err = rsp.Validate(); err != nil {
		logger.Error(ctx, err)
		h.writer.Response(ctx, w, NewValidationError(err))
		return
	}

	if err = h.store.AddComment(ctx, rsp); err != nil {
		logger.Error(ctx, err)
		h.writer.Response(ctx, w, NewInternalError(err))
		return
	}

	logger.Debug(ctx, "Success finish addComment")
}

func (h *Handler) AddPackage(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	logger := h.svc.Logger()
	logger.Debug(ctx, "Start AddPackage")

	defer r.Body.Close()
	all, err := io.ReadAll(r.Body)
	if err != nil {
		logger.Error(ctx, err)
		h.writer.Response(ctx, w, NewInternalError(err))
		return
	}

	req := new(pb.AddPackageRsp)
	if err = h.Unmarshal(all, req); err != nil {
		logger.Error(ctx, err)
		h.writer.Response(ctx, w, NewUnmarshalError(err))
		return
	}

	if err = req.Validate(); err != nil {
		logger.Error(ctx, err)
		h.writer.Response(ctx, w, NewValidationError(err))
		return
	}

	if h.git.IsClose() {
		logger.Error(ctx, "chan is closed")
	} else {
		h.chanUrl <- req
	}

	logger.Debug(ctx, "Success finish addPackage")
}

func NewHandler(svc micro.Service, w writer, client cligit.Client) *Handler {
	h := &Handler{
		svc:    svc,
		writer: w,
		git:    client,
	}
	h.EmitUnpopulated = true
	h.UseProtoNames = false
	return h
}

func (h *Handler) Init(ctx context.Context) error {
	store := cmsstorage.InterfaceFromContext(h.svc.Options().Context)
	if store == nil {
		return cmsstorage.ErrMissingStorage
	}
	st, ok := store.(storage.Storage)
	if !ok {
		return errors.New(config.ServiceName, "error init storage", 1)
	}

	h.chanUrl = h.git.Run(ctx, st)

	h.store = st

	return nil
}