#!/bin/bash

# Script to show installed packages and their repository origins
# Enhanced version with options and clearer output

show_help() {
    cat << EOF
Usage: $0 [OPTIONS]

OPTIONS:
    -h, --help          Show this help
    -p, --ppa-only      Show only PPA packages
    -a, --all           Show all packages with repositories (default)
    -f, --format FORMAT Output format (default|json|csv)
    -s, --sort          Sort by repository
    -c, --count         Show count per repository
    -o, --output FILE   Save output to file

EXAMPLES:
    $0                  # List all packages with repositories
    $0 -p               # Only PPA packages
    $0 -f json          # JSON output
    $0 -c               # Count per repository
    $0 -p -o ppa.txt    # Save PPA list to file

REPOSITORY TYPES:
    - PPA: Personal Package Archives (Ubuntu)
    - Official: Ubuntu/Debian main repositories
    - Third-party: External repositories
    - Local: Locally installed packages
EOF
}

# Function to get package information
get_package_info() {
    local format="$1"
    local ppa_only="$2"
    local sort_by_repo="$3"
    
    # Get list of installed packages
    local packages=$(dpkg --get-selections | grep -v deinstall$ | awk '{print $1}')
    
    # Arrays to store results
    declare -A results
    declare -A repo_counts
    
    echo "Processing packages..." >&2
    
    local total=0
    local processed=0
    
    # Count total packages
    total=$(echo "$packages" | wc -l)
    
    # Process each package
    while IFS= read -r package; do
        if [[ -n "$package" ]]; then
            ((processed++))
            
            # Show progress
            if (( processed % 50 == 0 )); then
                echo "Processed $processed/$total packages..." >&2
            fi
            
            # Get package policy
            local policy_output=$(apt-cache policy "$package" 2>/dev/null)
            
            if [[ -n "$policy_output" ]]; then
                # Extract main repository
                local repo=$(echo "$policy_output" | grep -E '^\s+\*\*\*|^\s+[0-9]+' | head -1 | \
                           perl -ne 'if (/^\s+(?:\*\*\*\s+)?[0-9]+\s+(.*)/) { 
                               $repo = $1; 
                               $repo =~ s/\s+\[.*?\]$//;  # remove architecture
                               if ($repo =~ /http:\/\/ppa\.launchpad\.net\/([^\/\s]+\/[^\/\s]+)/) {
                                   print "PPA: $1\n";
                               } elsif ($repo =~ /http:\/\/([^\/\s]+)/) {
                                   print "Repository: $1\n";
                               } elsif ($repo =~ /\/var\/lib\/dpkg\/status/) {
                                   print "Local\n";
                               } else {
                                   print "Other: $repo\n";
                               }
                           }')
                
                if [[ -n "$repo" ]]; then
                    # Filter PPA if requested
                    if [[ "$ppa_only" == "true" && ! "$repo" =~ ^PPA: ]]; then
                        continue
                    fi
                    
                    results["$package"]="$repo"
                    repo_counts["$repo"]=$((${repo_counts["$repo"]} + 1))
                fi
            fi
        fi
    done <<< "$packages"
    
    echo "Done! Found ${#results[@]} packages." >&2
    echo "" >&2
    
    # Output based on requested format
    case "$format" in
        "json")
            echo "{"
            local first=true
            for package in $(printf '%s\n' "${!results[@]}" | sort); do
                if [[ "$first" == "true" ]]; then
                    first=false
                else
                    echo ","
                fi
                printf '  "%s": "%s"' "$package" "${results[$package]}"
            done
            echo ""
            echo "}"
            ;;
        "csv")
            echo "Package,Repository"
            for package in $(printf '%s\n' "${!results[@]}" | sort); do
                echo "$package,${results[$package]}"
            done
            ;;
        "count")
            echo "================================================"
            echo "  Packages per Repository"
            echo "================================================"
            echo ""
            printf "%-60s %s\n" "Repository" "Count"
            echo "------------------------------------------------"
            for repo in "${!repo_counts[@]}"; do
                printf "%-60s %s\n" "$repo" "${repo_counts[$repo]}"
            done | sort -t$'\t' -k2 -nr
            echo ""
            echo "Total repositories: ${#repo_counts[@]}"
            echo "Total packages: ${#results[@]}"
            ;;
        *)
            echo "================================================"
            echo "  Installed Packages by Repository"
            echo "================================================"
            echo ""
            if [[ "$sort_by_repo" == "true" ]]; then
                # Sort by repository
                local current_repo=""
                for package in "${!results[@]}"; do
                    echo "${results[$package]}|$package"
                done | sort | while IFS='|' read -r repo package; do
                    if [[ "$repo" != "$current_repo" ]]; then
                        echo ""
                        echo "[$repo]"
                        echo "---"
                        current_repo="$repo"
                    fi
                    echo "  - $package"
                done
            else
                # Sort by package name
                printf "%-40s -> %s\n" "Package" "Repository"
                echo "------------------------------------------------"
                for package in $(printf '%s\n' "${!results[@]}" | sort); do
                    printf "%-40s -> %s\n" "$package" "${results[$package]}"
                done
            fi
            echo ""
            echo "Total packages: ${#results[@]}"
            ;;
    esac
}

# Parse arguments
format="default"
ppa_only=false
sort_by_repo=false
output_file=""

while [[ $# -gt 0 ]]; do
    case $1 in
        -h|--help)
            show_help
            exit 0
            ;;
        -p|--ppa-only)
            ppa_only=true
            shift
            ;;
        -a|--all)
            # Default behavior
            shift
            ;;
        -f|--format)
            format="$2"
            shift 2
            ;;
        -s|--sort)
            sort_by_repo=true
            shift
            ;;
        -c|--count)
            format="count"
            shift
            ;;
        -o|--output)
            output_file="$2"
            shift 2
            ;;
        *)
            echo "Unknown option: $1" >&2
            show_help
            exit 1
            ;;
    esac
done

# Check dependencies
if ! command -v apt-cache &> /dev/null; then
    echo "Error: apt-cache not found. This script works only on Debian/Ubuntu systems." >&2
    exit 1
fi

if ! command -v dpkg &> /dev/null; then
    echo "Error: dpkg not found." >&2
    exit 1
fi

# Execute main function
if [[ -n "$output_file" ]]; then
    get_package_info "$format" "$ppa_only" "$sort_by_repo" > "$output_file"
    echo "Output saved to: $output_file" >&2
else
    get_package_info "$format" "$ppa_only" "$sort_by_repo"
fi

chmod +x package-audit.sh

#!/bin/bash
# Analyze repository distribution

echo "Repository Distribution Analysis"
echo "================================="
echo ""

# Count by type
echo "By Type:"
./package-audit.sh -c | grep -E "^(PPA|Repository|Local)" | \
    awk -F: '{type=$1; count=$2; types[type]+=count} END {
        for (t in types) print t": "types[t]
    }' | sort -t: -k2 -nr

echo ""
echo "Top 10 Repositories:"
./package-audit.sh -c | head -12 | tail -10

# List packages from specific PPA
PPA_NAME="ppa-owner/ppa-name"

dpkg --get-selections | awk '!/deinstall$/ {print $1}' | \
while read package; do
    apt-cache policy "$package" | grep -q "$PPA_NAME" && echo "$package"
done

# Generate package list on system 1
./package-audit.sh -f csv > system1.csv

# Generate package list on system 2
./package-audit.sh -f csv > system2.csv

# Compare
diff <(sort system1.csv) <(sort system2.csv)

# Or use comm for better comparison
comm -3 <(sort system1.csv) <(sort system2.csv)

#!/bin/bash
# Find PPAs with no installed packages

echo "Checking for unused PPAs..."

# Get list of PPAs
ppas=$(grep -r --include '*.list' '^deb ' /etc/apt/sources.list.d/ | \
       grep ppa.launchpad.net | \
       sed 's/.*ppa\.launchpad\.net\/\([^\/]*\/[^\/]*\).*/\1/' | \
       sort -u)

# Check each PPA
while IFS= read -r ppa; do
    count=$(dpkg --get-selections | awk '!/deinstall$/ {print $1}' | \
            xargs -I{} apt-cache policy {} 2>/dev/null | \
            grep -c "ppa.launchpad.net/$ppa")
    
    if [ "$count" -eq 0 ]; then
        echo "Unused PPA: $ppa (0 packages)"
        echo "  Remove with: sudo add-apt-repository --remove ppa:$ppa"
    else
        echo "Active PPA: $ppa ($count packages)"
    fi
done <<< "$ppas"

#!/bin/bash
# Security audit report for packages

cat > security-audit.sh << 'EOF'
#!/bin/bash

echo "================================================"
echo "  Package Security Audit Report"
echo "  Generated: $(date)"
echo "================================================"
echo ""

echo "1. Packages from PPAs:"
echo "----------------------"
./package-audit.sh -p -c
echo ""

echo "2. Packages from Third-party Repositories:"
echo "-----------------------------------------"
./package-audit.sh | grep "Repository:" | \
    grep -v "archive.ubuntu.com" | \
    grep -v "security.ubuntu.com" | \
    grep -v "ppa.launchpad.net" | \
    sort -u
echo ""

echo "3. Locally Installed Packages:"
echo "------------------------------"
./package-audit.sh | grep "^Local" | wc -l
echo ""

echo "4. Unsigned Repositories:"
echo "------------------------"
grep -r "trusted=yes" /etc/apt/sources.list /etc/apt/sources.list.d/ 2>/dev/null
echo ""

echo "5. Outdated Package Keys:"
echo "------------------------"
apt-key list | grep -E "expired:"
EOF

chmod +x security-audit.sh
./security-audit.sh

# Cache apt-cache policy output
dpkg --get-selections | awk '!/deinstall$/ {print $1}' | \
    xargs apt-cache policy > /tmp/apt-policy-cache.txt

# Then process the cache file instead of calling apt-cache repeatedly
# This speeds up processing significantly

sudo apt install perl

# Alternative without perl
dpkg --get-selections | awk '!/deinstall$/ {print $1}' | \
while read pkg; do
    repo=$(apt-cache policy "$pkg" | grep -m1 "500 http" | \
           awk '{print $2}')
    echo "$pkg: $repo"
done

# Check package files directly
dpkg -L package-name | head -5

# Check package origin
dpkg -s package-name | grep -E "^(Package|Version|Status|Origin):"

# Check if it's a manually installed .deb
grep "Package: package-name" /var/lib/dpkg/status -A20 | grep "Origin"

#!/bin/bash
# Track repository changes

LOGDIR="$HOME/package-logs"
mkdir -p "$LOGDIR"

LOGFILE="$LOGDIR/packages-$(date +%Y%m%d-%H%M%S).csv"

# Generate current snapshot
./package-audit.sh -f csv > "$LOGFILE"

echo "Snapshot saved: $LOGFILE"

# Compare with previous
PREVIOUS=$(ls -t "$LOGDIR"/packages-*.csv | sed -n '2p')

if [ -n "$PREVIOUS" ]; then
    echo ""
    echo "Changes since last snapshot:"
    echo "----------------------------"
    
    # Added packages
    echo "Added:"
    comm -13 <(sort "$PREVIOUS") <(sort "$LOGFILE") | head -10
    
    # Removed packages
    echo "Removed:"
    comm -23 <(sort "$PREVIOUS") <(sort "$LOGFILE") | head -10
fi

# Run weekly
0 0 * * 0 /path/to/track-repos.sh

#!/bin/bash
# Generate installation documentation

cat > system-documentation.md << EOF
# System Package Documentation
Generated: $(date)
Hostname: $(hostname)

## Package Summary

$(./package-audit.sh -c)

## Packages by Repository

$(./package-audit.sh -s)

## PPA Packages

$(./package-audit.sh -p)

## Repository Configuration

\`\`\`
$(grep -r --include '*.list' '^deb ' /etc/apt/sources.list /etc/apt/sources.list.d/)
\`\`\`

## System Information

- OS: $(lsb_release -d | cut -f2-)
- Kernel: $(uname -r)
- Architecture: $(dpkg --print-architecture)

EOF

echo "Documentation generated: system-documentation.md"