diff --git a/formatter.go b/formatter.go index b929ba5..02cef31 100644 --- a/formatter.go +++ b/formatter.go @@ -30,7 +30,11 @@ func (fileContent *FileContent) FormatContent() string { for i, match := range re.FindAllSubmatch(data, -1) { if "!" == string(match[0][0:1]) { - link := fmt.Sprintf("=> %s %s\n", match[2], match[1]) + uPath := string(match[2][:]) + if strings.HasPrefix(uPath, "__a") { + uPath = "/" + uPath + } + link := fmt.Sprintf("=> %s %s\n", uPath, match[1]) data = bytes.Replace(data, match[0], []byte(link), 1) } else { replaceWithIndex := append(match[1], fmt.Sprintf("[%d]", i+1)...) diff --git a/main.go b/main.go index 11deff8..f89fa30 100644 --- a/main.go +++ b/main.go @@ -5,6 +5,7 @@ import ( "flag" "fmt" "google.golang.org/grpc/credentials/insecure" + "io" "log" "regexp" "strings" @@ -26,6 +27,10 @@ var ( key = "WgiPGpt5wM6SVEWo5iqI" ) +const DefaultLang = "sgs" + +var SupportedLang = []string{"en", "sgs", "lt", "prg", "eo"} + func init() { flag.StringVar(&hostname, "hostname", "", "capsule hostname") flag.StringVar(&port, "port", "1965", "capsule port") @@ -109,6 +114,8 @@ func process(_ context.Context, w gemini.ResponseWriter, r *gemini.Request) { lang := regexp.MustCompile(`^/(sgs|en)`).FindString(r.URL.Path) if lang != "" { lang = lang[1:] + } else { + lang = DefaultLang } conn, err := grpc.Dial(fileSrvHost+":"+fileSrvPort, grpc.WithTransportCredentials(insecure.NewCredentials())) @@ -120,35 +127,68 @@ func process(_ context.Context, w gemini.ResponseWriter, r *gemini.Request) { defer conn.Close() client := NewTreeManagerClient(conn) + langs := strings.Join(SupportedLang, "|") switch { case "/" == r.URL.Path: - w.WriteHeader(gemini.StatusPermanentRedirect, "/sgs") - case regexp.MustCompile(`^/(sgs|en)/?$`).MatchString(r.URL.Path): + w.WriteHeader(gemini.StatusPermanentRedirect, "/"+DefaultLang) + case regexp.MustCompile(`^/(` + langs + `)/?$`).MatchString(r.URL.Path): renderIndex(lang, w, client) - case regexp.MustCompile(`^/(sgs|en)/atom.xml$`).MatchString(r.URL.Path): + case regexp.MustCompile(`^/(` + langs + `)/atom.xml$`).MatchString(r.URL.Path): renderFeed(lang, w, r, client) - case regexp.MustCompile(`^/(sgs|en)/a/?$`).MatchString(r.URL.Path): + case regexp.MustCompile(`^/(` + langs + `)/a/?$`).MatchString(r.URL.Path): renderAbout(lang, w) - case regexp.MustCompile(`^/(sgs|en)/s/?$`).MatchString(r.URL.Path): + case regexp.MustCompile(`^/(` + langs + `)/s/?$`).MatchString(r.URL.Path): renderSearch(lang, w, r, client) - case regexp.MustCompile(`^/(sgs|en)/f/?$`).MatchString(r.URL.Path): + case regexp.MustCompile(`^/(` + langs + `)/f/?$`).MatchString(r.URL.Path): renderAllFiles(lang, w, client) - case regexp.MustCompile(`^/(sgs|en)/f/([\p{L}\d_+.]+/)+[\d\w]+/index.gmi$`).MatchString(r.URL.Path): + case regexp.MustCompile(`^/(` + langs + `)/f/([\p{L}\d_+.]+/)+[\d\w]+/index.gmi$`).MatchString(r.URL.Path): urlParts := strings.Split(r.URL.Path, "/") w.WriteHeader(gemini.StatusPermanentRedirect, strings.Join(urlParts[:(len(urlParts)-2)], "/")) - case regexp.MustCompile(`^/(sgs|en)/f/([\p{L}\d_+.]+/)+[\d\w]+/[\p{L}\d_+.]+.gmi$`).MatchString(r.URL.Path): + case regexp.MustCompile(`^/(` + langs + `)/f/([\p{L}\d_+.]+/)+[\d\w]+/[\p{L}\d_+.]+.gmi$`).MatchString(r.URL.Path): renderFile(lang, w, r, client) - case regexp.MustCompile(`^/(sgs|en)/f/[\p{L}\d_+.]+(/[\p{L}\d_+.]+)*/?$`).MatchString(r.URL.Path): + case regexp.MustCompile(`^/(` + langs + `)/f/[\p{L}\d_+.]+(/[\p{L}\d_+.]+)*/?$`).MatchString(r.URL.Path): renderCategory(lang, w, r, client) - case regexp.MustCompile(`^/(sgs|en)/t/[\p{L}\d_+.]+/?$`).MatchString(r.URL.Path): + case regexp.MustCompile(`^/(` + langs + `)/t/[\p{L}\d_+.]+/?$`).MatchString(r.URL.Path): renderTag(lang, w, r, client) - // case abruozdelee + case regexp.MustCompile(`^/__a/`).MatchString(r.URL.Path): + downloadAttachment(w, r, client) default: w.WriteHeader(gemini.StatusNotFound, "Out of space") } } +func downloadAttachment(w gemini.ResponseWriter, r *gemini.Request, client TreeManagerClient) { + urlParts := strings.Split(strings.Trim(r.URL.Path, "/"), "/") + filePath := urlParts[len(urlParts)-1] + ar := AttachmentRequest{Path: filePath} + response, err := client.DownloadAttachment(context.Background(), &ar) + if err != nil { + log.Fatalf("client.DownloadAttachment failed: %v", err) + w.WriteHeader(gemini.StatusNotFound, "We lost it!") + return + } + + //w.SetMediaType(http.DetectContentType([]byte(response.Content))) + w.SetMediaType("image/png") + for { + chunkResponse, err := response.Recv() + if err == io.EOF { + log.Println("received all chunks") + break + } + if err != nil { + log.Println("err receiving chunk:", err) + break + } + _, err = w.Write(chunkResponse.Chunk) + if err != nil { + w.WriteHeader(gemini.StatusTemporaryFailure, "Internal server error") + return + } + } +} + func renderIndex(lang string, w gemini.ResponseWriter, client TreeManagerClient) { filters := []*TreeRequest_Filter{{Key: "lang", Value: lang}} if lang == "sgs" { diff --git a/tree.pb.go b/tree.pb.go index 6c7598f..582f456 100644 --- a/tree.pb.go +++ b/tree.pb.go @@ -1,7 +1,7 @@ // Code generated by protoc-gen-go. DO NOT EDIT. // versions: -// protoc-gen-go v1.28.0 -// protoc v3.21.4 +// protoc-gen-go v1.30.0 +// protoc v3.21.12 // source: tree.proto package main @@ -461,6 +461,100 @@ func (x *FileContent) GetContent() string { return "" } +type AttachmentRequest struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + Path string `protobuf:"bytes,1,opt,name=path,proto3" json:"path,omitempty"` +} + +func (x *AttachmentRequest) Reset() { + *x = AttachmentRequest{} + if protoimpl.UnsafeEnabled { + mi := &file_tree_proto_msgTypes[7] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *AttachmentRequest) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*AttachmentRequest) ProtoMessage() {} + +func (x *AttachmentRequest) ProtoReflect() protoreflect.Message { + mi := &file_tree_proto_msgTypes[7] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use AttachmentRequest.ProtoReflect.Descriptor instead. +func (*AttachmentRequest) Descriptor() ([]byte, []int) { + return file_tree_proto_rawDescGZIP(), []int{7} +} + +func (x *AttachmentRequest) GetPath() string { + if x != nil { + return x.Path + } + return "" +} + +type AttachmentResponse struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + Chunk []byte `protobuf:"bytes,1,opt,name=chunk,proto3" json:"chunk,omitempty"` +} + +func (x *AttachmentResponse) Reset() { + *x = AttachmentResponse{} + if protoimpl.UnsafeEnabled { + mi := &file_tree_proto_msgTypes[8] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *AttachmentResponse) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*AttachmentResponse) ProtoMessage() {} + +func (x *AttachmentResponse) ProtoReflect() protoreflect.Message { + mi := &file_tree_proto_msgTypes[8] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use AttachmentResponse.ProtoReflect.Descriptor instead. +func (*AttachmentResponse) Descriptor() ([]byte, []int) { + return file_tree_proto_rawDescGZIP(), []int{8} +} + +func (x *AttachmentResponse) GetChunk() []byte { + if x != nil { + return x.Chunk + } + return nil +} + type TreeRequest_Filter struct { state protoimpl.MessageState sizeCache protoimpl.SizeCache @@ -473,7 +567,7 @@ type TreeRequest_Filter struct { func (x *TreeRequest_Filter) Reset() { *x = TreeRequest_Filter{} if protoimpl.UnsafeEnabled { - mi := &file_tree_proto_msgTypes[7] + mi := &file_tree_proto_msgTypes[9] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } @@ -486,7 +580,7 @@ func (x *TreeRequest_Filter) String() string { func (*TreeRequest_Filter) ProtoMessage() {} func (x *TreeRequest_Filter) ProtoReflect() protoreflect.Message { - mi := &file_tree_proto_msgTypes[7] + mi := &file_tree_proto_msgTypes[9] if protoimpl.UnsafeEnabled && x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { @@ -576,19 +670,29 @@ var file_tree_proto_rawDesc = []byte{ 0x04, 0x66, 0x69, 0x6c, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x0e, 0x2e, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x54, 0x72, 0x65, 0x65, 0x46, 0x69, 0x6c, 0x65, 0x52, 0x04, 0x66, 0x69, 0x6c, 0x65, 0x12, 0x18, 0x0a, 0x07, 0x63, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x18, 0x02, 0x20, 0x01, - 0x28, 0x09, 0x52, 0x07, 0x63, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x32, 0xa9, 0x01, 0x0a, 0x0b, - 0x54, 0x72, 0x65, 0x65, 0x4d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, 0x12, 0x2d, 0x0a, 0x0a, 0x47, - 0x65, 0x74, 0x53, 0x75, 0x6d, 0x6d, 0x65, 0x72, 0x79, 0x12, 0x11, 0x2e, 0x6d, 0x61, 0x69, 0x6e, - 0x2e, 0x54, 0x72, 0x65, 0x65, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x0a, 0x2e, 0x6d, - 0x61, 0x69, 0x6e, 0x2e, 0x54, 0x72, 0x65, 0x65, 0x22, 0x00, 0x12, 0x31, 0x0a, 0x07, 0x47, 0x65, - 0x74, 0x46, 0x69, 0x6c, 0x65, 0x12, 0x11, 0x2e, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x46, 0x69, 0x6c, - 0x65, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x11, 0x2e, 0x6d, 0x61, 0x69, 0x6e, 0x2e, - 0x46, 0x69, 0x6c, 0x65, 0x43, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x22, 0x00, 0x12, 0x38, 0x0a, - 0x0b, 0x52, 0x65, 0x62, 0x75, 0x69, 0x6c, 0x64, 0x54, 0x72, 0x65, 0x65, 0x12, 0x10, 0x2e, 0x6d, - 0x61, 0x69, 0x6e, 0x2e, 0x54, 0x72, 0x65, 0x65, 0x53, 0x65, 0x63, 0x72, 0x65, 0x74, 0x1a, 0x15, - 0x2e, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x52, 0x65, 0x62, 0x75, 0x69, 0x6c, 0x64, 0x52, 0x65, 0x73, - 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, 0x42, 0x03, 0x5a, 0x01, 0x2e, 0x62, 0x06, 0x70, 0x72, - 0x6f, 0x74, 0x6f, 0x33, + 0x28, 0x09, 0x52, 0x07, 0x63, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x22, 0x27, 0x0a, 0x11, 0x41, + 0x74, 0x74, 0x61, 0x63, 0x68, 0x6d, 0x65, 0x6e, 0x74, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, + 0x12, 0x12, 0x0a, 0x04, 0x70, 0x61, 0x74, 0x68, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x04, + 0x70, 0x61, 0x74, 0x68, 0x22, 0x2a, 0x0a, 0x12, 0x41, 0x74, 0x74, 0x61, 0x63, 0x68, 0x6d, 0x65, + 0x6e, 0x74, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x14, 0x0a, 0x05, 0x63, 0x68, + 0x75, 0x6e, 0x6b, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x05, 0x63, 0x68, 0x75, 0x6e, 0x6b, + 0x32, 0xf6, 0x01, 0x0a, 0x0b, 0x54, 0x72, 0x65, 0x65, 0x4d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, + 0x12, 0x2d, 0x0a, 0x0a, 0x47, 0x65, 0x74, 0x53, 0x75, 0x6d, 0x6d, 0x65, 0x72, 0x79, 0x12, 0x11, + 0x2e, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x54, 0x72, 0x65, 0x65, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, + 0x74, 0x1a, 0x0a, 0x2e, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x54, 0x72, 0x65, 0x65, 0x22, 0x00, 0x12, + 0x31, 0x0a, 0x07, 0x47, 0x65, 0x74, 0x46, 0x69, 0x6c, 0x65, 0x12, 0x11, 0x2e, 0x6d, 0x61, 0x69, + 0x6e, 0x2e, 0x46, 0x69, 0x6c, 0x65, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x11, 0x2e, + 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x46, 0x69, 0x6c, 0x65, 0x43, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, + 0x22, 0x00, 0x12, 0x38, 0x0a, 0x0b, 0x52, 0x65, 0x62, 0x75, 0x69, 0x6c, 0x64, 0x54, 0x72, 0x65, + 0x65, 0x12, 0x10, 0x2e, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x54, 0x72, 0x65, 0x65, 0x53, 0x65, 0x63, + 0x72, 0x65, 0x74, 0x1a, 0x15, 0x2e, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x52, 0x65, 0x62, 0x75, 0x69, + 0x6c, 0x64, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, 0x12, 0x4b, 0x0a, 0x12, + 0x44, 0x6f, 0x77, 0x6e, 0x6c, 0x6f, 0x61, 0x64, 0x41, 0x74, 0x74, 0x61, 0x63, 0x68, 0x6d, 0x65, + 0x6e, 0x74, 0x12, 0x17, 0x2e, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x41, 0x74, 0x74, 0x61, 0x63, 0x68, + 0x6d, 0x65, 0x6e, 0x74, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x18, 0x2e, 0x6d, 0x61, + 0x69, 0x6e, 0x2e, 0x41, 0x74, 0x74, 0x61, 0x63, 0x68, 0x6d, 0x65, 0x6e, 0x74, 0x52, 0x65, 0x73, + 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, 0x30, 0x01, 0x42, 0x03, 0x5a, 0x01, 0x2e, 0x62, 0x06, + 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, } var ( @@ -603,7 +707,7 @@ func file_tree_proto_rawDescGZIP() []byte { return file_tree_proto_rawDescData } -var file_tree_proto_msgTypes = make([]protoimpl.MessageInfo, 10) +var file_tree_proto_msgTypes = make([]protoimpl.MessageInfo, 12) var file_tree_proto_goTypes = []interface{}{ (*TreeSecret)(nil), // 0: main.TreeSecret (*RebuildResponse)(nil), // 1: main.RebuildResponse @@ -612,28 +716,32 @@ var file_tree_proto_goTypes = []interface{}{ (*TreeFile)(nil), // 4: main.TreeFile (*Tree)(nil), // 5: main.Tree (*FileContent)(nil), // 6: main.FileContent - (*TreeRequest_Filter)(nil), // 7: main.TreeRequest.Filter - nil, // 8: main.Tree.TagsEntry - nil, // 9: main.Tree.CategoriesEntry + (*AttachmentRequest)(nil), // 7: main.AttachmentRequest + (*AttachmentResponse)(nil), // 8: main.AttachmentResponse + (*TreeRequest_Filter)(nil), // 9: main.TreeRequest.Filter + nil, // 10: main.Tree.TagsEntry + nil, // 11: main.Tree.CategoriesEntry } var file_tree_proto_depIdxs = []int32{ - 7, // 0: main.TreeRequest.filter:type_name -> main.TreeRequest.Filter - 4, // 1: main.Tree.files:type_name -> main.TreeFile - 4, // 2: main.Tree.rootFiles:type_name -> main.TreeFile - 8, // 3: main.Tree.tags:type_name -> main.Tree.TagsEntry - 9, // 4: main.Tree.categories:type_name -> main.Tree.CategoriesEntry - 4, // 5: main.FileContent.file:type_name -> main.TreeFile - 3, // 6: main.TreeManager.GetSummery:input_type -> main.TreeRequest - 2, // 7: main.TreeManager.GetFile:input_type -> main.FileRequest - 0, // 8: main.TreeManager.RebuildTree:input_type -> main.TreeSecret - 5, // 9: main.TreeManager.GetSummery:output_type -> main.Tree - 6, // 10: main.TreeManager.GetFile:output_type -> main.FileContent - 1, // 11: main.TreeManager.RebuildTree:output_type -> main.RebuildResponse - 9, // [9:12] is the sub-list for method output_type - 6, // [6:9] is the sub-list for method input_type - 6, // [6:6] is the sub-list for extension type_name - 6, // [6:6] is the sub-list for extension extendee - 0, // [0:6] is the sub-list for field type_name + 9, // 0: main.TreeRequest.filter:type_name -> main.TreeRequest.Filter + 4, // 1: main.Tree.files:type_name -> main.TreeFile + 4, // 2: main.Tree.rootFiles:type_name -> main.TreeFile + 10, // 3: main.Tree.tags:type_name -> main.Tree.TagsEntry + 11, // 4: main.Tree.categories:type_name -> main.Tree.CategoriesEntry + 4, // 5: main.FileContent.file:type_name -> main.TreeFile + 3, // 6: main.TreeManager.GetSummery:input_type -> main.TreeRequest + 2, // 7: main.TreeManager.GetFile:input_type -> main.FileRequest + 0, // 8: main.TreeManager.RebuildTree:input_type -> main.TreeSecret + 7, // 9: main.TreeManager.DownloadAttachment:input_type -> main.AttachmentRequest + 5, // 10: main.TreeManager.GetSummery:output_type -> main.Tree + 6, // 11: main.TreeManager.GetFile:output_type -> main.FileContent + 1, // 12: main.TreeManager.RebuildTree:output_type -> main.RebuildResponse + 8, // 13: main.TreeManager.DownloadAttachment:output_type -> main.AttachmentResponse + 10, // [10:14] is the sub-list for method output_type + 6, // [6:10] is the sub-list for method input_type + 6, // [6:6] is the sub-list for extension type_name + 6, // [6:6] is the sub-list for extension extendee + 0, // [0:6] is the sub-list for field type_name } func init() { file_tree_proto_init() } @@ -727,6 +835,30 @@ func file_tree_proto_init() { } } file_tree_proto_msgTypes[7].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*AttachmentRequest); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + file_tree_proto_msgTypes[8].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*AttachmentResponse); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + file_tree_proto_msgTypes[9].Exporter = func(v interface{}, i int) interface{} { switch v := v.(*TreeRequest_Filter); i { case 0: return &v.state @@ -746,7 +878,7 @@ func file_tree_proto_init() { GoPackagePath: reflect.TypeOf(x{}).PkgPath(), RawDescriptor: file_tree_proto_rawDesc, NumEnums: 0, - NumMessages: 10, + NumMessages: 12, NumExtensions: 0, NumServices: 1, }, diff --git a/tree.proto b/tree.proto index e135fa9..45aaeb9 100644 --- a/tree.proto +++ b/tree.proto @@ -11,6 +11,7 @@ service TreeManager { rpc GetSummery(TreeRequest) returns (Tree) {} rpc GetFile(FileRequest) returns (FileContent) {} rpc RebuildTree(TreeSecret) returns (RebuildResponse) {} + rpc DownloadAttachment(AttachmentRequest) returns (stream AttachmentResponse) {} } message TreeSecret { @@ -59,3 +60,11 @@ message FileContent { TreeFile file = 1; string content = 2; } + +message AttachmentRequest { + string path = 1; +} + +message AttachmentResponse { + bytes chunk = 1; +} diff --git a/tree_grpc.pb.go b/tree_grpc.pb.go index ba9822e..0f284c9 100644 --- a/tree_grpc.pb.go +++ b/tree_grpc.pb.go @@ -1,7 +1,7 @@ // Code generated by protoc-gen-go-grpc. DO NOT EDIT. // versions: // - protoc-gen-go-grpc v1.2.0 -// - protoc v3.21.4 +// - protoc v3.21.12 // source: tree.proto package main @@ -25,6 +25,7 @@ type TreeManagerClient interface { GetSummery(ctx context.Context, in *TreeRequest, opts ...grpc.CallOption) (*Tree, error) GetFile(ctx context.Context, in *FileRequest, opts ...grpc.CallOption) (*FileContent, error) RebuildTree(ctx context.Context, in *TreeSecret, opts ...grpc.CallOption) (*RebuildResponse, error) + DownloadAttachment(ctx context.Context, in *AttachmentRequest, opts ...grpc.CallOption) (TreeManager_DownloadAttachmentClient, error) } type treeManagerClient struct { @@ -62,6 +63,38 @@ func (c *treeManagerClient) RebuildTree(ctx context.Context, in *TreeSecret, opt return out, nil } +func (c *treeManagerClient) DownloadAttachment(ctx context.Context, in *AttachmentRequest, opts ...grpc.CallOption) (TreeManager_DownloadAttachmentClient, error) { + stream, err := c.cc.NewStream(ctx, &TreeManager_ServiceDesc.Streams[0], "/main.TreeManager/DownloadAttachment", opts...) + if err != nil { + return nil, err + } + x := &treeManagerDownloadAttachmentClient{stream} + if err := x.ClientStream.SendMsg(in); err != nil { + return nil, err + } + if err := x.ClientStream.CloseSend(); err != nil { + return nil, err + } + return x, nil +} + +type TreeManager_DownloadAttachmentClient interface { + Recv() (*AttachmentResponse, error) + grpc.ClientStream +} + +type treeManagerDownloadAttachmentClient struct { + grpc.ClientStream +} + +func (x *treeManagerDownloadAttachmentClient) Recv() (*AttachmentResponse, error) { + m := new(AttachmentResponse) + if err := x.ClientStream.RecvMsg(m); err != nil { + return nil, err + } + return m, nil +} + // TreeManagerServer is the server API for TreeManager service. // All implementations must embed UnimplementedTreeManagerServer // for forward compatibility @@ -69,6 +102,7 @@ type TreeManagerServer interface { GetSummery(context.Context, *TreeRequest) (*Tree, error) GetFile(context.Context, *FileRequest) (*FileContent, error) RebuildTree(context.Context, *TreeSecret) (*RebuildResponse, error) + DownloadAttachment(*AttachmentRequest, TreeManager_DownloadAttachmentServer) error mustEmbedUnimplementedTreeManagerServer() } @@ -85,6 +119,9 @@ func (UnimplementedTreeManagerServer) GetFile(context.Context, *FileRequest) (*F func (UnimplementedTreeManagerServer) RebuildTree(context.Context, *TreeSecret) (*RebuildResponse, error) { return nil, status.Errorf(codes.Unimplemented, "method RebuildTree not implemented") } +func (UnimplementedTreeManagerServer) DownloadAttachment(*AttachmentRequest, TreeManager_DownloadAttachmentServer) error { + return status.Errorf(codes.Unimplemented, "method DownloadAttachment not implemented") +} func (UnimplementedTreeManagerServer) mustEmbedUnimplementedTreeManagerServer() {} // UnsafeTreeManagerServer may be embedded to opt out of forward compatibility for this service. @@ -152,6 +189,27 @@ func _TreeManager_RebuildTree_Handler(srv interface{}, ctx context.Context, dec return interceptor(ctx, in, info, handler) } +func _TreeManager_DownloadAttachment_Handler(srv interface{}, stream grpc.ServerStream) error { + m := new(AttachmentRequest) + if err := stream.RecvMsg(m); err != nil { + return err + } + return srv.(TreeManagerServer).DownloadAttachment(m, &treeManagerDownloadAttachmentServer{stream}) +} + +type TreeManager_DownloadAttachmentServer interface { + Send(*AttachmentResponse) error + grpc.ServerStream +} + +type treeManagerDownloadAttachmentServer struct { + grpc.ServerStream +} + +func (x *treeManagerDownloadAttachmentServer) Send(m *AttachmentResponse) error { + return x.ServerStream.SendMsg(m) +} + // TreeManager_ServiceDesc is the grpc.ServiceDesc for TreeManager service. // It's only intended for direct use with grpc.RegisterService, // and not to be introspected or modified (even as a copy) @@ -172,6 +230,12 @@ var TreeManager_ServiceDesc = grpc.ServiceDesc{ Handler: _TreeManager_RebuildTree_Handler, }, }, - Streams: []grpc.StreamDesc{}, + Streams: []grpc.StreamDesc{ + { + StreamName: "DownloadAttachment", + Handler: _TreeManager_DownloadAttachment_Handler, + ServerStreams: true, + }, + }, Metadata: "tree.proto", }