address feedback

Signed-off-by: Vinayak Kukreja <vinakuk@amazon.com>

packages/codemaker/src/codemaker.ts

@@ -196,6 +196,14 @@ export class CodeMaker {
     return caseutils.toSnakeCase(s, sep);
   }
 
+  /**
+   * Gets currently opened file path
+   * @returns Currently opened file path.
+   */
+  public getCurrentFilePath(): string | undefined {
+    return this.currentFile?.filePath;
+  }
+
   private makeIndent(): string {
     const length = this.currentIndentLength;
     if (length <= 0) {

packages/jsii-pacmak/lib/targets/go/package.ts

@@ -37,6 +37,7 @@ import { VersionFile } from './version-file';
 
 export const GOMOD_FILENAME = 'go.mod';
 export const GO_VERSION = '1.18';
+const MAIN_FILE = 'main.go';
 
 /*
  * Package represents a single `.go` source file within a package. This can be the root package file or a submodule
@@ -203,9 +204,9 @@ export abstract class Package {
     if (this.types.length > 0) {
       const { code } = context;
 
-      const initFile = join(this.directory, `main.go`);
+      const initFile = join(this.directory, MAIN_FILE);
       code.openFile(initFile);
-      code.line(`package ${this.packageName}`);
+      this.emitHeader(code);
       code.line();
       importGoModules(code, [GO_REFLECT, JSII_RT_MODULE]);
       code.line();
@@ -469,6 +470,19 @@ export class RootPackage extends Package {
     );
   }
 
+  protected emitHeader(code: CodeMaker) {
+    const currentFilePath = code.getCurrentFilePath();
+    if (
+      this.assembly.description !== '' &&
+      currentFilePath !== undefined &&
+      currentFilePath.includes(MAIN_FILE)
+    ) {
+      code.line(`// ${this.assembly.description}`);
+    }
+    code.line(`package ${this.packageName}`);
+    code.line();
+  }
+
   private emitJsiiPackage({ code }: EmitContext) {
     const dependencies = this.packageDependencies.sort((l, r) =>
       l.moduleName.localeCompare(r.moduleName),