go build: substitute a.yandex-team.ru/ for GO_ARCADIA_PROJECT_PREFIX

3 files changed, 8 insertions, 7 deletions

diff --git a/build/scripts/go_tool.py b/build/scripts/go_tool.py
index 058ff7ad0dc..b99a3338661 100644
--- a/build/scripts/go_tool.py
+++ b/build/scripts/go_tool.py
@@ -669,7 +669,7 @@ def gen_test_main(args, test_lib_args, xtest_lib_args):
     if test_main_package is None:
         lines.append('    "os"')
     lines.extend(['    "testing"', '    "testing/internal/testdeps"'])
-    lines.extend(['    _ "a.yandex-team.ru/library/go/test/yatest"'])
+    lines.extend(['    _ "{}library/go/test/yatest"'.format(args.arc_project_prefix)])
 
     if len(tests) > 0:
         lines.append('    _test "{}"'.format(test_module_path))
diff --git a/build/scripts/postprocess_go_fbs.py b/build/scripts/postprocess_go_fbs.py
index e6d5184a5e1..ffc2f7cb8ff 100644
--- a/build/scripts/postprocess_go_fbs.py
+++ b/build/scripts/postprocess_go_fbs.py
@@ -15,13 +15,14 @@ IMPORT_DECL=re.compile(r'''
 
 def parse_args():
     parser = argparse.ArgumentParser()
+    parser.add_argument('--arcadia-prefix', required=True)
     parser.add_argument('--input-dir', required=True)
     parser.add_argument('--map', nargs='*', default=None)
 
     return parser.parse_args()
 
 
-def process_go_file(file_name, import_map):
+def process_go_file(file_name, import_map, arcadia_prefix):
     content = ''
     with open(file_name, 'r') as f:
         content = f.read()
@@ -41,7 +42,7 @@ def process_go_file(file_name, import_map):
         ns = namespace.split('.')
         name = '__'.join(ns)
         import_path = '/'.join(ns)
-        imports = imports.replace('{} "{}"'.format(name, import_path), '{} "a.yandex-team.ru/{}"'.format(name, path))
+        imports = imports.replace('{} "{}"'.format(name, import_path), '{} "{}{}"'.format(name, arcadia_prefix, path))
 
     if imports != content[start:end]:
         with open(file_name, 'w') as f:
@@ -65,7 +66,7 @@ def main():
 
     for root, _, files in os.walk(args.input_dir):
         for src in (f for f in files if f.endswith('.go')):
-            process_go_file(os.path.join(root, src), import_map)
+            process_go_file(os.path.join(root, src), import_map, args.arcadia_prefix)
 
 
 if __name__ == '__main__':
diff --git a/build/scripts/vcs_info.py b/build/scripts/vcs_info.py
index ea32d200b13..88ad7005796 100644
--- a/build/scripts/vcs_info.py
+++ b/build/scripts/vcs_info.py
@@ -235,7 +235,7 @@ def print_java(json_file, output_file, argv):
     merge_java_mf(json_file, output_file, input)
 
 
-def print_go(json_file, output_file):
+def print_go(json_file, output_file, arc_project_prefix):
     def gen_map(info):
         lines = []
         for k, v in info.items():
@@ -247,7 +247,7 @@ def print_go(json_file, output_file):
             '// Code generated by vcs_info.py; DO NOT EDIT.',
             '',
             'package main',
-            'import "a.yandex-team.ru/library/go/core/buildinfo"',
+            'import "{}library/go/core/buildinfo"'.format(arc_project_prefix),
             'func init() {',
              '   buildinfo.InitBuildInfo(map[string]string {'] + gen_map(json_file) + ['})',
              '}']
@@ -323,7 +323,7 @@ if __name__ == '__main__':
         json_file = get_json(json_name)
 
     if lang == 'Go':
-        print_go(json_file, sys.argv[2])
+        print_go(json_file, sys.argv[2], sys.argv[3])
     elif lang == 'Java':
         print_java(json_file, sys.argv[2], sys.argv[3:])
     elif lang == 'JSON':