From 95f93b864187126a3f85f27486dc86b80daa8f45 Mon Sep 17 00:00:00 2001
From: Sauyon Lee <sauyon@github.com>
Date: Wed, 10 Mar 2021 07:23:06 -0800
Subject: [PATCH] Add FileFor utility function for trap files

---
 extractor/trap/trapwriter.go | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/extractor/trap/trapwriter.go b/extractor/trap/trapwriter.go
index ebbe8f44ed1..ff96defa3bd 100644
--- a/extractor/trap/trapwriter.go
+++ b/extractor/trap/trapwriter.go
@@ -26,14 +26,22 @@ type Writer struct {
 	Package      *packages.Package
 }
 
+func FileFor(path string) (string, error) {
+	trapFolder, err := trapFolder()
+	if err != nil {
+		return "", err
+	}
+
+	return filepath.Join(trapFolder, srcarchive.AppendablePath(path)+".trap.gz"), nil
+}
+
 // NewWriter creates a TRAP file for the given path and returns a writer for
 // writing to it
 func NewWriter(path string, pkg *packages.Package) (*Writer, error) {
-	trapFolder, err := trapFolder()
+	trapFilePath, err := FileFor(path)
 	if err != nil {
 		return nil, err
 	}
-	trapFilePath := filepath.Join(trapFolder, srcarchive.AppendablePath(path)+".trap.gz")
 	trapFileDir := filepath.Dir(trapFilePath)
 	err = os.MkdirAll(trapFileDir, 0755)
 	if err != nil {