Adding "src/" to paths list.

trifork · Jan 6, 2011 · 63916aa · 63916aa
1 parent 6da70c8
commit 63916aa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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),