add the compactPoll logic
Audric Ackermann
4 years ago
parent
675da5cdb2
commit
ad1d5a3c4c
@ -0,0 +1,211 @@
|
|||||||
|
import { getV2OpenGroupRoomByRoomId } from '../../data/opengroups';
|
||||||
|
import {
|
||||||
|
OpenGroupRequestCommonType,
|
||||||
|
OpenGroupV2CompactPollRequest,
|
||||||
|
parseMessages,
|
||||||
|
} from './ApiUtil';
|
||||||
|
import { parseStatusCodeFromOnionRequest } from './OpenGroupAPIV2Parser';
|
||||||
|
import _ from 'lodash';
|
||||||
|
import { sendViaOnion } from '../../session/onions/onionSend';
|
||||||
|
import { OpenGroupManagerV2 } from './OpenGroupManagerV2';
|
||||||
|
import { OpenGroupMessageV2 } from './OpenGroupMessageV2';
|
||||||
|
|
||||||
|
const COMPACT_POLL_ENDPOINT = 'compact_poll';
|
||||||
|
|
||||||
|
export const compactFetchEverything = async (
|
||||||
|
rooms: Array<OpenGroupRequestCommonType>
|
||||||
|
): Promise<null | any> => {
|
||||||
|
// fetch all we need
|
||||||
|
const compactPollRequest = await getCompactPollRequest(rooms);
|
||||||
|
if (!compactPollRequest) {
|
||||||
|
window.log.info('Nothing found to be fetched. returning');
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
const result = await sendOpenGroupV2RequestCompactPoll(compactPollRequest);
|
||||||
|
const statusCode = parseStatusCodeFromOnionRequest(result);
|
||||||
|
if (statusCode !== 200) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
return result;
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* This return body to be used to do the compactPoll
|
||||||
|
*/
|
||||||
|
const getCompactPollRequest = async (
|
||||||
|
rooms: Array<OpenGroupRequestCommonType>
|
||||||
|
): Promise<null | OpenGroupV2CompactPollRequest> => {
|
||||||
|
// first verify the rooms we got are all from on the same server
|
||||||
|
let firstUrl: string;
|
||||||
|
if (rooms) {
|
||||||
|
firstUrl = rooms[0].serverUrl;
|
||||||
|
const anotherUrl = rooms.some(r => r.serverUrl !== firstUrl);
|
||||||
|
if (anotherUrl) {
|
||||||
|
throw new Error('CompactPoll is designed for a single server');
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
window.log.warn('CompactPoll: No room given. nothing to do');
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
const allServerPubKeys: Array<string> = [];
|
||||||
|
|
||||||
|
const roomsRequestInfos = _.compact(
|
||||||
|
await Promise.all(
|
||||||
|
rooms.map(async ({ roomId, serverUrl }) => {
|
||||||
|
try {
|
||||||
|
const fetchedInfo = await getV2OpenGroupRoomByRoomId({
|
||||||
|
serverUrl,
|
||||||
|
roomId,
|
||||||
|
});
|
||||||
|
if (!fetchedInfo) {
|
||||||
|
window.log.warn('Could not find this room getMessages');
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
const {
|
||||||
|
lastMessageFetchedServerID,
|
||||||
|
lastMessageDeletedServerID,
|
||||||
|
token,
|
||||||
|
serverPublicKey,
|
||||||
|
} = fetchedInfo;
|
||||||
|
allServerPubKeys.push(serverPublicKey);
|
||||||
|
const roomRequestContent: Record<string, any> = {
|
||||||
|
room_id: roomId,
|
||||||
|
auth_token: token || '',
|
||||||
|
};
|
||||||
|
if (lastMessageDeletedServerID) {
|
||||||
|
roomRequestContent.from_deletion_server_id = lastMessageDeletedServerID;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (lastMessageFetchedServerID) {
|
||||||
|
roomRequestContent.from_message_server_id = lastMessageFetchedServerID;
|
||||||
|
}
|
||||||
|
|
||||||
|
return roomRequestContent;
|
||||||
|
} catch (e) {
|
||||||
|
window.log.warn('failed to fetch roominfos for room', roomId);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
})
|
||||||
|
)
|
||||||
|
);
|
||||||
|
if (!roomsRequestInfos?.length) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
// double check that all those server pubkeys are the same
|
||||||
|
let firstPubkey: string;
|
||||||
|
if (allServerPubKeys?.length) {
|
||||||
|
firstPubkey = allServerPubKeys[0];
|
||||||
|
const allMatch = allServerPubKeys.every(p => p === firstPubkey);
|
||||||
|
if (!allMatch) {
|
||||||
|
window.log.warn('All pubkeys do not match:', allServerPubKeys);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
window.log.warn('No pubkeys found:', allServerPubKeys);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
const body = JSON.stringify({
|
||||||
|
requests: roomsRequestInfos,
|
||||||
|
});
|
||||||
|
return {
|
||||||
|
body,
|
||||||
|
server: firstUrl,
|
||||||
|
serverPubKey: firstPubkey,
|
||||||
|
endpoint: COMPACT_POLL_ENDPOINT,
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* This call is separate as a lot of the logic is custom (statusCode handled separately, etc)
|
||||||
|
*/
|
||||||
|
async function sendOpenGroupV2RequestCompactPoll(
|
||||||
|
request: OpenGroupV2CompactPollRequest
|
||||||
|
): Promise<Object | null> {
|
||||||
|
const { server, endpoint, body, serverPubKey } = request;
|
||||||
|
// this will throw if the url is not valid
|
||||||
|
const builtUrl = new URL(`${server}/${endpoint}`);
|
||||||
|
|
||||||
|
console.warn(`sending compactPoll request: ${request.body}`);
|
||||||
|
|
||||||
|
const res = await sendViaOnion(serverPubKey, builtUrl, {
|
||||||
|
method: 'POST',
|
||||||
|
body,
|
||||||
|
});
|
||||||
|
|
||||||
|
const statusCode = parseStatusCodeFromOnionRequest(res);
|
||||||
|
if (!statusCode) {
|
||||||
|
window.log.warn(
|
||||||
|
'sendOpenGroupV2Request Got unknown status code; res:',
|
||||||
|
res
|
||||||
|
);
|
||||||
|
return res as object;
|
||||||
|
}
|
||||||
|
|
||||||
|
const results = await parseCompactPollResults(res);
|
||||||
|
|
||||||
|
throw new Error(
|
||||||
|
'See how we handle needs of new tokens, and save stuff to db (last deleted, ... conversation commit, etc'
|
||||||
|
);
|
||||||
|
|
||||||
|
return res as object;
|
||||||
|
}
|
||||||
|
|
||||||
|
type ParsedRoomCompactPollResults = {
|
||||||
|
roomId: string;
|
||||||
|
deletions: Array<number>;
|
||||||
|
messages: Array<OpenGroupMessageV2>;
|
||||||
|
moderators: Array<string>;
|
||||||
|
};
|
||||||
|
|
||||||
|
const parseCompactPollResult = async (
|
||||||
|
singleRoomResult: any
|
||||||
|
): Promise<ParsedRoomCompactPollResults | null> => {
|
||||||
|
const {
|
||||||
|
room_id,
|
||||||
|
deletions: rawDeletions,
|
||||||
|
messages: rawMessages,
|
||||||
|
moderators: rawMods,
|
||||||
|
} = singleRoomResult;
|
||||||
|
|
||||||
|
if (
|
||||||
|
!room_id ||
|
||||||
|
rawDeletions === undefined ||
|
||||||
|
rawMessages === undefined ||
|
||||||
|
rawMods === undefined
|
||||||
|
) {
|
||||||
|
window.log.warn('Invalid compactPoll result', singleRoomResult);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
const validMessages = await parseMessages(rawMessages);
|
||||||
|
const moderators = rawMods as Array<string>;
|
||||||
|
const deletions = rawDeletions as Array<number>;
|
||||||
|
|
||||||
|
return { roomId: room_id, deletions, messages: validMessages, moderators };
|
||||||
|
};
|
||||||
|
|
||||||
|
const parseCompactPollResults = async (
|
||||||
|
res: any
|
||||||
|
): Promise<Array<ParsedRoomCompactPollResults> | null> => {
|
||||||
|
if (
|
||||||
|
!res ||
|
||||||
|
!res.result ||
|
||||||
|
!res.result.results ||
|
||||||
|
!res.result.results.length
|
||||||
|
) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
const arrayOfResults = res.result.results as Array<any>;
|
||||||
|
|
||||||
|
const parsedResults: Array<ParsedRoomCompactPollResults> = _.compact(
|
||||||
|
await Promise.all(arrayOfResults.map(parseCompactPollResult))
|
||||||
|
);
|
||||||
|
|
||||||
|
if (!parsedResults || !parsedResults.length) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
return parsedResults;
|
||||||
|
};
|
||||||
Loading…
Reference in New Issue