misskey/src/misc/cafy-id.ts

34 lines
659 B
TypeScript
Raw Normal View History

2018-04-24 11:13:06 +02:00
import * as mongo from 'mongodb';
2018-07-05 16:36:07 +02:00
import { Context } from 'cafy';
2018-04-24 11:13:06 +02:00
2018-06-18 02:54:53 +02:00
export const isAnId = (x: any) => mongo.ObjectID.isValid(x);
export const isNotAnId = (x: any) => !isAnId(x);
2018-04-24 11:13:06 +02:00
/**
* ID
*/
2018-07-05 16:36:07 +02:00
export default class ID extends Context<mongo.ObjectID> {
2018-05-02 11:06:16 +02:00
constructor() {
super();
2018-04-24 11:13:06 +02:00
this.transform = v => {
if (isAnId(v) && !mongo.ObjectID.prototype.isPrototypeOf(v)) {
return new mongo.ObjectID(v);
} else {
return v;
}
};
2018-05-02 11:06:16 +02:00
this.push(v => {
2018-04-24 11:13:06 +02:00
if (!mongo.ObjectID.prototype.isPrototypeOf(v) && isNotAnId(v)) {
return new Error('must-be-an-id');
}
return true;
});
}
2018-07-06 05:38:07 +02:00
public getType() {
return super.getType('string');
}
2018-04-24 11:13:06 +02:00
}