feat: add banner image and finalize v0.1.0

fezcode · fezcode · commit 49e25a5d9ae5 · 2026-03-11T23:58:06.000+03:00
diff --git a/banner-image.png b/banner-image.png
diff --git a/internal/db/db.go b/internal/db/db.go
@@ -59,6 +59,33 @@ func Connect(connStr string) (Database, error) {
 }
 
 func (s *sqlDB) Query(query string) (*Result, error) {
+	upperQuery := strings.TrimSpace(strings.ToUpper(query))
+	isSelect := strings.HasPrefix(upperQuery, "SELECT") || 
+				strings.HasPrefix(upperQuery, "PRAGMA") || 
+				strings.HasPrefix(upperQuery, "SHOW") || 
+				strings.HasPrefix(upperQuery, "EXPLAIN") ||
+				strings.HasPrefix(upperQuery, "WITH")
+
+	if !isSelect {
+		res, err := s.db.Exec(query)
+		if err != nil {
+			return nil, err
+		}
+
+		rowsAffected, _ := res.RowsAffected()
+		lastInsertID, _ := res.LastInsertId()
+
+		msg := fmt.Sprintf("Success. Rows affected: %d", rowsAffected)
+		if lastInsertID > 0 {
+			msg += fmt.Sprintf(", Last Insert ID: %d", lastInsertID)
+		}
+
+		return &Result{
+			Columns: []string{"Status"},
+			Rows:    [][]string{{msg}},
+		}, nil
+	}
+
 	rows, err := s.db.Query(query)
 	if err != nil {
 		return nil, err
diff --git a/internal/ui/ui.go b/internal/ui/ui.go
@@ -125,14 +125,15 @@ func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 			}
 
 		case "h":
-			if m.focus == focusInput {
+			if m.focus == focusInput && !m.showHelp {
 				m.input, cmd = m.input.Update(msg)
 				return m, cmd
 			}
 			m.showHelp = !m.showHelp
 			if m.showHelp {
 				m.showDetail = false
 			}
+			return m, nil
 
 		case "c":
 			if m.focus == focusTable && m.result != nil {
@@ -196,6 +197,10 @@ func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 			}
 
 		case "esc":
+			if m.showHelp {
+				m.showHelp = false
+				return m, nil
+			}
 			if m.showDetail {
 				m.showDetail = false
 				return m, nil
@@ -444,6 +449,13 @@ func (m Model) helpView() string {
 		tutorial.Render("  SELECT * FROM users WHERE age > 18;"),
 		tutorial.Render("  -- Join tables"),
 		tutorial.Render("  SELECT u.name, p.title FROM users u JOIN posts p ON u.id = p.user_id;"),
+		"",
+		tutorial.Render("  -- Create / Insert / Update / Delete"),
+		tutorial.Render("  CREATE TABLE items (id INTEGER PRIMARY KEY, name TEXT);"),
+		tutorial.Render("  INSERT INTO items (name) VALUES ('Laptop'), ('Mouse');"),
+		tutorial.Render("  UPDATE items SET name = 'Mechanical Mouse' WHERE id = 2;"),
+		tutorial.Render("  DELETE FROM items WHERE id = 1;"),
+		"",
 		tutorial.Render("  -- SQLite Meta: List Tables / Schema"),
 		tutorial.Render("  SELECT name FROM sqlite_master WHERE type='table';"),
 		tutorial.Render("  PRAGMA table_info(table_name); -- Describe table"),
