package_manager.js 8.5 KB

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