diff --git a/src/protobuffs_compile.erl b/src/protobuffs_compile.erl index aed0b0ce..34181681 100644 --- a/src/protobuffs_compile.erl +++ b/src/protobuffs_compile.erl @@ -45,7 +45,7 @@ scan_file(ProtoFile) -> scan_file(ProtoFile,Options) when is_list(ProtoFile) -> Basename = filename:basename(ProtoFile, ".proto") ++ "_pb", {ok,FirstParsed} = parse(ProtoFile), - ImportPaths = ["./" | proplists:get_value(imports_dir, Options, [])], + ImportPaths = ["./", "src/" | proplists:get_value(imports_dir, Options, [])], Parsed = parse_imports(FirstParsed, ImportPaths), {{msg,UntypedMessages},{enum,Enums}} = collect_full_messages(Parsed), Messages = resolve_types(UntypedMessages,Enums), @@ -71,7 +71,7 @@ generate_source(ProtoFile) -> generate_source(ProtoFile,Options) when is_list (ProtoFile) -> Basename = filename:basename(ProtoFile, ".proto") ++ "_pb", {ok,FirstParsed} = parse(ProtoFile), - ImportPaths = ["./" | proplists:get_value(imports_dir, Options, [])], + ImportPaths = ["./", "src/" | proplists:get_value(imports_dir, Options, [])], Parsed = parse_imports(FirstParsed, ImportPaths), {{msg,UntypedMessages},{enum,Enums}} = collect_full_messages(Parsed), Messages = resolve_types(UntypedMessages,Enums),