From 58a85e82abd3609a76a5444eaf67cd92a8aabfd1 Mon Sep 17 00:00:00 2001
From: Masahiro Yamada <masahiro.yamada@canonical.com>
Date: Tue, 6 Dec 2022 09:43:50 +0900
Subject: [PATCH] annotations: add missing CONFIG_ prefix in main()

This is common for do_query(), do_note(), do_write() (and whatever
command might be added in the future).

Move it to main().

Signed-off-by: Masahiro Yamada <masahiro.yamada@canonical.com>
Signed-off-by: Andrea Righi <andrea.righi@canonical.com>
---
 annotations | 39 ++++++++++++++-------------------------
 1 file changed, 14 insertions(+), 25 deletions(-)

diff --git a/annotations b/annotations
index 877ee22..9ae320c 100755
--- a/annotations
+++ b/annotations
@@ -76,11 +76,8 @@ def do_query(args):
     if args.arch is None and args.flavour is not None:
         arg_fail('error: --flavour requires --arch')
     a = Annotation(args.file)
-    for config in (args.config, 'CONFIG_' + args.config if args.config else None):
-        res = a.search_config(config=config, arch=args.arch, flavour=args.flavour)
-        if res:
-            break
-    print_result(config, res)
+    res = a.search_config(config=args.config, arch=args.arch, flavour=args.flavour)
+    print_result(args.config, res)
 
 def do_autocomplete(args):
     a = Annotation(args.file)
@@ -90,57 +87,45 @@ def do_autocomplete(args):
 def do_source(args):
     if args.config is None:
         arg_fail('error: --source requires --config')
-    if args.config.startswith('CONFIG_'):
-        config = args.config
-    else:
-        config = 'CONFIG_' + args.config
     if not os.path.exists('tags'):
         print('tags not found in the current directory, try: `make tags`')
         exit(1)
-    os.system(f'vim -t {config}')
+    os.system(f'vim -t {args.config}')
 
 def do_note(args):
     if args.config is None:
         arg_fail('error: --note requires --config')
-    if args.config.startswith('CONFIG_'):
-        config = args.config
-    else:
-        config = 'CONFIG_' + args.config
 
     # Set the note in annotations
     a = Annotation(args.file)
-    a.set(config, note=args.note)
+    a.set(args.config, note=args.note)
 
     # Save back to annotations
     a.save(args.file)
 
     # Query and print back the value
     a = Annotation(args.file)
-    res = a.search_config(config=config)
-    print_result(config, res)
+    res = a.search_config(config=args.config)
+    print_result(args.config, res)
 
 def do_write(args):
     if args.config is None:
         arg_fail('error: --write requires --config')
-    if args.config.startswith('CONFIG_'):
-        config = args.config
-    else:
-        config = 'CONFIG_' + args.config
 
     # Set the value in annotations ('null' means remove)
     a = Annotation(args.file)
     if args.value == 'null':
-        a.remove(config, arch=args.arch, flavour=args.flavour)
+        a.remove(args.config, arch=args.arch, flavour=args.flavour)
     else:
-        a.set(config, arch=args.arch, flavour=args.flavour, value=args.value, note=args.note)
+        a.set(args.config, arch=args.arch, flavour=args.flavour, value=args.value, note=args.note)
 
     # Save back to annotations
     a.save(args.file)
 
     # Query and print back the value
     a = Annotation(args.file)
-    res = a.search_config(config=config)
-    print_result(config, res)
+    res = a.search_config(config=args.config)
+    print_result(args.config, res)
 
 def do_export(args):
     if args.arch is None:
@@ -236,6 +221,10 @@ def main():
     # Main annotations program
     args = _ARGPARSER.parse_args()
     autodetect_annotations(args)
+
+    if args.config and not args.config.startswith('CONFIG_'):
+        args.config = 'CONFIG_' + args.config
+
     if args.value:
         do_write(args)
     elif args.note:
-- 
2.31.1