*: add file path

Author: gyuho

parse/markdown.go

@@ -41,7 +41,7 @@ func (p *Proto) Markdown(title string, parseOpts []ParseOption, lopts ...string)
 		switch opt {
 		case ParseService:
 			for _, svs := range p.Services {
-				buf.WriteString(fmt.Sprintf("##### service `%s`\n\n", svs.Name))
+				buf.WriteString(fmt.Sprintf("##### service `%s` (%s)\n\n", svs.Name, svs.FilePath))
 				if svs.Description != "" {
 					buf.WriteString(svs.Description)
 					buf.WriteString("\n\n")
@@ -65,7 +65,7 @@ func (p *Proto) Markdown(title string, parseOpts []ParseOption, lopts ...string)
 
 		case ParseMessage:
 			for _, msg := range p.Messages {
-				buf.WriteString(fmt.Sprintf("##### message `%s`\n\n", msg.Name))
+				buf.WriteString(fmt.Sprintf("##### message `%s` (%s)\n\n", msg.Name, msg.FilePath))
 				if msg.Description != "" {
 					buf.WriteString(msg.Description)
 					buf.WriteString("\n\n")

parse/parse.go

@@ -91,6 +91,16 @@ func ReadFile(fpath string) (*Proto, error) {
 	if err != nil {
 		return nil, err
 	}
+	var (
+		wd, _ = os.Getwd()
+		fs    = fpath
+	)
+	if strings.HasPrefix(fs, wd) {
+		fs = strings.Replace(fs, wd, "", 1)
+		if strings.HasPrefix(fs, "/") {
+			fs = strings.Replace(fs, "/", "", 1)
+		}
+	}
 
 	lines, err := readLines(f)
 	if err != nil {
@@ -161,6 +171,7 @@ func ReadFile(fpath string) (*Proto, error) {
 
 		case parsingMessage:
 			if strings.HasSuffix(line, "}") { // closing of message
+				protoMessage.FilePath = fs
 				rp.Messages = append(rp.Messages, protoMessage)
 				protoMessage = ProtoMessage{}
 				comments = []string{}
@@ -214,6 +225,7 @@ func ReadFile(fpath string) (*Proto, error) {
 				protoService.Methods = append(protoService.Methods, protoMethod)
 				comments = []string{}
 			} else if !strings.HasSuffix(line, "{}") && strings.HasSuffix(line, "}") { // closing of service
+				protoService.FilePath = fs
 				rp.Services = append(rp.Services, protoService)
 				protoService = ProtoService{}
 				comments = []string{}

parse/proto.go

@@ -42,6 +42,7 @@ func (p *Proto) Sort() {
 // ProtoService represents the 'service' type in Protocol Buffer.
 // (https://developers.google.com/protocol-buffers/docs/proto3#services)
 type ProtoService struct {
+	FilePath    string
 	Name        string
 	Description string
 	Methods     []ProtoMethod
@@ -58,6 +59,7 @@ type ProtoMethod struct {
 // ProtoMessage represents the 'message' type in Protocol Buffer.
 // (https://developers.google.com/protocol-buffers/docs/proto3#simple)
 type ProtoMessage struct {
+	FilePath    string
 	Name        string
 	Description string
 	Fields      []ProtoField