usememos/memos
1
1
Fork 0
mirror of https://github.com/usememos/memos.git synced 2024-12-26 23:22:47 +08:00
Code Issues Projects Releases Packages Wiki Activity

chore: implement create resource

Browse source
This commit is contained in:
Steven 2024-01-15 08:13:06 +08:00
parent 3d23c01e26
commit 7ae4299df2
4 changed files with 44 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
api/v2/memo_resource_service.go
View file

@ -48,7 +48,7 @@ func (s *APIV2Service) SetMemoResources(ctx context.Context, request *apiv2pb.Se
MemoID: &request.Id,
UpdatedTs: &updatedTs,
}); err != nil {
return nil, status.Errorf(codes.Internal, "failed to update resource")
return nil, status.Errorf(codes.Internal, "failed to update resource: %v", err)
}
}

35
api/v2/resource_service.go
View file

@ -2,6 +2,7 @@ package v2
import (
"context"
"net/url"
"time"
"google.golang.org/grpc/codes"
@ -12,6 +13,40 @@ import (
"github.com/usememos/memos/store"
)
func (s *APIV2Service) CreateResource(ctx context.Context, request *apiv2pb.CreateResourceRequest) (*apiv2pb.CreateResourceResponse, error) {
user, err := getCurrentUser(ctx, s.Store)
if err != nil {
return nil, status.Errorf(codes.Internal, "failed to get current user: %v", err)
}
if request.ExternalLink != "" {
// Only allow those external links scheme with http/https
linkURL, err := url.Parse(request.ExternalLink)
if err != nil {
return nil, status.Errorf(codes.InvalidArgument, "invalid external link: %v", err)
}
if linkURL.Scheme != "http" && linkURL.Scheme != "https" {
return nil, status.Errorf(codes.InvalidArgument, "invalid external link scheme: %v", linkURL.Scheme)
}
}
create := &store.Resource{
CreatorID: user.ID,
Filename: request.Filename,
ExternalLink: request.ExternalLink,
Type: request.Type,
}
if request.MemoId != nil {
create.MemoID = request.MemoId
}
resource, err := s.Store.CreateResource(ctx, create)
if err != nil {
return nil, status.Errorf(codes.Internal, "failed to create resource: %v", err)
}
return &apiv2pb.CreateResourceResponse{
Resource: s.convertResourceFromStore(ctx, resource),
}, nil
}
func (s *APIV2Service) ListResources(ctx context.Context, _ *apiv2pb.ListResourcesRequest) (*apiv2pb.ListResourcesResponse, error) {
user, err := getCurrentUser(ctx, s.Store)
if err != nil {

4
web/src/helpers/api.ts
View file

@ -44,10 +44,6 @@ export function signout() {
return axios.post("/api/v1/auth/signout");
}
export function createResource(resourceCreate: ResourceCreate) {
return axios.post<Resource>("/api/v1/resource", resourceCreate);
}
export function createResourceWithBlob(formData: FormData) {
return axios.post<Resource>("/api/v1/resource/blob", formData);
}

13
web/src/store/module/resource.ts
View file

@ -1,6 +1,6 @@
import { resourceServiceClient } from "@/grpcweb";
import * as api from "@/helpers/api";
import { Resource, UpdateResourceRequest } from "@/types/proto/api/v2/resource_service";
import { CreateResourceRequest, Resource, UpdateResourceRequest } from "@/types/proto/api/v2/resource_service";
import { useTranslate } from "@/utils/i18n";
import store, { useAppSelector } from "../";
import { patchResource, setResources } from "../reducer/resource";
@ -17,8 +17,11 @@ export const useResourceStore = () => {
getState: () => {
return store.getState().resource;
},
async createResource(resourceCreate: ResourceCreate): Promise<Resource> {
const { data: resource } = await api.createResource(resourceCreate);
async createResource(create: CreateResourceRequest): Promise<Resource> {
const { resource } = await resourceServiceClient.createResource(create);
if (!resource) {
throw new Error("resource is null");
}
const resourceList = state.resources;
store.dispatch(setResources([resource, ...resourceList]));
return resource;
@ -36,8 +39,8 @@ export const useResourceStore = () => {
store.dispatch(setResources([resource, ...resourceList]));
return resource;
},
async updateResource(request: UpdateResourceRequest): Promise<Resource> {
const { resource } = await resourceServiceClient.updateResource(request);
async updateResource(update: UpdateResourceRequest): Promise<Resource> {
const { resource } = await resourceServiceClient.updateResource(update);
if (!resource) {
throw new Error("resource is null");
}