install.js 8.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206
  1. import { writeFileSync, mkdirSync, readFileSync } from "fs";
  2. import yaml from 'js-yaml';
  3. import { execSync } from "child_process";
  4. import { docker } from "../server.js";
  5. import DockerodeCompose from "dockerode-compose";
  6. import { Syslog } from "../database/models.js";
  7. // This entire page hurts to look at.
  8. export const Install = async (req, res) => {
  9. let data = req.body;
  10. let { service_name, name, image, command_check, command, net_mode, restart_policy } = data;
  11. let { port0, port1, port2, port3, port4, port5 } = data;
  12. let { volume0, volume1, volume2, volume3, volume4, volume5 } = data;
  13. let { env0, env1, env2, env3, env4, env5, env6, env7, env8, env9, env10, env11 } = data;
  14. let { label0, label1, label2, label3, label4, label5, label6, label7, label8, label9, label10, label11 } = data;
  15. let ports = [port0, port1, port2, port3, port4, port5]
  16. let docker_volumes = [];
  17. if (image.startsWith('https://')){
  18. mkdirSync(`./appdata/${name}`, { recursive: true });
  19. execSync(`curl -o ./appdata/${name}/${name}_stack.yml -L ${image}`);
  20. console.log(`Downloaded stackfile: ${image}`);
  21. let stackfile = yaml.load(readFileSync(`./appdata/${name}/${name}_stack.yml`, 'utf8'));
  22. let services = Object.keys(stackfile.services);
  23. for ( let i = 0; i < services.length; i++ ) {
  24. try {
  25. console.log(stackfile.services[Object.keys(stackfile.services)[i]].environment);
  26. } catch { console.log('no env') }
  27. }
  28. } else {
  29. let compose_file = `version: '3'`;
  30. compose_file += `\nservices:`
  31. compose_file += `\n ${service_name}:`
  32. compose_file += `\n container_name: ${name}`;
  33. compose_file += `\n image: ${image}`;
  34. // Command
  35. if (command_check == 'on') {
  36. compose_file += `\n command: ${command}`
  37. }
  38. // Network mode
  39. if (net_mode == 'host') {
  40. compose_file += `\n network_mode: 'host'`
  41. }
  42. else if (net_mode != 'host' && net_mode != 'docker') {
  43. compose_file += `\n network_mode: '${net_mode}'`
  44. }
  45. // Restart policy
  46. if (restart_policy != '') {
  47. compose_file += `\n restart: ${restart_policy}`
  48. }
  49. // Ports
  50. for (let i = 0; i < ports.length; i++) {
  51. if ((ports[i] == 'on') && (net_mode != 'host')) {
  52. compose_file += `\n ports:`
  53. break;
  54. }
  55. }
  56. for (let i = 0; i < ports.length; i++) {
  57. if ((ports[i] == 'on') && (net_mode != 'host')) {
  58. compose_file += `\n - ${data[`port_${i}_external`]}:${data[`port_${i}_internal`]}/${data[`port_${i}_protocol`]}`
  59. }
  60. }
  61. // Volumes
  62. let volumes = [volume0, volume1, volume2, volume3, volume4, volume5]
  63. for (let i = 0; i < volumes.length; i++) {
  64. if (volumes[i] == 'on') {
  65. compose_file += `\n volumes:`
  66. break;
  67. }
  68. }
  69. for (let i = 0; i < volumes.length; i++) {
  70. // if volume is on and neither bind or container is empty, it's a bind mount (ex /mnt/user/appdata/config:/config )
  71. if ((data[`volume${i}`] == 'on') && (data[`volume_${i}_bind`] != '') && (data[`volume_${i}_container`] != '')) {
  72. compose_file += `\n - ${data[`volume_${i}_bind`]}:${data[`volume_${i}_container`]}:${data[`volume_${i}_readwrite`]}`
  73. }
  74. // if bind is empty create a docker volume (ex container_name_config:/config) convert any '/' in container name to '_'
  75. else if ((data[`volume${i}`] == 'on') && (data[`volume_${i}_bind`] == '') && (data[`volume_${i}_container`] != '')) {
  76. let volume_name = data[`volume_${i}_container`].replace(/\//g, '_');
  77. compose_file += `\n - ${name}_${volume_name}:${data[`volume_${i}_container`]}:${data[`volume_${i}_readwrite`]}`
  78. docker_volumes.push(`${name}_${volume_name}`);
  79. }
  80. }
  81. // Environment variables
  82. let env_vars = [env0, env1, env2, env3, env4, env5, env6, env7, env8, env9, env10, env11]
  83. for (let i = 0; i < env_vars.length; i++) {
  84. if (env_vars[i] == 'on') {
  85. compose_file += `\n environment:`
  86. break;
  87. }
  88. }
  89. for (let i = 0; i < env_vars.length; i++) {
  90. if (env_vars[i] == 'on') {
  91. compose_file += `\n - ${data[`env_${i}_name`]}=${data[`env_${i}_default`]}`
  92. }
  93. }
  94. // Labels
  95. let labels = [label0, label1, label2, label3, label4, label5, label6, label7, label8, label9, label10, label11]
  96. for (let i = 0; i < labels.length; i++) {
  97. if (labels[i] == 'on') {
  98. compose_file += `\n labels:`
  99. break;
  100. }
  101. }
  102. for (let i = 0; i < 12; i++) {
  103. if (data[`label${i}`] == 'on') {
  104. compose_file += `\n - ${data[`label_${i}_name`]}=${data[`label_${i}_value`]}`
  105. }
  106. }
  107. // Privileged mode
  108. if (data.privileged == 'on') {
  109. compose_file += `\n privileged: true`
  110. }
  111. // Hardware acceleration
  112. for (let i = 0; i < env_vars.length; i++) {
  113. if ((env_vars[i] == 'on') && (data[`env_${i}_name`] == 'DRINODE')) {
  114. compose_file += `\n deploy:`
  115. compose_file += `\n resources:`
  116. compose_file += `\n reservations:`
  117. compose_file += `\n devices:`
  118. compose_file += `\n - driver: nvidia`
  119. compose_file += `\n count: 1`
  120. compose_file += `\n capabilities: [gpu]`
  121. break;
  122. }
  123. }
  124. // add any docker volumes to the docker-compose file
  125. if ( docker_volumes.length > 0 ) {
  126. compose_file += `\n`
  127. compose_file += `\nvolumes:`
  128. // check docker_volumes for duplicates and remove them completely
  129. docker_volumes = docker_volumes.filter((item, index) => docker_volumes.indexOf(item) === index)
  130. for (let i = 0; i < docker_volumes.length; i++) {
  131. if ( docker_volumes[i] != '') {
  132. compose_file += `\n ${docker_volumes[i]}:`
  133. }
  134. }
  135. }
  136. try {
  137. mkdirSync(`./appdata/${name}`, { recursive: true });
  138. writeFileSync(`./appdata/${name}/docker-compose.yml`, compose_file, function (err) { console.log(err) });
  139. var compose = new DockerodeCompose(docker, `./appdata/${name}/docker-compose.yml`, `${name}`);
  140. } catch {
  141. const syslog = await Syslog.create({
  142. user: req.session.user,
  143. email: null,
  144. event: "App Installation",
  145. message: `${name} installation failed - error creating directory or compose file : ${err}`,
  146. ip: req.socket.remoteAddress
  147. });
  148. }
  149. try {
  150. (async () => {
  151. await compose.pull();
  152. await compose.up();
  153. const syslog = await Syslog.create({
  154. user: req.session.user,
  155. email: null,
  156. event: "App Installation",
  157. message: `${name} installed successfully`,
  158. ip: req.socket.remoteAddress
  159. });
  160. })();
  161. } catch (err) {
  162. const syslog = await Syslog.create({
  163. user: req.session.user,
  164. email: null,
  165. event: "App Installation",
  166. message: `${name} installation failed: ${err}`,
  167. ip: req.socket.remoteAddress
  168. });
  169. }
  170. }
  171. res.redirect('/');
  172. }