Added relations to redis ORM

nodejs/models/dns_provider.js

@@ -0,0 +1,181 @@
+'use strict';
+
+const crypto = require("crypto");
+
+const conf = require('../conf');
+const Table = require('../utils/redis_model');
+const ModelPs = require('../utils/model_pubsub');
+
+const tldExtract = require('tld-extract').parse_host;
+
+const providers = {
+	Cloudflare: require('./dns_provider/cloudflare'),
+	DigitalOcean: require('./dns_provider/digitalocean'),
+	PorkBun: require('./dns_provider/porkbun'),
+};
+
+class Domain extends Table{
+	static _key = 'domain';
+	static _keyMap = {
+		'created_by': {isRequired: true, type: 'string', min: 3, max: 500},
+		'created_on': {default: function(){return (new Date).getTime()}},
+		'updated_by': {default:"__NONE__", isRequired: false, type: 'string',},
+		'updated_on': {default: function(){return (new Date).getTime()}, always: true},
+		'domain': {isRequired: true, type: 'string'},
+		'dnsProvider_id': {isRequired: true, type: 'string'},
+		'provider': {model: 'DnsProvider', rel:'one', localKey: 'dnsProvider_id'},
+		'zoneId': {isRequired: false, type: 'string'},
+	}
+
+	static async get(domain, ...args){
+		try{
+			domain = tldExtract(domain).domain;
+		}catch{}
+		
+		let instance = await super.get(domain, ...args);
+
+		return instance;
+	}
+
+	static async getByProviderId(id, ...args){
+		let domains = await this.listDetail(...args);
+		let results = [];
+		for(let domain of domains){
+			if(domain.dnsProvider_id == id) results.push(domain);
+		}
+
+		return results;
+	}
+
+	// toJSON(){
+	// 	return {
+	// 		...super.toJSON(),
+	// 		provider: {
+	// 			displayName: this.provider.constructor.displayName,
+	// 			displayIconUni: this.provider.constructor.displayIconUni,
+	// 			displayIconHtml: this.provider.constructor.displayIconHtml,
+	// 		}
+	// 	}
+	// }
+}
+
+Domain.register(ModelPs(Domain))
+
+class DnsProvider extends Table{
+	static _key = 'id';
+	static _keyMap = {
+		'created_by': {isRequired: true, type: 'string', min: 3, max: 500},
+		'created_on': {default: function(){return (new Date).getTime()}},
+		'updated_by': {default:"__NONE__", isRequired: false, type: 'string',},
+		'updated_on': {default: function(){return (new Date).getTime()}, always: true},
+		'id': {default: ()=>crypto.randomBytes(8).toString("hex")},
+		'name': {isRequired: true, type: 'string'},
+		'dnsProvider': {isRequired: true, type: 'string'},
+		'domains': {model:'Domain', rel: 'many', remoteKey: 'dnsProvider_id'}
+	}
+
+	static __intraModel(provider){
+		if(!Object.keys(providers).includes(provider)){
+			throw new Error('Invalid DNS provider');
+		}
+
+		let Provider = providers[provider];
+		let _keyMap = {...this._keyMap, ...Provider._keyMap};
+
+		return ({
+			[this.name] : class extends this {
+				static _keyMap = _keyMap;
+				static Provider = Provider;
+			}
+		})[this.name];
+	}
+
+	static async create(data, ...args){
+		let __intraModel = this.__intraModel(data.dnsProvider)
+		let provider = new __intraModel.Provider(data, ...args);
+
+		return await super.create.call(__intraModel, data, ...args);
+	}
+
+	static async get(data, ...args){
+		let instance = await super.get(data, ...args);
+		let __intraModel = this.__intraModel(instance.dnsProvider)
+
+		return await super.get.call(__intraModel, data, ...args);
+	}
+
+	static listProviders(){
+		let out = [];
+		for(let provider in providers){
+			out.push({
+				name: provider,
+				displayName: providers[provider].displayName || provider,
+				displayIconUni: providers[provider].displayIconUni || '?',
+				displayIconHtml:  providers[provider].displayIconHtml || '<i class="fa-solid fa-question"></i>',
+				fields: providers[provider]._keyMap,
+			});
+		}
+		return out;
+	}
+
+	async updateDomains(){
+		let domains = await this.provider.listDomains();
+		for(let domain of domains){
+			if(!(await Domain.exists(domain.domain))){
+				await Domain.create({
+					created_by: this.created_by,
+					domain: domain.domain,
+					dnsProvider_id: this.id,
+					zoneId: domain.id,
+				});
+			}
+		}
+	}
+
+	async getDomains(){
+		return Domain.getByProviderId(this.id);
+	}
+
+	async remove(){
+		let id = this.id;
+		let instance = await super.remove();
+		for(let domain of await this.getDomains()){
+			await domain.remove();
+		}
+
+		return instance;
+	}
+
+	toJSON(){
+		return {
+			...super.toJSON(),
+			...this.constructor.Provider.toJSON()
+		};
+	}
+}
+
+DnsProvider.register(ModelPs(DnsProvider))
+
+
+if(require.main === module){(async function(){try{
+	const conf = require('../conf');
+
+	console.log(Table.models)
+
+	// console.log(await Domain.listDetail())
+
+	// console.log(JSON.stringify(await Domain.get('ipa.wtf'), null, 2))
+
+	// console.log(JSON.stringify(await Domain.listDetail() ,null, 2))
+	console.log(JSON.stringify(await Table.models.DnsProvider.listDetail() ,null, 2))
+
+
+	// console.log(await Domain.getByProviderId('e8443e03ac503c7b'));
+
+	// console.log(aw)
+
+	process.exit(0)
+
+}catch(error){
+	console.log('IIFE Error:', error);
+}})()}