pullfiles.sh

#!/bin/bash
#VillainMod Git Handler

##################
# FUNCTIONS BELOW
##################

# detect changes to a given git repo
detect_changes()
{
    changes=0
    # detect changes to existing files
    git diff --no-ext-diff --quiet --exit-code || changes=1
    # if no changed files, did they add any more?
    if [[ $changes = 0 ]]; then
	printf "No changes to commit"
        changes=`git ls-files --exclude-standard --others| wc -l`
    fi
    # now look at changes variable to see if changes exist

}

# function to actually sync our local repositories with the remote ones, without losing any untracked changes (COUGH REPO)
fetch_remotes()
{
    # for each subfolder, decide if it is a valid git repository
    for folder in *
    do
        cd ${folder}
        if [[ -d ${folder}/.git ]]
        then
            changes=0
            detect_changes
            
            # sync sources and merge
            git fetch origin
            git merge origin/master
            
            #if changes were stashed, let's re-apply them and hope for the best!
            if [[ $changes = 1 ]]
            then
                git stash pop
            fi
            
        #else
        #	something something
        fi
    done
}

push_changes() 
{
    # for each subfolder, decide if it is a valid git repository
    for folder in *
    do
        cd ${folder}
        if [[ -d ${folder}/.git ]]
        then
            changes=0
            detect_changes
            # now let's see if we should commit them
            if [[ $changes = 1 ]]
            then
                # changes were made, let's ask the user if they want to push them
                printf "Uncommitted changes detected, add all and commit? (this will add all new and changed files, ensure your .gitignore is set up correctly! [y/N]\n"
                read want_to_commit
                shopt -s nocasematch
                if [[ "$want_to_commit" == "Y" ]] 
                then
                    # user wants to add and commit all changes
                    git add .
                    git commit
                    # user will enter commit message and commit is now done
                    
                    # if user says they are allowed to push to the remote tree
                    if [[ $IS_REPO_ADMIN = 1 ]]
                    then
                        printf "Push these changes now? [y/N]\n"
                        read push_changes
                        if [[ "$push_changes" == "Y" ]]
                        then
                        
                            branch_name="$(git symbolic-ref HEAD 2>/dev/null)" || 
                            branch_name="(unnamed branch)"     # detached HEAD
                            branch_name=${branch_name##refs/heads/}
                        
                            git push origin $branch_name
                        fi
                    fi
                fi
            fi
        fi
    done    
}

device_selection()
{
	printf "Please select the manufacturer of your device:\n"
	printf "1:  HTC\n"
	printf "2:  Samsung\n"
	printf "3:  HP\n"
	printf "Type the corresponding number and press ENTER:\n"
	read device_manufacturer
	
	if [[ "$device_manufacturer" == "1" ]]
	then
		printf "Device Manufacturer: HTC"
		manu="htc"
		printf "Please choose a device:\n"
		printf "1. EVO 3D\n"
		printf "2: Sensation\n"
		printf "Type the corresponding number and press ENTER:\n"
		read manufacturer_model
		[[ "$manufacturer_model" == "1" ]] && device="shooteru"
		[[ "$manufacturer_model" == "2" ]] && device="pyramid"
	elif [[ "$device_manufacturer" == "2" ]]
	then
		printf "Device Manufacturer: Samsung\n"
		manu="samsung"
		printf "Please choose a device:\n"
		printf "1.  Galaxy S2\n"
		printf "2:  Galaxy Nexus\n"
		printf "3:  Nexus S\n"
		printf "Type the corresponding number and press ENTER:\n"
		read manufacturer_model
		[[ "$manufacturer_model" == "1" ]] && device="galaxys2"
		[[ "$manufacturer_model" == "2" ]] && device="maguro"
		[[ "$manufacturer_model" == "3" ]] && device="crespo"
	elif [[ "$device_manufacturer" == "3" ]]
	then
		printf "Device Manufacturer: HP\n"
		manu="hp"
		printf "Please choose a device:\n"
		printf "1:  Touchpad\n"
		printf "Type the corresponding number and press ENTER:\n"
		read manufacturer_model
		[[ "$manufacturer_model" == "1" ]] && device="tenderloin"
	else
		printf "\n\nInput a valid entry!\n\n"
		device_selection
	fi
}

pull_selected_devices()
{
	printf "You can only pull device trees for devices VillainMod officially supports.\n"
	printf "If you would like to build for an unsupported device, please pull the tree manually.\n"
	printf "To get official support for your device, please submit a patch via gerrit.\n"
	device_selection
	
	#Check device folder exists
	if [[ ! -d "device" ]]
	then
		printf "Device directory does not exist!\n"
		printf "Creating directory..\n"
		mkdir -p "device"
		cd "device"
	else
		cd "device"
	fi
	if [[ ! -d "$manu" ]]
	then
		printf "$manu directory does not exist!\n"
		printf "Creating directory..\n"
		mkdir -p "$manu"
		cd "$manu"
	else
		cd "$manu"
	fi
	git clone "git@github.com:VillainMod/android_device_"$manu"_"$device".git" "$device"

}

submit_gerrit_patch()
{
	printf "Please type the path of the directory you have made your changes in and press ENTER:\n"
	printf "$PWD\n"
	read change_dir
	while [[ ! -d "$change_dir" ]]; do
      		printf "\nNo such directory! Specify directory changes have been made in:\n"
      		read change_dir
	done
	cd "$change_dir"
	printf "\n\nCurrent directory: $PWD\n"
	printf "Is this the direcory you have made your changes in?\n"
	printf "Y/n:\n"
	read confirm_dir
	shopt -s nocasematch
	if [[ "$confirm_dir" == "Y" ]] 
	then
		printf "Gathering local changes..\n\n"
		shopt -u nocasematch
		check_for_changes
	elif [[ "$confirm_dir" == "n" ]]
	then
		printf "Would you like to switch directories?\n"
		printf "Y/n:\n"
	else
		printf "Gathering local changes..\n\n"
		shopt -u nocasematch
		check_for_changes
	fi
}

check_for_changes()
{
    for folder in *
    do
        cd ${folder}
        if [[ -d ${folder}/.git ]]
        then
            changes=0
            detect_changes
            # now let's see if we should commit them
            if [[ $changes = 1 ]]
            then
		printf "Changes detected!"
	    fi
	fi
   done
}

display_menu()
{
while [[ "$menu_choice" != "X" || "$menu_choice" != "x" ]]; do
  printf "1: Sync local tree\n"
  printf "2: Push local changes\n"
  printf "3: Pull device trees\n"
  printf "4: Submit gerrit patch\n"
  printf "X: Type X to exit\n"
  printf "What would you like to do? [1-4]:\n"
  read menu_choice
  if [[ "$menu_choice" == "1" ]]
  then
    printf "Syncing local tree\n"
    fetch_remotes
  elif [[ "$menu_choice" == "2" ]]
  then
	printf "Gathering local changes\n"
	detect_changes
  elif [[ "$menu_choice" == "3" ]]
  then
	pull_selected_devices
  elif [[ "$menu_choice" == "4" ]]
  then
	submit_gerrit_patch
  elif [[ "$menu_choice" == "X" || "$menu_choice" == "x" ]]
  then
	exit
  else
    printf "\nPlease select a valid option!\n"
  fi
done
}

##################
# FUNCTIONS ABOVE
##################

if [[ -n "$1" ]]
# Test whether command-line argument is present (non-empty).
then
  lines=$1
else  
  lines=$LINES # Default, if not specified on command-line.
fi

DEF_DIR="VillainMod"
# if the user has write access to push to the repo (this is self-selected by the user, it's not a security feature, just enables push features)
IS_REPO_ADMIN="1"

printf "Welcome to the VillainMod Git Handler\n"

# Check for git
if [[ ! $(which git) > /dev/null ]]; then
   printf "\nCommand "git" not found! Please install git and run the script again.\n\n"
   exit 0
fi

printf "Current directory:\n$PWD\n\n"

# Set up working dir and display menu
if [[ ! -d VillainMod ]]
then
  printf "Create working directory here? [Y/n]: "
  read confirm_dir_create
  if [[ "$confirm_dir_create" == "Y" || "$confirm_dir_create" == "" || "$confirm_dir_create" == "y" ]] 
  then
    printf "\nCreating directory 'VillainMod'...\n"
    mkdir "$PWD/$DEF_DIR"
    cd "VillainMod"
    display_menu
  else
    printf "Specify working directory:\n"
    read alt_working
    while [[ ! -d $alt_working ]]; do
      printf "\nNo such directory! Specify working directory:\n"
      read alt_working
    done
    cp $PWD/pullfiles.sh $alt_working
    cd $alt_working
  fi
else
  printf "Found working directory 'VillainMod'!\n"
  #cd into working dir
  #start menu
  cd "VillainMod"
  printf "Current directory:\n$PWD\n\n"
  display_menu
fi