From 3e989992df4e918f2d7ecbcb4ff3994a9895b9d3 Mon Sep 17 00:00:00 2001
From: William Mantly <wmantly@gmail.com>
Date: Sat, 10 Aug 2024 23:09:55 -0400
Subject: [PATCH] Added digital ocean

---
 nodejs/models/dns_provider/digitalocean.js | 113 +++++++++++++++++++++
 1 file changed, 113 insertions(+)
 create mode 100644 nodejs/models/dns_provider/digitalocean.js

diff --git a/nodejs/models/dns_provider/digitalocean.js b/nodejs/models/dns_provider/digitalocean.js
new file mode 100644
index 0000000..8ff5306
--- /dev/null
+++ b/nodejs/models/dns_provider/digitalocean.js
@@ -0,0 +1,113 @@
+'use strict';
+
+const axios = require('axios');
+
+
+class DigitalOcean{
+	static _keyMap = {
+		token: {isRequired: true, type: 'string', isPrivate: true, displayName: 'API Token'},
+	}
+
+	static displayName = 'DigitalOcean'
+	static displayIconHtml = '<i class="fa-brands fa-digital-ocean"></i>'
+	static displayIconUni =  '&#xf391;'
+
+	constructor(token){
+		this.token = token.token || token;
+	}
+
+	__typeCheck(type){
+		if(!type) return;
+		if(!['A', 'MX', 'CNAME', 'ALIAS', 'TXT', 'NS', 'AAAA', 'SRV', 'TLSA', 'CAA', 'HTTPS', 'SVCB'].includes(type)) throw new Error(`${this.constructor.name} API: Invalid 'type' passed`)
+	}
+
+	__parseName(domain, name){
+		if(name && !name.endsWith('.'+domain)){
+			return `${name}.${domain}`
+		}
+		return name;
+	}
+
+	get axios(){
+		try{
+			return axios.create({
+				baseURL: 'https://api.digitalocean.com/v2/',
+				headers: {Authorization: `Bearer ${this.token}`}
+			});
+		}catch(error){
+			console.log(error.data);
+		}
+	}
+
+	async listDomains(){
+		try{
+			let res = await this.axios.get('/domains');
+			for(let domain of res.data.domains){
+				domain.domain = domain.name
+			}
+			return res.data.domains;
+		}catch{}
+	}
+
+	async getRecords(domain, options={}){
+		this.__typeCheck(options.type);
+		let res = await this.axios.get(`/domains/${domain}/records`, {params: options})
+		let records = [];
+
+		for(let record of res.data.domain_records){
+			let matchCount = 0
+			for(let key in options){
+				if(record[key] === options[key] && ++matchCount === Object.keys(options).length){
+					records.push(record)
+				}
+			}
+		}
+		return records;
+	}
+
+	async createRecord(domain, options){
+		this.__typeCheck(options.type);
+		if(!options.data) throw new Error(`${this.constructor.name} API: 'data' key is required for this action`)
+		if(options.name) options.name = this.__parseName(domain, options.name);
+
+		let res = await this.axios.post(`/domains/${domain}/records`, options);
+		return res.data;
+	}
+
+	async deleteRecordById(domain, id){
+		let res = await this.axios.delete(`/domains/${domain}/records/${id}`);
+	}
+
+	async deleteRecords(domain, options){
+		let records = await this.getRecords(domain, options)
+		for(let record of records){
+			let res = await this.deleteRecordById(domain, record.id);
+		}
+
+		return true;
+	}
+}
+
+module.exports = DigitalOcean;
+
+
+if(require.main === module){(async function(){try{
+	// const conf = require('../conf');
+
+	// console.log(await digi.listDomains())
+
+	// console.log('make', await digi.createRecord('rm-rf.stream', {name:'_test', data: '890', type: "TXT"}))
+	// console.log('delete', await digi.deleteRecords('rm-rf.stream', {type: 'TXT'}))
+	// console.log('get', await digi.getRecords('rm-rf.stream', {type:'TXT', data:'890'}))
+
+
+	// let porkBun = new PorkBun(conf.porkBun.apiKey, conf.porkBun.secretApiKey);
+
+	// console.log(await porkBun.listDomains())
+
+	// console.log(await porkBun.deleteRecordById('holycore.quest', '415509355'))
+	// console.log('IIFE', await porkBun.createRecordForce('holycore.quest', {type:'A', name: 'testapi', content: '127.0.0.5'}))
+	// console.log('IIFE', await porkBun.getRecords('holycore.quest', {type:'A', name: 'testapi'}))
+}catch(error){
+	console.log('IIFE Error:', error)
+}})()}