fortrangoingonforty/fortress / 2367faf

+program fortress

+    use iso_fortran_env, only: output_unit

+    use terminal_control

+    use filesystem_ops

+    use git_ops

+    use ui_display

+    ! State variables

+    character(len=MAX_PATH) :: current_dir, parent_dir, temp_dir, exit_dir

+    integer :: selected = 1, parent_selected = -1

+    integer :: scroll_offset = 0, parent_scroll_offset = 0

+    character(len=256) :: repo_name, branch_name

+    logical :: in_git_repo = .false., running = .true., cd_on_exit = .false.

+    call setup_raw_mode()

+        ! Get terminal size

+        visible_height = rows - 3

+        ! Handle navigation signals from previous iteration

+        ! Bounds check

+        ! Adjust scroll to keep cursor visible

+        if (selected < scroll_offset + 1) scroll_offset = max(0, selected - 1)

+        if (selected > scroll_offset + visible_height) scroll_offset = selected - visible_height

+            if (parent_selected < parent_scroll_offset + 1) parent_scroll_offset = max(0, parent_selected - 1)

+            if (parent_selected > parent_scroll_offset + visible_height) parent_scroll_offset = parent_selected - visible_height

+        ! Draw

+        call draw_interface(rows, cols, current_dir, current_files, current_is_dir, current_is_exec, &

+                           current_is_staged, current_is_unstaged, current_is_untracked, current_count, &

+                           parent_files, parent_is_dir, parent_is_exec, parent_count, &

+                           selected, parent_selected, scroll_offset, parent_scroll_offset, &

+                           in_git_repo, repo_name, branch_name)

+        case(27)  ! ESC - arrow keys

+                if (selected > 1) selected = selected - 1

+                if (selected < current_count .and. current_count > 0) selected = selected + 1

+            case('C')  ! Right - enter directory

+                        selected = -1

+            case('D')  ! Left - go back

+                    selected = -1

+        case(113, 81)  ! 'q' or 'Q' - quit

+                selected = -2

+        case(85, 117)  ! 'U' or 'u' - git unstage

+        case(77, 109)  ! 'M' or 'm' - git commit

+            if (in_git_repo) then

+                call git_commit_prompt(current_dir, repo_name)

+            end if

+    call restore_terminal()

+end program fortress

+module filesystem_ops

+    implicit none

+    private

+

+    public :: get_file_list, get_pwd, get_parent_path, join_path

+    public :: find_in_parent, find_file_in_list, fzf_search, write_exit_dir

+    public :: MAX_PATH, MAX_FILES

+

+    integer, parameter :: MAX_PATH = 512

+    integer, parameter :: MAX_FILES = 500

+

+contains

+

+    subroutine get_file_list(dir, files, is_dir, is_exec, count)

+        character(len=*), intent(in) :: dir

+        character(len=*), dimension(*), intent(out) :: files

+        logical, dimension(*), intent(out) :: is_dir, is_exec

+        integer, intent(out) :: count

+        integer :: unit, ios, stat

+        character(len=MAX_PATH) :: fullpath

+

+        call execute_command_line("ls -1a '" // trim(dir) // "' > .fortress_ls 2>/dev/null", wait=.true.)

+

+        open(newunit=unit, file=".fortress_ls", status='old', iostat=ios)

+        if (ios /= 0) then

+            count = 0

+            return

+        end if

+

+        count = 0

+        do

+            count = count + 1

+            if (count > MAX_FILES) exit

+            read(unit, '(a)', iostat=ios) files(count)

+            if (ios /= 0) then

+                count = count - 1

+                exit

+            end if

+

+            fullpath = join_path(dir, files(count))

+            call execute_command_line("test -d '" // trim(fullpath) // "'", exitstat=stat, wait=.true.)

+            is_dir(count) = (stat == 0)

+

+            ! Check if executable (but not directories)

+            if (.not. is_dir(count)) then

+                call execute_command_line("test -x '" // trim(fullpath) // "'", exitstat=stat, wait=.true.)

+                is_exec(count) = (stat == 0)

+            else

+                is_exec(count) = .false.

+            end if

+        end do

+

+        close(unit)

+        call execute_command_line("rm -f .fortress_ls 2>/dev/null")

+    end subroutine get_file_list

+

+    function get_pwd() result(path)

+        character(len=MAX_PATH) :: path

+        integer :: unit, ios

+

+        call execute_command_line("pwd > .fortress_pwd 2>/dev/null", wait=.true.)

+        open(newunit=unit, file=".fortress_pwd", status='old', iostat=ios)

+        if (ios == 0) then

+            read(unit, '(a)') path

+            close(unit)

+        else

+            path = "."

+        end if

+        call execute_command_line("rm -f .fortress_pwd 2>/dev/null")

+    end function get_pwd

+

+    function get_parent_path(path) result(parent)

+        character(len=*), intent(in) :: path

+        character(len=MAX_PATH) :: parent

+        integer :: pos

+

+        pos = index(path, "/", back=.true.)

+        if (pos > 1) then

+            parent = path(1:pos-1)

+        else if (pos == 1) then

+            parent = "/"

+        else

+            parent = "."

+        end if

+    end function get_parent_path

+

+    function join_path(base, name) result(full)

+        character(len=*), intent(in) :: base, name

+        character(len=MAX_PATH) :: full

+

+        if (base == "/") then

+            full = "/" // trim(name)

+        else

+            full = trim(base) // "/" // trim(name)

+        end if

+    end function join_path

+

+    function find_in_parent(dir, files, count) result(idx)

+        character(len=*), intent(in) :: dir

+        character(len=*), dimension(*), intent(in) :: files

+        integer, intent(in) :: count

+        integer :: idx, pos

+        character(len=256) :: basename

+

+        pos = index(dir, "/", back=.true.)

+        if (pos > 0) then

+            basename = dir(pos+1:)

+        else

+            basename = dir

+        end if

+

+        do idx = 1, count

+            if (trim(files(idx)) == trim(basename)) return

+        end do

+        idx = 1

+    end function find_in_parent

+

+    function find_file_in_list(target_path, files, count) result(idx)

+        character(len=*), intent(in) :: target_path

+        character(len=*), dimension(*), intent(in) :: files

+        integer, intent(in) :: count

+        integer :: idx, pos

+        character(len=MAX_PATH) :: basename

+

+        ! Extract basename from target_path

+        pos = index(target_path, "/", back=.true.)

+        if (pos > 0) then

+            basename = target_path(pos+1:)

+        else

+            basename = target_path

+        end if

+

+        ! Search for the file in the list

+        do idx = 1, count

+            if (trim(files(idx)) == trim(basename)) return

+        end do

+

+        ! Default to first item if not found

+        idx = 1

+    end function find_file_in_list

+

+    subroutine fzf_search(search_dir, result_path)

+        character(len=*), intent(in) :: search_dir

+        character(len=*), intent(out) :: result_path

+        character(len=MAX_PATH) :: temp_file, fzf_cmd

+        integer :: unit, ios, stat

+

+        result_path = ""

+

+        ! Create temp file for fzf output

+        call get_environment_variable("HOME", temp_file)

+        temp_file = trim(temp_file) // "/.fortress_fzf"

+

+        ! Restore terminal for fzf

+        call execute_command_line("stty icanon echo 2>/dev/null")

+

+        ! Build fzf command: find files, pipe to fzf, save selection

+        fzf_cmd = "cd '" // trim(search_dir) // "' && " // &

+                  "find . -type f -o -type d | " // &

+                  "sed 's|^\./||' | " // &

+                  "fzf --height=40% --reverse --border --preview 'ls -lh {}' " // &

+                  "> " // trim(temp_file) // " 2>/dev/null"

+

+        ! Run fzf

+        call execute_command_line(trim(fzf_cmd), exitstat=stat, wait=.true.)

+

+        ! Restore raw mode

+        call execute_command_line("stty -icanon -echo min 1 time 0 2>/dev/null")

+

+        ! Read result if fzf succeeded

+        if (stat == 0) then

+            open(newunit=unit, file=temp_file, status='old', iostat=ios)

+            if (ios == 0) then

+                read(unit, '(a)', iostat=ios) result_path

+                if (ios == 0) then

+                    ! Convert relative path to absolute

+                    result_path = join_path(search_dir, result_path)

+                end if

+                close(unit)

+            end if

+        end if

+

+        ! Cleanup

+        call execute_command_line("rm -f " // trim(temp_file) // " 2>/dev/null")

+    end subroutine fzf_search

+

+    subroutine write_exit_dir(dir)

+        character(len=*), intent(in) :: dir

+        character(len=MAX_PATH) :: temp_file

+        integer :: unit, ios

+

+        ! Create temp file in HOME directory

+        call get_environment_variable("HOME", temp_file)

+        temp_file = trim(temp_file) // "/.fortress_cd"

+

+        open(newunit=unit, file=temp_file, status='replace', action='write', iostat=ios)

+        if (ios == 0) then

+            write(unit, '(a)') trim(dir)

+            close(unit)

+        end if

+    end subroutine write_exit_dir

+

+end module filesystem_ops

+module git_ops

+    use iso_fortran_env, only: output_unit

+    use terminal_control, only: GREEN, RED, GREY, RESET, BOLD, CLEAR

+    use filesystem_ops, only: MAX_PATH

+    implicit none

+    private

+

+    public :: detect_git_repo, get_git_status, write_git_indicators

+    public :: git_add_file, git_unstage_file, git_commit_prompt

+

+contains

+

+    subroutine detect_git_repo(dir, is_git, repo, branch)

+        character(len=*), intent(in) :: dir

+        logical, intent(out) :: is_git

+        character(len=*), intent(out) :: repo, branch

+        integer :: stat, i

+        character(len=MAX_PATH) :: temp_file

+

+        is_git = .false.

+        repo = ""

+        branch = ""

+

+        ! Check if .git directory exists

+        call execute_command_line("git -C '" // trim(dir) // "' rev-parse --git-dir > /dev/null 2>&1", &

+                                  exitstat=stat, wait=.true.)

+        is_git = (stat == 0)

+

+        if (is_git) then

+            ! Get repo name (basename of repo root)

+            call get_environment_variable("HOME", temp_file)

+            temp_file = trim(temp_file) // "/.fortress_repo"

+            call execute_command_line("git -C '" // trim(dir) // "' rev-parse --show-toplevel 2>/dev/null | " // &

+                                     "xargs basename > " // trim(temp_file), wait=.true.)

+            open(newunit=stat, file=temp_file, status='old', iostat=i)

+            if (i == 0) then

+                read(stat, '(a)', iostat=i) repo

+                close(stat)

+            end if

+            call execute_command_line("rm -f " // trim(temp_file) // " 2>/dev/null")

+

+            ! Get current branch name

+            temp_file = trim(temp_file) // "_branch"

+            call execute_command_line("git -C '" // trim(dir) // "' rev-parse --abbrev-ref HEAD 2>/dev/null > " // &

+                                     trim(temp_file), wait=.true.)

+            open(newunit=stat, file=temp_file, status='old', iostat=i)

+            if (i == 0) then

+                read(stat, '(a)', iostat=i) branch

+                close(stat)

+            end if

+            call execute_command_line("rm -f " // trim(temp_file) // " 2>/dev/null")

+        end if

+    end subroutine detect_git_repo

+

+    subroutine get_git_status(dir, files, count, is_staged, is_unstaged, is_untracked)

+        character(len=*), intent(in) :: dir

+        character(len=*), dimension(*), intent(in) :: files

+        integer, intent(in) :: count

+        logical, dimension(*), intent(out) :: is_staged, is_unstaged, is_untracked

+        character(len=MAX_PATH) :: temp_file, line, file_path, git_status

+        integer :: unit, ios, stat, i

+

+        ! Initialize all to false

+        do i = 1, count

+            is_staged(i) = .false.

+            is_unstaged(i) = .false.

+            is_untracked(i) = .false.

+        end do

+

+        ! Get git status

+        call get_environment_variable("HOME", temp_file)

+        temp_file = trim(temp_file) // "/.fortress_git_status"

+        call execute_command_line("cd '" // trim(dir) // "' && git status --porcelain 2>/dev/null > " // &

+                                 trim(temp_file), exitstat=stat, wait=.true.)

+

+        if (stat /= 0) return

+

+        ! Parse git status output

+        open(newunit=unit, file=temp_file, status='old', iostat=ios)

+        if (ios /= 0) return

+

+        do

+            read(unit, '(a)', iostat=ios) line

+            if (ios /= 0) exit

+

+            if (len_trim(line) > 3) then

+                git_status = line(1:2)

+                file_path = trim(adjustl(line(4:)))

+

+                ! Match against our file list

+                do i = 1, count

+                    if (trim(files(i)) == trim(file_path)) then

+                        ! Parse git status (XY format)

+                        is_untracked(i) = (git_status == '??')

+                        is_staged(i) = (git_status(1:1) /= ' ' .and. git_status(1:1) /= '?')

+                        is_unstaged(i) = (git_status(2:2) /= ' ' .and. .not. is_untracked(i))

+                        exit

+                    end if

+                end do

+            end if

+        end do

+

+        close(unit)

+        call execute_command_line("rm -f " // trim(temp_file) // " 2>/dev/null")

+    end subroutine get_git_status

+

+    subroutine write_git_indicators(staged, unstaged, untracked, highlighted)

+        logical, intent(in) :: staged, unstaged, untracked, highlighted

+

+        ! Write indicators without RESET (caller handles that)

+        if (staged) then

+            if (highlighted) then

+                write(output_unit, '(a)', advance='no') GREEN // " ↑"

+            else

+                write(output_unit, '(a)', advance='no') GREEN // " ↑" // RESET

+            end if

+        end if

+        if (unstaged) then

+            if (highlighted) then

+                write(output_unit, '(a)', advance='no') RED // " ✗"

+            else

+                write(output_unit, '(a)', advance='no') RED // " ✗" // RESET

+            end if

+        end if

+        if (untracked) then

+            if (highlighted) then

+                write(output_unit, '(a)', advance='no') GREY // " ✗"

+            else

+                write(output_unit, '(a)', advance='no') GREY // " ✗" // RESET

+            end if

+        end if

+    end subroutine write_git_indicators

+

+    subroutine git_add_file(dir, filename)

+        character(len=*), intent(in) :: dir, filename

+        character(len=MAX_PATH*2) :: git_cmd

+        integer :: stat

+

+        ! Build git add command

+        git_cmd = "cd '" // trim(dir) // "' && git add '" // trim(filename) // "' 2>/dev/null"

+        call execute_command_line(trim(git_cmd), exitstat=stat, wait=.true.)

+    end subroutine git_add_file

+

+    subroutine git_unstage_file(dir, filename)

+        character(len=*), intent(in) :: dir, filename

+        character(len=MAX_PATH*2) :: git_cmd

+        integer :: stat

+

+        ! Build git restore --staged command

+        git_cmd = "cd '" // trim(dir) // "' && git restore --staged '" // trim(filename) // "' 2>/dev/null"

+        call execute_command_line(trim(git_cmd), exitstat=stat, wait=.true.)

+    end subroutine git_unstage_file

+

+    subroutine git_commit_prompt(dir, repo_name)

+        character(len=*), intent(in) :: dir, repo_name

+        character(len=512) :: commit_msg

+        character(len=MAX_PATH*2) :: git_cmd

+        character(len=1) :: key

+        integer :: stat, ios

+

+        ! Clear screen and show prompt

+        write(output_unit, '(a)', advance='no') CLEAR

+        write(output_unit, '(a)', advance='no') BOLD // "Git Commit" // RESET // " - " // trim(repo_name)

+        write(output_unit, *)

+        write(output_unit, *)

+        write(output_unit, '(a)', advance='no') "Commit message: "

+

+        ! Restore terminal to canonical mode for reading input

+        call execute_command_line("stty icanon echo 2>/dev/null")

+

+        ! Read commit message

+        read(*, '(a)', iostat=ios) commit_msg

+

+        ! Restore raw mode

+        call execute_command_line("stty -icanon -echo min 1 time 0 2>/dev/null")

+

+        if (ios == 0 .and. len_trim(commit_msg) > 0) then

+            ! Execute git commit (use single quotes for message to avoid escaping issues)

+            git_cmd = "cd '" // trim(dir) // "' && git commit -m '" // trim(commit_msg) // "' 2>&1"

+            call execute_command_line(trim(git_cmd), exitstat=stat, wait=.true.)

+

+            ! Show result briefly

+            write(output_unit, *)

+            if (stat == 0) then

+                write(output_unit, '(a)') GREEN // "✓ Committed successfully!" // RESET

+            else

+                write(output_unit, '(a)') RED // "✗ Commit failed (nothing to commit?)" // RESET

+            end if

+            write(output_unit, '(a)') "Press any key to continue..."

+

+            ! Wait for keypress

+            read(*, '(a1)', advance='no') key

+        end if

+    end subroutine git_commit_prompt

+

+end module git_ops

+module terminal_control

+    use iso_fortran_env, only: output_unit

+    implicit none

+    private

+

+    public :: get_term_size, setup_raw_mode, restore_terminal, read_arrow_key

+    public :: ESC, CLEAR, BOLD, DIM, REVERSE, RESET

+    public :: BLUE, GREEN, RED, GREY, WHITE

+

+    ! ANSI escape codes

+    character(len=*), parameter :: ESC = char(27)

+    character(len=*), parameter :: CLEAR = ESC // "[2J" // ESC // "[H"

+    character(len=*), parameter :: BOLD = ESC // "[1m"

+    character(len=*), parameter :: DIM = ESC // "[2m"

+    character(len=*), parameter :: REVERSE = ESC // "[7m"

+    character(len=*), parameter :: RESET = ESC // "[0m"

+    character(len=*), parameter :: BLUE = ESC // "[34m"

+    character(len=*), parameter :: GREEN = ESC // "[32m"

+    character(len=*), parameter :: RED = ESC // "[31m"

+    character(len=*), parameter :: GREY = ESC // "[90m"

+    character(len=*), parameter :: WHITE = ESC // "[37m"

+

+contains

+

+    subroutine get_term_size(r, c)

+        integer, intent(out) :: r, c

+        integer :: unit, ios

+

+        call execute_command_line("tput lines > .fortress_size 2>/dev/null", wait=.true.)

+        open(newunit=unit, file=".fortress_size", status='old', iostat=ios)

+        if (ios == 0) then

+            read(unit, *) r

+            close(unit)

+        else

+            r = 24

+        end if

+

+        call execute_command_line("tput cols > .fortress_size 2>/dev/null", wait=.true.)

+        open(newunit=unit, file=".fortress_size", status='old', iostat=ios)

+        if (ios == 0) then

+            read(unit, *) c

+            close(unit)

+        else

+            c = 80

+        end if

+

+        call execute_command_line("rm -f .fortress_size 2>/dev/null")

+    end subroutine get_term_size

+

+    subroutine setup_raw_mode()

+        call execute_command_line("stty -icanon -echo min 1 time 0 2>/dev/null")

+    end subroutine setup_raw_mode

+

+    subroutine restore_terminal()

+        call execute_command_line("stty icanon echo 2>/dev/null")

+    end subroutine restore_terminal

+

+    subroutine read_arrow_key(k)

+        character(len=1), intent(out) :: k

+        character(len=1) :: ch

+

+        read(*, '(a1)', advance='no') ch

+        if (ch == '[') then

+            read(*, '(a1)', advance='no') k

+        else

+            k = ch

+        end if

+    end subroutine read_arrow_key

+

+end module terminal_control

+module ui_display

+    use iso_fortran_env, only: output_unit

+    use terminal_control

+    use git_ops, only: write_git_indicators

+    use filesystem_ops, only: MAX_PATH, MAX_FILES

+    implicit none

+    private

+

+    public :: draw_interface, get_file_color

+

+contains

+

+    subroutine draw_interface(r, c, current_dir, current_files, current_is_dir, current_is_exec, &

+                              current_is_staged, current_is_unstaged, current_is_untracked, current_count, &

+                              parent_files, parent_is_dir, parent_is_exec, parent_count, &

+                              selected, parent_selected, scroll_offset, parent_scroll_offset, &

+                              in_git_repo, repo_name, branch_name)

+        integer, intent(in) :: r, c, current_count, parent_count, selected, parent_selected

+        integer, intent(in) :: scroll_offset, parent_scroll_offset

+        character(len=*), intent(in) :: current_dir, repo_name, branch_name

+        character(len=*), dimension(*), intent(in) :: current_files, parent_files

+        logical, dimension(*), intent(in) :: current_is_dir, parent_is_dir

+        logical, dimension(*), intent(in) :: current_is_exec, parent_is_exec

+        logical, dimension(*), intent(in) :: current_is_staged, current_is_unstaged, current_is_untracked

+        logical, intent(in) :: in_git_repo

+        integer :: left_w, i, parent_idx, current_idx, vis_h

+        character(len=256) :: fname

+        character(len=20) :: color_code

+

+        left_w = c * 3 / 10

+        vis_h = r - 3  ! Visible height

+

+        ! Header

+        write(output_unit, '(a)') BOLD // "FORTRESS" // RESET // " - " // trim(current_dir)

+

+        ! Files (render based on scroll offsets)

+        do i = 1, vis_h

+            parent_idx = i + parent_scroll_offset

+            current_idx = i + scroll_offset

+

+            ! Parent pane

+            if (parent_idx >= 1 .and. parent_idx <= parent_count) then

+                fname = parent_files(parent_idx)

+                if (parent_is_dir(parent_idx) .and. fname /= "." .and. fname /= "..") then

+                    fname = trim(fname) // "/"

+                end if

+

+                ! Get color for parent file

+                color_code = get_file_color(parent_files(parent_idx), parent_is_dir(parent_idx), parent_is_exec(parent_idx))

+

+                if (parent_idx == parent_selected) then

+                    write(output_unit, '(a)', advance='no') DIM // BOLD // trim(color_code) // &

+                        fname(1:min(len_trim(fname),left_w)) // RESET

+                else

+                    write(output_unit, '(a)', advance='no') DIM // trim(color_code) // &

+                        fname(1:min(len_trim(fname),left_w)) // RESET

+                end if

+                write(output_unit, '(a)', advance='no') repeat(" ", max(0, left_w - len_trim(fname)))

+            else

+                write(output_unit, '(a)', advance='no') repeat(" ", left_w)

+            end if

+

+            ! Separator

+            write(output_unit, '(a)', advance='no') " │ "

+

+            ! Current pane

+            if (current_idx >= 1 .and. current_idx <= current_count) then

+                fname = current_files(current_idx)

+                if (current_is_dir(current_idx) .and. fname /= "." .and. fname /= "..") then

+                    fname = trim(fname) // "/"

+                end if

+

+                ! Get color for current file

+                color_code = get_file_color(current_files(current_idx), current_is_dir(current_idx), current_is_exec(current_idx))

+

+                if (current_idx == selected) then

+                    write(output_unit, '(a)', advance='no') REVERSE // trim(color_code) // trim(fname)

+                    ! Add git indicators if in repo

+                    if (in_git_repo) then

+                        call write_git_indicators(current_is_staged(current_idx), &

+                                                  current_is_unstaged(current_idx), &

+                                                  current_is_untracked(current_idx), .true.)

+                    end if

+                    write(output_unit, '(a)') RESET

+                else

+                    write(output_unit, '(a)', advance='no') trim(color_code) // trim(fname)

+                    ! Add git indicators if in repo

+                    if (in_git_repo) then

+                        call write_git_indicators(current_is_staged(current_idx), &

+                                                  current_is_unstaged(current_idx), &

+                                                  current_is_untracked(current_idx), .false.)

+                    end if

+                    write(output_unit, '(a)') RESET

+                end if

+            else

+                write(output_unit, *)

+            end if

+        end do

+

+        ! Footer

+        if (in_git_repo) then

+            write(output_unit, '(a)') DIM // trim(repo_name) // ":" // trim(branch_name) // " | " // RESET // &

+                                     DIM // "↑↓:nav →:enter ←:back f:find A:add U:unstage M:commit c:cd q:quit" // RESET

+        else

+            write(output_unit, '(a)') DIM // "↑↓:nav →:enter ←:back f:find c:cd q:quit" // RESET

+        end if

+    end subroutine draw_interface

+

+    function get_file_color(filename, is_dir, is_exec) result(color)

+        character(len=*), intent(in) :: filename

+        logical, intent(in) :: is_dir, is_exec

+        character(len=20) :: color

+

+        ! Directories: Blue and bold

+        if (is_dir) then

+            color = BOLD // BLUE

+        ! Dotfiles: Grey

+        else if (filename(1:1) == '.') then

+            color = GREY

+        ! Executable files: Green

+        else if (is_exec) then

+            color = GREEN

+        ! All other files: White

+        else

+            color = WHITE

+        end if

+    end function get_file_color

+

+end module ui_display