This repository has been archived by the owner on Jun 27, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 2
/
post_install.sh
executable file
·183 lines (163 loc) · 6.21 KB
/
post_install.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
#!/usr/bin/env bash
## The primary service for this jail will be installed in a Python Virtualenv
## When the python version changes, the virtualenv should be recreated
service_name='esphome'
service_port='6052'
## Which version of Python to use
service_python='python3.8'
## Location of the Virtualenv
service_venv="/usr/local/srv/${service_name}"
## Config directory for the service
service_config_dir="/var/db/${service_name}"
## Who will run the jail's primary service
## Here the user and group are both set to service_name
service_user="${service_name}"
service_group="${service_name}"
## Also, the UID is set to the service_port
service_uid="${service_port}"
## Home directory for the service_user
service_home="/usr/local/home/${service_user}"
# shellcheck disable=SC1091
. /etc/rc.subr && load_rc_config ${service_name}
service_python="${esphome_python:-$(which ${service_python})}"
service_venv="${esphome_venv:-${service_venv}}"
service_config_dir="${esphome_config_dir:-${service_config_dir}}"
service_user="${esphome_user:-${service_user}}"
service_group="${esphome_group:-${service_group}}"
run_once() {
## Add service_user, install the service_home and service_config_dir directories
pw adduser -u "${service_uid}" -n "${service_user}" -d "${service_home}" -w no -s /usr/local/bin/bash
install -d -m 775 -o "${service_user}" -g "${service_group}" -- "${service_home}"
install -d -m 775 -o "${service_user}" -g "${service_group}" -- "${service_config_dir}"
## Launch the console menu upon login
{ echo -e "\n# Start console menu upon login." ; \
echo "if ( -x /root/bin/menu ) menu" ; } >> /root/.login
}
set_rcvars() {
echo -e "\nSetting rcvars for service...\n"
sysrc ${service_name}_umask="002"
sysrc ${service_name}_user="${service_user}"
sysrc ${service_name}_group="${service_group}"
sysrc ${service_name}_config_dir="${service_config_dir}"
sysrc ${service_name}_python="${service_python}"
sysrc ${service_name}_venv="${service_venv}"
}
install_toolchains() {
## This function is also called during a Plugin UPDATE
local url pkg1 pkg2
url=https://github.com/tprelog/iocage-esphome/raw/toolchains/toolchains
pkg1=toolchain-xtensa-FreeBSD.12.amd64-2.40802.191122.pkg
pkg2=toolchain-xtensa32-FreeBSD.12.amd64-2.50200.0.pkg
echo -e "\nAttempting to add ESP8266 support for FreeBSD"
fetch --quiet ${url}/${pkg1} -o /tmp/${pkg1} \
&& pkg install -y /tmp/${pkg1}
echo -e "\nAttempting to add ESP32 support for FreeBSD"
fetch --quiet ${url}/${pkg2} -o /tmp/${pkg2} \
&& pkg install -y /tmp/${pkg2}
# shellcheck disable=SC2016
su - "${service_user}" -c '
mkdir -p ~/.platformio/packages
cd ~/.platformio/packages || exit
rm -rf -- toolchain-xtensa toolchain-xtensa32 xtensa-lx106-elf
ln -s /usr/local/esp-quick-toolchain-gcc48/xtensa-lx106-elf .
ln -s /usr/local/xtensa-esp32-elf-idf3 toolchain-xtensa32
'
}
install_service() {
echo -e "\nCreating virtualenv...\n"
local required R='https://github.com/raw/esphome/esphome/release/requirements.txt'
local optional O='https://github.com/raw/esphome/esphome/release/requirements_optional.txt'
if [ -d "${service_venv}" ] && [ -n "$(ls -A "${service_venv}")" ]; then
echo -e "${red} directory is not empty:${end} ${service_venv}" ; exit 1
else
required=$(mktemp -t "${service_name}.required")
fetch --quiet ${R} -o "${required}" \
&& chown "${service_user}" "${required}"
optional=$(mktemp -t "${service_name}.optional")
fetch --quiet ${O} -o "${optional}" \
&& chown "${service_user}" "${optional}"
install -d -g "${service_group}" -m 775 -o "${service_user}" -- "${service_venv}"
# shellcheck disable=SC2016
su - "${service_user}" -c '
${1} -m venv ${2}
source ${2}/bin/activate || exit 1
pip install --upgrade pip wheel
pip install ${3} -r ${4} -r ${5} || exit 1
deactivate
' _ "${service_python}" "${service_venv}" "${service_name}" "${required}" "${optional}"
rm -f -- "${optional}" "${required}"
fi
}
enablestart_service() {
sysrc ${service_name}_enable=YES
service ${service_name} start
}
upgrade_service() {
local restart='N'
service ${service_name} stop && restart='Y'
# shellcheck disable=SC2016
su - "${service_user}" -c '
source ${1}/bin/activate || exit 1
pip install --upgrade ${2}
deactivate
' _ "${service_venv}" "${service_name}"
if [ ${restart} == 'Y' ]; then
service ${service_name} start
fi
}
disablestop_service() {
sysrc ${service_name}_enable=NO
service ${service_name} onestop
}
remove_service() {
local home; home="$(getent passwd "${service_user}" | cut -d: -f6)"
echo -e "\n${orn}WARN:${end} You are about to remove ${service_name}"
echo -e "${orn}WARN:${end} The following directory will be deleted -- ${service_venv}"
echo -e "${orn}WARN:${end} The following directory will be deleted -- ${home}/.platformio"
echo -e "${orn}WARN:${end} The following directory will be deleted -- ${home}/.cache\n"
read -rp " Type 'YES' to continue: " response
[[ "${response}" == [Yy][Ee][Ss] ]] || return 1 ; echo
disablestop_service
rm -rf -- "${service_venv}" "${home}/.platformio" "${home}/.cache"
}
unset_rcvars() {
for var in $(eval sysrc -ae | grep ^"${service_name}"); do
echo "removing rcvar: ${var}"
sysrc -x "${var}" | cut -d= -f1
done
}
show_url(){
## Get the jail's current ip address and show a link to the default admin portal
service_ip=$(ifconfig | sed -En 's/127.0.0.1//;s/.*inet (addr:)?(([0-9]*\.){3}[0-9]*).*/\2/p')
echo -e "\nAdmin Portal:\nhttp://${service_ip}:${service_port}\n"
}
colors() {
red=$'\e[1;31m'
orn=$'\e[38;5;208m'
end=$'\e[0m'
}; colors
if [ -z "${1}" ]; then
## Install the ESPHome iocage-plugin
run_once \
&& set_rcvars \
&& install_service \
&& install_toolchains \
&& enablestart_service
elif [ "${1}" == 'install' ]; then
install_service \
&& install_toolchains \
&& enablestart_service \
&& show_url
elif [ "${1}" == 'remove' ]; then
remove_service
elif [ "${1}" == 'upgrade' ]; then
upgrade_service
elif [ "${1}" == 'install-toolchains' ]; then
install_toolchains
elif [ "${1}" == 'set-rcvars' ]; then
set_rcvars
elif [ "${1}" == 'unset-rcvars' ]; then
unset_rcvars
else
echo "unknown: ${1}"
fi