1 files changed, 38 insertions, 0 deletions
diff --git a/internal/profiling/flags.go b/internal/profiling/flags.go
new file mode 100644
index 0000000..59a6d78
--- /dev/null
+++ b/internal/profiling/flags.go
@@ -0,0 +1,38 @@
+package profiling
+
+import "flag"
+
+// Flags holds command-line flags for profiling
+type Flags struct {
+	// Enable CPU profiling
+	CPUProfile bool
+	// Enable memory profiling
+	MemProfile bool
+	// Enable all profiling (CPU + memory)
+	Profile bool
+	// Directory to store profiles
+	ProfileDir string
+}
+
+// AddFlags adds profiling flags to the flag set
+func AddFlags(f *Flags) {
+	flag.BoolVar(&f.CPUProfile, "cpuprofile", false, "Enable CPU profiling")
+	flag.BoolVar(&f.MemProfile, "memprofile", false, "Enable memory profiling")
+	flag.BoolVar(&f.Profile, "profile", false, "Enable all profiling (CPU + memory)")
+	flag.StringVar(&f.ProfileDir, "profiledir", "profiles", "Directory to store profiles")
+}
+
+// ToConfig converts flags to profiler config
+func (f *Flags) ToConfig(commandName string) Config {
+	return Config{
+		CPUProfile:  f.CPUProfile || f.Profile,
+		MemProfile:  f.MemProfile || f.Profile,
+		ProfileDir:  f.ProfileDir,
+		CommandName: commandName,
+	}
+}
+
+// Enabled returns true if any profiling is enabled
+func (f *Flags) Enabled() bool {
+	return f.CPUProfile || f.MemProfile || f.Profile
+}
+\ No newline at end of file