Current File : //snap/firefox/5091/patch-default-profile.py
#!/usr/bin/python3
# Copyright (C) 2021 Canonical Ltd.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 3, as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
import configparser
import os.path
import sys
# Mozilla products, when not packaged as a snap, use dedicated profiles per
# installation by default, unless instructed otherwise (legacy mode). See
# https://support.mozilla.org/en-US/kb/understanding-depth-profile-installation
# for details.
# We want to import existing profiles from a package of Firefox typically
# installed in a well-known location. The following is a list of hashes for
# such well-known locations, in descending order of popularity/likeliness
# on distributions compatible with snaps:
KNOWN_INSTALL_HASHES = [
'4F96D1932A9F858E', # /usr/lib/firefox (Ubuntu, Debian, Mint, Arch,
# Pop!_OS, Zorin OS, Manjaro, KDE Neon, elementary OS)
'11457493C5A56847', # /usr/lib64/firefox (Fedora)
'3B6073811A6ABF12', # /usr/lib/firefox-esr (Debian)
'46F492E0ACFF84D4', # /usr/lib/firefox-developer-edition (Arch)
]
def _patch_imported_profiles(profiles_file):
profiles = configparser.RawConfigParser()
profiles.optionxform = lambda option: option
profiles.read(profiles_file)
known_install = False
for known_hash in KNOWN_INSTALL_HASHES:
install_section = 'Install{}'.format(known_hash)
if install_section in profiles:
known_install = True
break
if not known_install:
return
try:
profile_path = profiles.get(install_section, 'Default')
except configparser.NoOptionError:
return
if not profile_path:
return
print('Found default profile: {}'.format(profile_path))
for section in profiles:
if section.startswith('Profile'):
try:
path = profiles.get(section, 'Path')
except configparser.NoOptionError:
continue
else:
if path == profile_path:
# We found the section for the default profile,
# explicitly mark it as such …
profiles.set(section, 'Default', '1')
# … and remove the default marker from any other profile
# that might have had it.
for other_section in profiles:
if other_section.startswith('Profile') and \
other_section != section:
profiles.remove_option(other_section, 'Default')
# Delete the Install section as it is meaningless
# (and unused) in legacy mode.
profiles.remove_section(install_section)
# Write back the modified profiles.ini
with open(profiles_file, 'w') as profiles_fd:
profiles.write(profiles_fd, False)
return
if __name__ == '__main__':
if len(sys.argv) != 2 or not os.path.isfile(sys.argv[1]):
expected_arg = '/path/to/profiles_dir/profiles.ini'
print('Usage: {} {}'.format(sys.argv[0], expected_arg))
sys.exit(1)
_patch_imported_profiles(sys.argv[1])
Mini Shell