mirror of
https://github.com/ItsDrike/dotfiles.git
synced 2025-04-28 05:02:26 +00:00
113 lines
3.6 KiB
Python
113 lines
3.6 KiB
Python
from lib import Input, Print, Install, Path
|
|
import os
|
|
|
|
|
|
def make_backup(backup_floder):
|
|
if Input.yes_no('Do you wish to create a backup of current state of your dotfiles?'):
|
|
Print.action('Creating dotfiles backup')
|
|
Print.err('Backup is not yet implemented')
|
|
# TODO: Create backup
|
|
return True
|
|
else:
|
|
Print.warning('Proceeding without backup')
|
|
return False
|
|
|
|
|
|
def check_installation():
|
|
if Install.check_not_installed('zsh'):
|
|
if not Install.package('zsh', 'default + (This is required shell for dotfiles)'):
|
|
Print.err('Dotfiles installation cancelled - zsh not installed')
|
|
return False
|
|
|
|
global oh_my_zsh_path
|
|
oh_my_zsh_path = None
|
|
if Path.check_dir_exists('~/.oh-my-zsh'):
|
|
oh_my_zsh_path = '$HOME/.oh-my-zsh'
|
|
elif Path.check_dir_exists('~/oh-my-zsh'):
|
|
oh_my_zsh_path = '$HOME/oh-my-zsh'
|
|
elif Path.check_dir_exists('~/ohmyzsh'):
|
|
oh_my_zsh_path = '$HOME/ohmyzsh'
|
|
elif Path.check_dir_exists('~/.config/oh-my-zsh'):
|
|
oh_my_zsh_path = '$HOME/.config/oh-my-zsh'
|
|
elif Path.check_dir_exists('/usr/share/oh-my-zsh'):
|
|
oh_my_zsh_path = '/usr/share/oh-my-zsh'
|
|
|
|
if oh_my_zsh_path:
|
|
return True
|
|
else:
|
|
Print.err('oh-my-zsh is not installed, cannot proceed...')
|
|
# TODO: Option to install
|
|
return False
|
|
|
|
|
|
def personalized_changes(file):
|
|
if '.zshrc' in file:
|
|
filedata = None
|
|
with open(file, 'r') as f:
|
|
filedata = f.read()
|
|
filedata_old = filedata
|
|
filedata = filedata.replace('"$HOME/.config/oh-my-zsh"', f'"{oh_my_zsh_path}"')
|
|
if filedata_old != filedata:
|
|
Print.commend('Changing oh-my-zsh location in .zshrc')
|
|
with open(file, 'w') as f:
|
|
f.write(filedata)
|
|
|
|
|
|
def init(symlink):
|
|
# Get path to files/ floder (contains all dotfiles)
|
|
files_dir = os.path.join(
|
|
Path.get_parent(__file__), 'files')
|
|
# Create optional backup
|
|
make_backup(files_dir)
|
|
# Go through all files
|
|
for file in Path.get_all_files(files_dir):
|
|
# Make personalized changes to files
|
|
personalized_changes(file)
|
|
# Set symlink position ($HOME/filepath)
|
|
position = os.path.join(
|
|
Path.get_home(), file.replace(f'{files_dir}/', ''))
|
|
|
|
if symlink:
|
|
# Make symlink
|
|
Path.create_symlink(file, position)
|
|
else:
|
|
Path.copy(file, position)
|
|
|
|
|
|
def main():
|
|
Print.action('Installing dotfiles')
|
|
|
|
if not check_installation():
|
|
return False
|
|
|
|
choice = Input.multiple('Do you wish to create dotfiles', [
|
|
'symlinks (dotfiles/ dir will be required)', 'files (dotfiles/ dir can be removed afterwards)'])
|
|
# Create symlinks
|
|
if choice == 'symlinks (dotfiles/ dir will be required)':
|
|
init(symlink=True)
|
|
|
|
# Final prints
|
|
Print.action('Symlink installation complete')
|
|
Print.warning(
|
|
'Do not delete this floder, all dotfile symlinks are linked to it')
|
|
Print.warning(
|
|
'Deletion would cause errors on all your dotfiles managed by this program')
|
|
Print.comment(
|
|
'If you wish to remove this floder, please select files instead of symlinks for dotfile creation')
|
|
|
|
# Create files
|
|
elif choice == 'files (dotfiles/ dir can be removed afterwards)':
|
|
init(symlink=False)
|
|
|
|
# Final prints
|
|
Print.action('Dotfiles successfully created')
|
|
Print.comment('This directory can now be removed')
|
|
|
|
# Don't create dotfiles
|
|
else:
|
|
Print.cancel('Dotfiles installation cancelled')
|
|
return False
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|